gorm框架：user role用户角色一对一关联Model编写

一、目录结构

我们建立RBAC应用的目录结构如下图所示

config目录:配置文件例如数据库连接信息

controller目录:所有请求注入控制器来进行处理

model目录:与数据表映射和请求响应需要用到的结构体

static目录:应用静态文件gin需要加载这些静态文件以供应用使用

view目录:应用模板视图文件存放处

二、使用Docker搭建一个简单的mysql环境(确保系统已安装docker)

1.拉取镜像

docker pull mysql:5.7.28

2.镜像拉取成功后，我们在宿主机上来定义Mysql的配置文件，用于挂载容器中的配置文件

新建文件:/usr/local/docker/mysql/my.cnf

[mysql]
prompt = [\\u@\\p][\\d]>\\_
auto-rehash[mysqld]
pid-file    = /var/run/mysqld/mysqld.pid
socket      = /var/run/mysqld/mysqld.sock
datadir     = /var/lib/mysql
#log-error  = /var/log/mysql/error.log
bind-address    = 0.0.0.0
# Disabling symbolic-links is recommended to prevent assorted security risks
symbolic-links=0
port=3306
sql_mode = "STRICT_TRANS_TABLES,NO_ENGINE_SUBSTITUTION,NO_ZERO_DATE,NO_ZERO_IN_DATE,ERROR_FOR_DIVISION_BY_ZERO,NO_AUTO_CREATE_USER"
autocommit = 1
server-id = 11
character_set_server=utf8mb4
transaction_isolation = READ-COMMITTED
explicit_defaults_for_timestamp = 1
max_allowed_packet = 64M
event_scheduler = 1

4.根据镜像生成Mysql容器

docker run -d \
--name mysql \
-p 3306:3306 \
-v /usr/local/docker/mysql/my.cnf:/etc/mysql/mysql.conf.d/mysqld.cnf \
-e MYSQL_ROOT_PASSWORD=root \
--restart="always" \
mysql:5.7.28

上面命令运行了一个mysql实例，并且设置了密码为:root

我们进入容器并创建一个数据库名为:ginrbac

三.使用Gorm来连接并生成Rbac的表结构

什么是ORM?

即Object-Relationl Mapping，它的作用是在关系型数据库和对象之间作一个映射，这样，我们在具体的操作数据库的时候，就不需要再去和复杂的SQL语句打交道，只要像平时操作对象一样操作它就可以了。

Gorm中文文档地址:https://gorm.io/zh_CN/docs/index.html

1.我们在主目录下新建一个config.ini文件，来作为我们的配置文件内容如下：

#gin运行模式:'debug','release'
AppMode = debug
#运行的端口号
AppPort = 8081
#mysql相关配置
MysqlUser = root
MysqlPwd  = root
MysqlDb   = ginrbac
MysqlOption = "charset=utf8mb4&parseTime=True&loc=Local"

我们使用：https://github.com/go-ini/ini类库来读取配置文件

首先我们安装需要用到的2个类库

go get -u gopkg.in/ini.v1
go get -u github.com/jinzhu/gorm"

2.我们在model目录下新建3个文件来对应“用色”，“用户”，“权限”表

2.1 role.go文件:

package modelimport ("github.com/jinzhu/gorm"
)type Role struct {gorm.ModelName   string `json:"name" gorm:"type:VARCHAR(30);not null;comment:'角色名称'"`Rules  string `json:"rules" gorm:"type:VARCHAR(255);not null;comment:'角色拥有的权限节点'"`Status int    `json:"status" gorm:"type:TINYINT(1);not null;default:1;comment:'状态：1 启用 0 禁用'"`
}

2.2 user.go文件:

package modelimport ("github.com/jinzhu/gorm"
)type User struct {gorm.ModelUsername string `json:"username" gorm:"type:varchar(50);not null; comment:'用户名'"`Password string `json:"password" gorm:"type:varchar(128);not null; comment:'密码'"`RoleId   int    `json:"role_id" gorm:"type:int;not null; comment:'角色id'"`Status   int    `json:"status" gorm:"type:TINYINT(1);not null; comment:'状态：1启用 0禁用';default:1"`
}

2.3 node.go文件(权限节点表):

package modelimport ("github.com/jinzhu/gorm"
)type Node struct {gorm.ModelName   string `json:"name" gorm:"type:VARCHAR(50);not null;comment:'节点名称'"`Path   string `json:"path" gorm:"type:VARCHAR(50);not null;comment:'节点路径'"`Pid    int    `json:"pid" gorm:"type:INT(11);not null;comment:'所属节点id'"`Sort   int    `json:"sort" gorm:"type:INT(4);not null;default:0;comment:'排序'"`Icon   string `json:"icon" gorm:"type:VARCHAR(50);default:'';comment:'图标'"`IsMenu int    `json:"is_menu" gorm:"type:TINYINT(1);not null;default:1;comment:'是否是菜单项 1 不是 2 是'"`Status int    `json:"status" gorm:"type:TINYINT(1);not null;default:1;comment:'状态：1 启用 0 禁用'"`
}

3.我们在config目录下新建config.go文件内容如下:

 package configimport ("fmt"_ "github.com/go-sql-driver/mysql""github.com/jinzhu/gorm""gopkg.in/ini.v1""log""myblog/model"
)var Conf *config //应用配置变量，可以通过它来读取
var Db *gorm.DB  //应用mysql连接池type config struct {AppMode     stringAppPort     stringMysqlUser   stringMysqlPwd    stringMysqlIpPort stringMysqlDb     stringMysqlOption string
}func init() {initConfig()initDb()   //调用同级目录下db.go中的方法genTable() //生成修改表结构
}
func initConfig() {cfg, err := ini.Load("./config.ini")if err != nil {log.Panic("加载ini错误！")}Conf = &config{}err = cfg.MapTo(Conf)if err != nil {log.Panic("解析ini文件错误", err)}
}func initDb() {var err error//gorm.Open("mysql", "user:password@tcp(127.0.0.1:3306)/dbname?charset=utf8&parseTime=True&loc=Local")dnsStr := fmt.Sprintf("%s:%s@tcp(%s)/%s?%s", Conf.MysqlUser, Conf.MysqlPwd, Conf.MysqlIpPort, Conf.MysqlDb, Conf.MysqlOption)Db, err = gorm.Open("mysql", dnsStr)if err != nil {log.Panic("连接mysql有错误:", err)}
}
func genTable() {Db.AutoMigrate(&model.User{})Db.AutoMigrate(&model.Role{})Db.AutoMigrate(&model.Node{})
}

config.go文件中生成表结构的关键函数：genTable()

参考链接：http://gincms.com/blog/detail-2.html