golang webserver 引入数据库连接的较好方式
🕡 Mon, 03 Jul 2017 by GolangNote
这是个人认为golang webserver 引入数据库连接的较好方式,代码也比较优美
package main
import (
"database/sql"
"net/http"
"strconv"
_ "github.com/go-sql-driver/mysql"
)
type Dependencies struct {
Database *sql.DB
// also logging
// and metrics
// and service consumers
// whatever
}
func (d Dependencies) GetDatabase() *sql.DB {
return d.Database
}
type APIv1 struct {
Dependencies
// any APIv1-specific dependencies
}
type Databaser interface {
GetDatabase() *sql.DB
}
func main() {
myDB, err := sql.Open("mysql", "mysql-dsn-goes-here")
if err != nil {
panic(err)
}
coreDeps := Dependencies{Database: myDB}
apiv1 := APIv1{Dependencies: coreDeps}
http.Handle("/v1", apiv1)
// ListenAndServe, you know the rest
}
func minGophersMiddleware(d Databaser, next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// call our handler only if there are more than 5 gophers
db := d.GetDatabase()
count, err := getCount(db)
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
return
}
if count < 5 {
w.WriteHeader(http.StatusBadRequest)
w.Write([]byte("Error: not enough gophers"))
return
}
next.ServeHTTP(w, r)
})
}
func (a APIv1) ServeHTTP(w http.ResponseWriter, r *http.Request) {
// use a router package here
// in this example we’re just gonna respond with the same
// handler to every request
minGophersMiddleware(a.Dependencies, http.HandlerFunc(a.handler)).ServeHTTP(w, r)
}
func (a APIv1) handler(w http.ResponseWriter, r *http.Request) {
count, err := getCount(a.Database)
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
return
}
w.Write([]byte(strconv.Itoa(count) + " gophers"))
}
func getCount(db *sql.DB) (int, error) {
var count int
err := db.QueryRow("SELECT COUNT(*) FROM animals WHERE name=?", "gopher").Scan(&count)
return count, err
}
本文网址: https://golangnote.com/topic/176.html (转载注明出处)
关于GolangNote:记录在工作中使用golang 遇到、面临的相关问题及解决方法。如果你在这里获得一些知识或信息,解决你的编程问题,请考虑捐赠给不幸的人或者你喜欢的慈善机构,除捐赠外,种植树木、志愿服务或减少排碳的行为也很有益处。如果你有任何问题可以在下面 留言
Be the first to comment!
Relative Articles
Recent Go Articles
- Golang 把cookie 字符串解析为cookie 结构
- Golang 计算字符串中包含某个或某些字符集的个数
- 使用Golang 对文件增删写读操作备忘
- Go Modules 使用备忘
- 使用Golang 简单删除图片exif 信息
- 谷歌翻译的 golang 库推荐
- Go 1.13.2 与1.13.3 紧急更新
- golang 人脸检测识别库
- Go build 错误 “stackcheck redeclared in this block previous declaration”的解决方法
- Golang phantomjs 动态代理实现
- Golang chrome debug protocol 库推荐
- Golang 随机打乱数组/Slice
- Golang sync.WaitGroup 的 Wait 超时处理
- Golang实现简单的Socks5代理
- Golang 用snappy + Base64 简单压缩加密进行网络传输
- Golang http IPv4/IPv6 服务
- golang 全角半角相互转换
- 在自己的网站部署TLS 1.3
- Golang 实现/打印菜单树
- Golang telegram 机器人小试
Top Go Articles
- 用golang 标准库regexp 验证用户名和密码的合法性
- Golang http IPv4/IPv6 服务
- mismatched types time.Duration and int
- golang 用gzip 压缩、解压缩字符串
- golang 简单全局计数器实现
- Golang sync.WaitGroup 的 Wait 超时处理
- Golang telegram 机器人小试
- Golang 并发控制的两种模式
- go 获取硬盘的可用空间的方法
- iris websocket 自定义信息结构
- golang map 按value 大小排序,降序和升序
- Golang Leveldb 基本使用的例子
- go post 上传文件的例子
- golang 生成良好的唯一ID/uuid库比较
- golang 获取用户真实访问ip 地址
- Golang webview 做桌面应用及跨平台编译
- Golang http.Post 用最小的内存发送大文件
- golang image 智能合并图片
- Golang string 和byte 操作性能比较
- golang 使用crypto/rand 生成随机数