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

野性酷女 2023-10-01 10:20 9阅读 0赞

![在这里插入图片描述][d56836f8702141c38e6c08091598f33c.png]

##### 一、目录结构 #####

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

![img][]

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 model
    
    import (
    	"github.com/jinzhu/gorm"
    )
    
    type Role struct {
    	gorm.Model
    	Name   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 model
    
    import (
    	"github.com/jinzhu/gorm"
    )
    
    type User struct {
    	gorm.Model
    	Username 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 model
    
    import (
    	"github.com/jinzhu/gorm"
    )
    
    type Node struct {
    	gorm.Model
    	Name   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 config
    
    import (
    	"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     string
    	AppPort     string
    	MysqlUser   string
    	MysqlPwd    string
    	MysqlIpPort string
    	MysqlDb     string
    	MysqlOption 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

[d56836f8702141c38e6c08091598f33c.png]: https://img-blog.csdnimg.cn/d56836f8702141c38e6c08091598f33c.png
[img]: https://img-blog.csdnimg.cn/img_convert/cacc3632dbe03cdc5d18ce7e89e883a2.png