使用配置文件配置程序启动

This commit is contained in:
yumaojun03 2024-11-16 18:11:24 +08:00
parent bb8d788610
commit fda31d9172
5 changed files with 179 additions and 6 deletions

10
book/application.yaml Normal file
View File

@ -0,0 +1,10 @@
app:
host: 127.0.0.1
port: 8080
mysql:
host: 127.0.0.1
port: 3306
database: test
username: "root"
password: "123456"
debug: false

View File

@ -1,8 +1,6 @@
# 程序配置管理
## 测试验证包
要考虑测序的是否如预期运行

View File

@ -1,6 +1,6 @@
app:
host: localhost
port: 8090
port: 8080
mysql:
host: 127.0.0.1
port: 3306

View File

@ -1,6 +1,13 @@
package config
import "encoding/json"
import (
"encoding/json"
"fmt"
"sync"
"gorm.io/driver/mysql"
"gorm.io/gorm"
)
// 定义程序配置
// dsn := "root:123456@tcp(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local"
@ -31,6 +38,10 @@ type App struct {
Port int `json:"port"`
}
func (c *App) Address() string {
return fmt.Sprintf("%s:%d", c.Host, c.Port)
}
// mysql:
//
// host: 127.0.0.1
@ -46,4 +57,35 @@ type MySQL struct {
Username string `json:"username"`
Password string `json:"password"`
Debug bool `json:"debug"`
db *gorm.DB
lock sync.Mutex
}
// 初始化数据库, 能过与数据库交互的 连接池对象: db
// 复用, 只要有就用 之前的没有才初始化新的db对象
func (c *MySQL) DB() *gorm.DB {
c.lock.Lock()
defer c.lock.Unlock()
if c.db == nil {
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
c.Username,
c.Password,
c.Host,
c.Port,
c.Database,
)
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
if err != nil {
panic("failed to connect database")
}
if c.Debug {
db = db.Debug()
}
c.db = db
}
return c.db
}

View File

@ -2,11 +2,19 @@ package main
import (
"fmt"
"log"
"net/http"
"os"
"github.com/gin-gonic/gin"
"gitlab.com/go-course-project/go17/book/config"
)
func Failed(ctx *gin.Context, err error) {
ctx.JSON(http.StatusBadRequest, gin.H{"code": 0, "msg": err.Error()})
}
// 规定好风格: JSON Restful Api
func main() {
// 从配置文件中加载配置
// 加载配置
@ -14,9 +22,124 @@ func main() {
if path == "" {
path = "application.yaml"
}
config.LoadConfigFromYaml(path)
if err := config.LoadConfigFromYaml(path); err != nil {
fmt.Printf("加载配置错误: %s\n", err)
os.Exit(1)
}
// 访问加载后的配置
conf := config.Get()
fmt.Println(conf)
// gin Engine, 它包装了http server
server := gin.Default()
db := conf.MySQL.DB()
// 配置业务路有, Book类型的资源的 一套简单的CRUD
// 创建Book -> Book
// POST, Body
book := server.Group("/api/books")
book.POST("", func(ctx *gin.Context) {
// 获取Book用户传达的参数
ins := new(Book)
if err := ctx.ShouldBindJSON(ins); err != nil {
Failed(ctx, err)
return
}
// book, save
if err := db.Save(ins).Error; err != nil {
Failed(ctx, err)
return
}
ctx.JSON(http.StatusOK, ins)
})
// 查询Book列表 -> []*Book
// SELECT * FROM `books`
book.GET("", func(ctx *gin.Context) {
var books []Book
if err := db.Find(&books).Error; err != nil {
Failed(ctx, err)
return
}
ctx.JSON(http.StatusOK, books)
})
// 查询Book详情
book.GET("/:isbn", func(ctx *gin.Context) {
var ins Book
id := ctx.Param("isbn")
if err := db.Where("isbn = ?", id).Take(&ins).Error; err != nil {
Failed(ctx, fmt.Errorf("Book not found"))
return
}
ctx.JSON(http.StatusOK, ins)
})
// 更新Book
book.PUT("/:isbn", func(ctx *gin.Context) {
id := ctx.Param("isbn")
// 获取用户参数, 读取用户的更新参数
req := BookSpec{}
if err := ctx.ShouldBindJSON(&req); err != nil {
Failed(ctx, err)
return
}
// gorm更新是如果字段为零值 就不更新该字段, is_sale 没办法更新为false
// 如果越到零值,也需要更新,则需要转化为 指针类型
if err := db.Where("isbn = ?", id).Model(&Book{}).Updates(req).Error; err != nil {
Failed(ctx, err)
return
}
// 针对有零值的字段独立更新
// if err := db.Where("isbn = ?", id).Model(&Book{}).Update("is_sale", req.IsSale).Error; err != nil {
// Failed(ctx, err)
// return
// }
// // ......
// 再次查询出来
var ins Book
if err := db.Where("isbn = ?", id).Take(&ins).Error; err != nil {
Failed(ctx, fmt.Errorf("Book not found"))
return
}
// 查询出更新后的数据
ctx.JSON(http.StatusOK, ins)
})
// 删除Book
book.DELETE("/:isbn", func(ctx *gin.Context) {
id := ctx.Param("isbn")
if err := db.Where("isbn = ?", id).Delete(&Book{}).Error; err != nil {
Failed(ctx, err)
return
}
})
if err := server.Run(conf.App.Address()); err != nil {
log.Println(err)
}
}
// Book 结构体定义
type Book struct {
// grom:"column:isbn;", 具体文档: https://gorm.io/docs/models.html#Fields-Tags
IsBN uint `json:"isbn" gorm:"primaryKey;column:isbn"`
BookSpec
}
type BookSpec struct {
Title string `json:"title" gorm:"column:title;type:varchar(200)"`
Author string `json:"author" gorm:"column:author;type:varchar(200);index"`
Price float64 `json:"price" gorm:"column:price"`
// bool false
// nil 是零值, false
IsSale *bool `json:"is_sale" gorm:"column:is_sale"`
}
// 定义该对象映射到数据里 表的名称
func (t *Book) TableName() string {
return "books"
}