使用配置文件配置程序启动
This commit is contained in:
parent
bb8d788610
commit
fda31d9172
10
book/application.yaml
Normal file
10
book/application.yaml
Normal 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
|
||||
@ -1,8 +1,6 @@
|
||||
# 程序配置管理
|
||||
|
||||
|
||||
|
||||
|
||||
## 测试验证包
|
||||
|
||||
要考虑测序的是否如预期运行
|
||||
|
||||
@ -1,6 +1,6 @@
|
||||
app:
|
||||
host: localhost
|
||||
port: 8090
|
||||
port: 8080
|
||||
mysql:
|
||||
host: 127.0.0.1
|
||||
port: 3306
|
||||
|
||||
@ -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
|
||||
}
|
||||
|
||||
127
book/main.go
127
book/main.go
@ -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"
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user