网站建设公司-信科网络,wordpress落叶插件,企业免费网站优化方案,wordpress 301跳转目录文件结构数据库准备配置文件处理config.godb_list.gogorm_mysql.gosystem.go初始化数据库gorm.gogorm_mysql.go开始初始化测试数据库定义实体类 Userserviceapi开始测试#xff01;文件结构 本文章将使用到上一节创建的 CommonService 接口#xff0c;用于测试连接数据库…
目录文件结构数据库准备配置文件处理config.godb_list.gogorm_mysql.gosystem.go初始化数据库gorm.gogorm_mysql.go开始初始化测试数据库定义实体类 Userserviceapi开始测试文件结构 本文章将使用到上一节创建的 CommonService 接口用于测试连接数据库状况当然您也可以自制简单接口作为测试不会影响整体工作流程 下方树状图为最终项目结构
文件夹 api 与 service 助力数据库查询请求
initialize/gorm.go 初始化 GORM initialize/gorm_mysql.go 执行 MySQL 数据库对应的初始化过程
user.go 一个实体类
│ config.yaml
│ go.mod
│ go.sum
│ main.go
│
├─api
│ │ enter.go
│ │
│ ├─example
│ └─system
│ api_common.go
│ enter.go
│
├─config
│ config.go
│ db_list.go
│ gorm_mysql.go
│ jwt.go
│ local.go
│ system.go
│
├─core
│ constant.go
│ server.go
│ viper.go
│
├─global
│ global.go
│
├─initialize
│ │ gorm.go
│ │ gorm_mysql.go
│ │ router.go
│ │
│ └─internal
│ gorm.go
│
├─model
│ user.go
│
├─router
│ │ enter.go
│ │
│ ├─example
│ └─system
│ enter.go
│ sys_common.go
│
└─service 之后回到项目根目录打开命令行输入指令安装 MySQL 驱动 go get -u gorm.io/driver/mysql 数据库准备
这是我本地使用的数据库参数 username:root password:123456 dbname:golang
这是数据库“golang”中的数据表 user 的数据注意到有四个属性 来到 go 项目中根据上述参数打开 config.yaml添加 mysql 配置 这里为了便于大家操作省去了源码中的注册 zap 的部分
同时添加 system 对象内含一个 db-type 属性用来定义我们希望后端链接的数据库类型
system:db-type: mysqlmysql:path: localhostport: 3306config: db-name: golangusername: rootpassword: 123456max-idle-conns: 10max-open-conns: 100配置文件处理 由于我们为 config.yaml 添加了新的配置项所以需要按照我们之前文章所属为其配置指定的结构 config.go 打开如果你跟随上上节文章或者新建文件 config/config.go 在主 config 入口添加 mysql 属性
type Server struct {...Mysql Mysql mapstructure:mysql json:mysql yaml:mysql
}db_list.go 新建文件 config/db_list.go 由于我们有可能需要支持多种类型的数据库比如 PostgreSQL所以最佳解决办法就是创建一个通用的数据库参数结构体
DSNProvider 接口用来给 GORM 提供 DSN
type DSNProvider interface {Dsn() string
}type GeneralDB struct {Path string mapstructure:path json:path yaml:path // 服务器地址:端口Port string mapstructure:port json:port yaml:port //:端口Config string mapstructure:config json:config yaml:config // 高级配置Dbname string mapstructure:db-name json:db-name yaml:db-name // 数据库名Username string mapstructure:username json:username yaml:username // 数据库用户名Password string mapstructure:password json:password yaml:password // 数据库密码Prefix string mapstructure:prefix json:prefix yaml:prefix //全局表前缀单独定义TableName则不生效Singular bool mapstructure:singular json:singular yaml:singular //是否开启全局禁用复数true表示开启Engine string mapstructure:engine json:engine yaml:engine default:InnoDB //数据库引擎默认InnoDBMaxIdleConns int mapstructure:max-idle-conns json:max-idle-conns yaml:max-idle-conns // 空闲中的最大连接数MaxOpenConns int mapstructure:max-open-conns json:max-open-conns yaml:max-open-conns // 打开到数据库的最大连接数
}gorm_mysql.go 新建文件 gorm_mysql.go 在这里定义 mysql 结构体直接使用通用数据库参数结构体 GeneralDB 即可
然后于此实现接口 DSNProvider 构造 DSN
type Mysql struct {GeneralDB yaml:,inline mapstructure:,squash
}func (m *Mysql) Dsn() string {return m.Username : m.Password tcp( m.Path : m.Port )/ m.Dbname ? m.Config
}system.go 打开文件 config/system.go 老样子为 db_type 配置对应结构体内容
type System struct {DbType string mapstructure:db-type json:db-type yaml:db-type
}初始化数据库
gorm.go 新建文件 initialize/gorm.go 此处的逻辑很简单即通过全局变量获取 db_type 属性得知开发者想要用那种数据库 由于我们只会用到 MySQL故这里的判断就都启动 mysql 了
GormMysql 目前处于飘红状态因为还没有定义下面我们将为其定义初始化 MySQL 的方法
import (ginlogin/globalgorm.io/gorm
)func Gorm() *gorm.DB {switch global.G_CONFIG.System.DbType {case mysql:return GormMysql()default:return GormMysql()}
}gorm_mysql.go 新建文件 initialize/gorm_mysql.go 具体通过 GORM 链接 mysql 的过程相信大家也很熟悉这里就不多费口舌了直接看注释吧
import (fmtginlogin/globalgorm.io/driver/mysqlgorm.io/gorm
)// 初始化mysql方法返回一个gorm.DB引用
func GormMysql() *gorm.DB {// 第一步// 全局变量获取config中定义的mysql参数// 判断如果不急于数据库名直接返回nil表示初始化数据库失败info : global.G_CONFIG.Mysqlif info.Dbname {return nil}// 第二步// 构造mysqlConfigmysqlConfig : mysql.Config{DSN: info.Dsn(),DefaultStringSize: 191,SkipInitializeWithVersion: false,}// 第三步// gorm.Open连接到数据库并处理错误if db, err : gorm.Open(mysql.New(mysqlConfig)); err ! nil {fmt.Println(完蛋连接不上数据库)return nil} else {// 如果成功链接设置好连接池啥的就可以返回db了db.InstanceSet(gorm:table_options, ENGINEinfo.Engine)sqlDB, _ : db.DB()sqlDB.SetMaxIdleConns(info.MaxIdleConns)sqlDB.SetMaxOpenConns(info.MaxOpenConns)return db}
}开始初始化
哎这还没完你还要调用初始化方法
打开 main.go添加如下代码调用 GORM 初始化方法
func main() {// 初始化数据库DBglobal.G_DB initialize.Gorm()...
}此处关于整合 mysql 的部分已经完结了下面将是测试数据库的内容您可以直接略过 测试数据库
定义实体类 User 新建文件 model/user.go 根据开头所述 user 表故可定义如下实体类
type User struct {Id int json:idUsername string json:usernamePassword string json:passwordRole int json:role
}// 显式指定表名
func (User) TableName() string {return user
}service 打开文件 service/system/service_common.go service 层内简单实现了查询整张表的功能并作了错误处理
type CommonService struct{}func (s *CommonService) CreateApi(c *gin.Context) (err error) {// 调用Find方法查询表将查询成功与否的结果存到res变量var userList []model.Userres : global.G_DB.Find(userList)// 判断res是否出现了错误if errors.Is(res.Error, gorm.ErrRecordNotFound) {return errors.New(找不到这个元素)}// 如果没错那就返回数据c.JSON(http.StatusOK, gin.H{data: userList,})// 给api层传递一个errorreturn res.Error
}api 打开文件 api/system/api_common.go 很简单调用 service然后在 api 中处理错误并返回 JSON 数据这一步也可以直接全部写在 service 层里此时就可以把该 api 层看做是 springboot 中的 controller 层了
import (github.com/gin-gonic/ginnet/http
)type CommonApi struct{}func (s *CommonApi) CreateApi(c *gin.Context) {err : commonService.CreateApi(c)if err ! nil {c.JSON(http.StatusBadRequest, gin.H{status: err.Error(),})}
}开始测试
万事俱备只欠东风打开你的 postman|apifox|apipost请求以下看看结果吧
