go语言实现订单管理系统

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› Go›Go教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

写此系统的初衷实为加强对所学部分东西的理解与掌握，简洁的同时达到目的，本文代码/项目的形成正是一个学习和加强实践的过程。

基于macaron+viper+gorm框架（关于macaron、gorm可移步：https://blog.csdn.net/HYZX_9987/article/details/100022849

https://blog.csdn.net/HYZX_9987/article/details/99681632）

项目结构

配置文件部分

路由控制器部分

对象关系映射部分

项目结构

配置文件部分
1、配置文件内容及其存放路径

2、配置文件的解析与内容的注入

结构体

注入配置

提示：启动项目时也可以刻意不写配置文件，viper读取配置时会加载路径，如果没有文件会提示在xx路径下找不到config.yaml

这时你再把文件放到xx路径也行。

路由控制器部分
1、初始化马卡龙组件

m := macaron.Classic()
m.Use(macaron.Renderer())
m.Use(macaron.Recovery())
//test
m.Get("/", func(ctx *macaron.Context) macaron.ReturnStruct {
return macaron.ReturnStruct{Msg: "Hello world!", Data: "Your macaron."}
})
2、分组路由使用

本系统不需要太复杂的逻辑，暂定实体为客户、订单、订单项，因此分三组路由，前缀以/api开始

3、各模块路由控制器（以Customer部分为例）

路由定义：

/*
马卡龙数据绑定：go get github.com/go-macaron/binding，移入pkg包
*/
func CusRoute(m *macaron.Macaron) {
//http://localhost:8081/api/cus
m.Get("/cus", ListCustomer)
//http://localhost:8081/api/cus/1
m.Get("/cus/:ID", GetCustomer)
m.Post("/cus", binding.Bind(model.Customer{}), CreateCustomer)
m.Delete("/cus/:ID", DeleteCustomer)
m.Put("/cus", binding.Bind(model.Customer{}), UpdateCustomer)
}
对应处理器：

func CusRoute(m *macaron.Macaron) {

//http://localhost:8081/api/cus
m.Get("/cus", ListCustomer)
//http://localhost:8081/api/cus/1
m.Get("/cus/:ID", GetCustomer)
m.Post("/cus", binding.Bind(model.Customer{}), CreateCustomer)
m.Delete("/cus/:ID", DeleteCustomer)
m.Put("/cus", binding.Bind(model.Customer{}), UpdateCustomer)
}

func ListCustomer(ctx *macaron.Context) macaron.ReturnStruct {
var err error
cusList, err := service.ListCustomerInternal()
if err != nil {
errMsg := "list customer err:" + err.Error()
ctx.Resp.Header().Set("error-info", errMsg)
log.Fatal(errMsg)
return macaron.ReturnStruct{Msg: errMsg}
}
return macaron.ReturnStruct{Data: cusList, Code: http.StatusOK}
}

func GetCustomer(ctx *macaron.Context) macaron.ReturnStruct {
cus, err := service.GetCustomerByIDInternal(ctx.Params("ID"))
if err != nil {
errMsg := "get customer err:" + err.Error()
ctx.Resp.Header().Set("error-info", errMsg)
log.Fatal(errMsg)
return macaron.ReturnStruct{Msg: errMsg}
}
return macaron.ReturnStruct{Data: cus, Code: http.StatusOK}
}

func CreateCustomer(customer model.Customer, ctx *macaron.Context) macaron.ReturnStruct {
err := service.CreateCustomerInternal(customer)
if err != nil {
errMsg := "create customer err:" + err.Error()
ctx.Resp.Header().Set("error-info", errMsg)
log.Fatal(errMsg)
return macaron.ReturnStruct{Msg: errMsg}
}
return macaron.ReturnStruct{Code: http.StatusOK}
}

func DeleteCustomer(ctx *macaron.Context) macaron.ReturnStruct {
err := service.DeleteCustomerByIDInternal(ctx.Params("ID"))
if err != nil {
errMsg := "del customer err:" + err.Error()
ctx.Resp.Header().Set("error-info", errMsg)
log.Fatal(errMsg)
return macaron.ReturnStruct{Msg: errMsg}
}
return macaron.ReturnStruct{Code: http.StatusOK}
}

func UpdateCustomer(customer model.Customer, ctx *macaron.Context) macaron.ReturnStruct {
err := service.UpdateCustomerInternal(customer)
if err != nil {
errMsg := "update customer err:" + err.Error()
ctx.Resp.Header().Set("error-info", errMsg)
log.Fatal(errMsg)
return macaron.ReturnStruct{Msg: errMsg}
}
return macaron.ReturnStruct{Code: http.StatusOK}
}
Service层（非必须，web层可直接与持久层交互）：

func ListCustomerInternal() (cusList []model.Customer, err error) {
cusList, err = dao.ListCustomerByDB()
return cusList, err
}

func GetCustomerByIDInternal(ID string) (cus model.Customer, err error) {
cus, err = dao.GetCustomerByIDByDB(ID)
return cus, err
}

func DeleteCustomerByIDInternal(ID string) (err error) {
err = dao.DeleteCustomerByIDByDB(ID)
return err
}

func CreateCustomerInternal(cus model.Customer) (err error) {
err = dao.CreateCustomerByDB(cus)
return err
}

func UpdateCustomerInternal(cus model.Customer) (err error) {
err = dao.UpdateCustomerByDB(cus)
return err
}
对象关系映射部分
1、gorm开启连接、初始化配置

2、表初始化

3、各模块db操作（持久层）

func ListCustomerByDB() (list []model.Customer, err error) {
db := config.Connection()
err = db.Find(&list).Error
return list, err
}

func GetCustomerByIDByDB(ID string) (cus model.Customer, err error) {
db := config.Connection()
err = db.Where("id = ?", ID).Find(&cus).Error
return cus, err
}

func DeleteCustomerByIDByDB(ID string) (err error) {
db := config.Connection()
err = db.Where("id = ?", ID).Delete(&model.Customer{}).Error
return err
}

func CreateCustomerByDB(cus model.Customer) (err error) {
db := config.Connection()
err = db.Save(&cus).Error
return err
}

func UpdateCustomerByDB(cus model.Customer) (err error) {
db := config.Connection()
err = db.Update(&cus).Error
return err
}
一切就绪后，启动项目：

调接口看效果：

控制台日志：

访问控制
实际项目中往往都加了登录控制以及token校验之类的东西，我们在这模拟一下：

m.Group("/api", func() {
m.Group("", func() {
controller.CusRoute(m)
})

m.Group("", func() {
controller.OrderRoute(m)
})

m.Group("", func() {
controller.ItemRoute(m)
})
},CheckToken)
让所有进来的请求都得执行CheckToken函数，函数如下：

func CheckToken(ctx *macaron.Context) {
if auth := ctx.Req.Header.Get("auth"); auth == "123" {
log.Println("token success.")
return
}
ctx.JSON(http.StatusUnauthorized, macaron.ReturnStruct{Code: http.StatusUnauthorized, Msg: "invalid Token."})
}
此时如果继续按刚才的请求访问，那么会被校验住：

如果加了token的验证，那么将一路顺风：

————————————————
版权声明：本文为CSDN博主「_雨落山岚」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/HYZX_9987/article/details/100583800