当前位置：首页 > news >正文

go gin学习记录5

news 2025/8/23 0:13:48

有了前面几节的学习，如果做个简单的web服务端已经可以完成了。
这节来做一下优化。

我们实验了3种SQL写入的方法，但是发现每一种都需要在方法中去做数据库链接的操作，有些重复了。
所以，我们把这部分提取出来，数据库链接的功能提取出来，数据库的配置也提取出来。
这就是今天这节要做的内容，配置化。

配置文件通常是放在conf目录下，所以我们在t_gin/下新建一个conf目录，用来存放配置文件。
然后在conf目录下，新建一个service.conf文件，作为实验用的配置文件。
完成后的目录结果如下：
在这里插入图片描述
配置文件这里使用.ini的格式。
要解析.ini格式，我们需要下载对应的package：

$ go get gopkg.in/ini.v1

将package安装好之后，我们再来想一下配置文件里放什么。
首先，我们可以将数据库配置放进去，简单点的就是dsn，这个在多场景都会用到，就不用在每个文件中去重复定义了。
在service.conf中添加数据库配置：

[mysql]
dsn = "root:@tcp(127.0.0.1:3306)/t_gin?charset=utf8&parseTime=true"

配置添加好了，怎么用呢？
方便一点，就继续使用上一节的worker.go文件来进行改造。
要做的是：
1.将数据库链接操作提取出来，不用每个方法都写一遍；
2.将dsn配置改为从配置文件service.conf中获取，不用每个文件都写一遍。
开始撸码：

func (w *WorkerController) init() *sql.DB {cfg, err := ini.Load("conf/service.conf")if err != nil {log.Println(err.Error())}mysqlDsn := cfg.Section("mysql").Key("dsn").String()db, err := sql.Open("mysql", mysqlDsn)if err != nil {log.Panic(err.Error())}return db
}

init方法是我们的简单实现。
从service.conf中读取了dsn信息，然后建立数据库链接，并返回链接。
那方法提取出来了，文件中之前的代码（插入、查询、更新）我们也同步进行调整。
调整之后的worker.go文件内容如下：

package controllerimport ("database/sql""github.com/gin-gonic/gin""github.com/huandu/go-sqlbuilder""gopkg.in/ini.v1""log""net/http"
)type WorkerController struct {
}type Worker struct {ID    int    `json:"id"`Name  string `json:"name"`Birth string `json:"birth"`
}func (w *WorkerController) init() *sql.DB {cfg, err := ini.Load("conf/service.conf")if err != nil {log.Println(err.Error())}mysqlDsn := cfg.Section("mysql").Key("dsn").String()db, err := sql.Open("mysql", mysqlDsn)if err != nil {log.Panic(err.Error())}return db
}func (w *WorkerController) CreateWorker(c *gin.Context) {db := w.init()name := c.PostForm("name")birth := c.PostForm("birth")sb := sqlbuilder.NewInsertBuilder()sb.InsertInto("worker")sb.Cols("name", "birth")sb.Values(name, birth)sqlString, args := sb.Build()log.Println(sqlString, args)_, err := db.Exec(sqlString, args...)if err != nil {log.Panic(err.Error())}c.JSON(http.StatusOK, gin.H{"code": 0,"data": true,})
}func (w *WorkerController) GetWorkerInfo(c *gin.Context) {db := w.init()id := c.Query("id")sb := sqlbuilder.NewSelectBuilder()sb.From("worker")sb.Select("id,name,birth")sb.Where(sb.Equal("id", id))sqlString, args := sb.Build()log.Println(sqlString, args)result := db.QueryRow(sqlString, args...)var worker Workererr := result.Scan(&worker.ID, &worker.Name, &worker.Birth)if err != nil {log.Panic(err.Error())}c.JSON(http.StatusOK, gin.H{"code": 0,"data": worker,})
}func (w *WorkerController) UpdateWorkerInfo(c *gin.Context) {db := w.init()id := c.PostForm("id")name := c.PostForm("name")sb := sqlbuilder.NewUpdateBuilder()sb.Update("worker")sb.Where(sb.Equal("id", id))sb.SetMore(sb.Assign("name", name))sqlString, args := sb.Build()log.Println(sqlString, args)_, err := db.Exec(sqlString, args...)if err != nil {log.Panic(err.Error())}c.JSON(http.StatusOK, gin.H{"code": 0,"data": true,})
}