上一节抽离了captcha验证码功能,集成了验证码微服务功能,这一节来看看后台Rbac功能,并抽离其中的用户登录,管理员管理,角色管理,权限管理等功能作为微服务来调用

一.引入

后台操作从登录到后台首页,然后其中的管理员管理,角色管理,权限管理等功能可以抽离出来作为一个Rbac微服务,因为他们都是共同关系的,并且使用人员也不是太多,当然,也可以每个管理功能单独作为一个微服务来进行操作

二.创建权限管理Rbac微服务服务端以及实现后台用户登录微服务

1.生成权限管理Rbac微服务代码

在server目录下运行: go-micro new service rbac,生成 rbac微服务服务端代码

2.编写proto/rbac.proto文件,实现用户登录微服务逻辑

在这个文件中编写用户登录相关代码,可参考: [golang gin框架] 13.Gin 商城项目-配置公共基类实现公共的成功,失败提示页面用户登录、退出登录、以及权限判断

参考代码 controllers/admin/loginController.go如下:

//执行登录操作
func (con LoginController) DoIndex(c *gin.Context) {//获取表单中的数据captchaId := c.PostForm("captchaId")     // 验证码idverifyValue := c.PostForm("verifyValue") //验证码的值//获取用户名以及密码username := c.PostForm("username")password := c.PostForm("password")// 1.判断验证码是否验证成功: 这里调用的是上一节验证码微服务的功能if flag := models.VerifyCaptcha(captchaId, verifyValue); flag {//2.查询数据库,判断用户以及密码是否正确userinfo := []models.Manager{}password = models.Md5(password)models.DB.Where("username = ? and password = ? ", username, password).Find(&userinfo)if len(userinfo) > 0 {//3.执行登录,保存用户信息,执行跳转操作session := sessions.Default(c)//注意: session.Set没法保存结构体对应的切片,所以需要把结构体转换成json字符串userinfoSlice, _ := json.Marshal(userinfo)session.Set("userinfo_admin", string(userinfoSlice))session.Save()con.Success(c, "登录成功", "/admin")} else {con.Error(c, "用户名或密码错误", "/admin/login")}} else {con.Error(c, "验证码验证失败", "/admin/login")}
}

从上面可以看出:在权限管理Rbac微服务中要实现:

//2.查询数据库,判断用户以及密码是否正确
userinfo := []models.Manager{}
password = models.Md5(password)
models.DB.Where("username = ? and password = ? ", username, password).Find(&userinfo)

而实现上面的逻辑,则需要连接数据库,创建数据模型,这里使用GORM方式

(1).创建数据模型

在models下创建manager.go,可参考[golang gin框架] 13.Gin 商城项目-配置公共基类实现公共的成功,失败提示页面用户登录、退出登录、以及权限判断

package models//管理员表type Manager struct { // 结构体首字母大写, 和数据库表名对应, 默认访问数据表users, 可以设置访问数据表的方法Id  intUsername stringPassword stringMobile stringEmail stringStatus intRoleId intAddTime intIsSuper int
}//配置数据库操作的表名称
func (Manager) TableName() string {return "manager"
}

(2).创建conf/app.ini文件

在微服务项目Rbac下创建conf/app.ini配置文件,一些配置,如:数据库配置,redis配置,consul配置信息都可以放到该文件下,以便使用,使用了这个文件下的配置,改变里面的数据不用重启项目,具体参考:[golang gin框架] 9.Gin GORM 中使用事务以及go-ini加载.ini配置文件

app_name   = rbac
# possible values: DEBUG, INFO, WARNING, ERROR, FATAL[mysql]
ip       = 127.0.0.1
port     = 3306
user     = root
password = 123456
database = ginshop[consul]
#指定微服务的ip,一般一个微服务对应一台服务器,当在同一台服务器上有多个微服务时,以不同端口区分
addr   = localhost:8082

(3).定义initInt.go文件

在models下创建initInt.go文件,自动初始化app.ini,很多地方调用app.ini时,就可以从这里介入,该文件目的是减少代码的复用

package models//自动初始化app.ini:这里很多地方都要用到import ("fmt""os""gopkg.in/ini.v1"
)var Config *ini.File
var iniErr errorfunc init() {Config, iniErr = ini.Load("./conf/app.ini")if iniErr != nil {fmt.Printf("Fail to read file: %v", iniErr)os.Exit(1)}
}

(4).创建initMysql.go

在models下创建initMysql.go文件,该文件目的是创建mysql数据库连接引擎

package models//https://gorm.io/zh_CN/docs/connecting_to_the_database.html
//微服务项目中: 推荐一个微服务对应一个数据库
import ("fmt""gorm.io/driver/mysql""gorm.io/gorm"
)var DB *gorm.DB
var err errorfunc init() {//读取.ini里面的数据库配置,相关配置从initIni.go中读取ip := Config.Section("mysql").Key("ip").String()port := Config.Section("mysql").Key("port").String()user := Config.Section("mysql").Key("user").String()password := Config.Section("mysql").Key("password").String()database := Config.Section("mysql").Key("database").String()// dsn := "root:123456@tcp(192.168.0.6:3306)/gin?charset=utf8mb4&parseTime=True&loc=Local"dsn := fmt.Sprintf("%v:%v@tcp(%v:%v)/%v?charset=utf8mb4&parseTime=True&loc=Local", user, password, ip, port, database)DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{QueryFields: true, //打印sql//SkipDefaultTransaction: true, //禁用事务})// DB.Debug()if err != nil {fmt.Println(err)}
}

(5).编写proto/rbac.proto文件

编写proto/rbac.proto文件,实现微服务相关数据逻辑,可参考:[golang gin框架] 39.Gin商城项目-微服务实战之微服务架构

syntax = "proto3";package rbac;option go_package = "./proto/rbac";service Rbac {//登录操作rpc Login(LoginRequest) returns (LoginResponse) {}
}//用户信息model:参考models/manager.go,一一对应
message ManagerModel{int64 id=1;string username=2;string password=3;string mobile=4;string email=5;int64 status=6;int64 roleId=7;int64 addTime=8;int64 isSuper=9;
}//登录请求参数
message LoginRequest{string username=1;string password=2;
}//登录返回参数
message LoginResponse{bool isLogin=1;repeated ManagerModel userlist=2;  //因为在client端需要返回用户相关信息userinfo,故定义一个切片
}

(6).生成proto相关文件

参考[golang 微服务] 7. go-micro框架介绍,go-micro脚手架,go-micro结合consul搭建微服务案例,windows下运行 Makefile里面的proto下的代码 :protoc --proto_path=. --micro_out=. --go_out=:. proto/captcha.proto, 当然,如果是初次使用go-micro,则还需运行init下面的代码(@go xxx),引入相关包

(7).初始化服务端项目

初始化项目,引入项目所需要的包

go mod init rbac
go mod tidy

如果出现:

F:\www\go-data\src\go_code\micro\shop\server\rbac>go mod tidy
go: finding module for package go-micro.dev/v4/client
go: finding module for package gopkg.in/ini.v1
go: finding module for package go-micro.dev/v4
go: finding module for package go-micro.dev/v4/logger
go: finding module for package go-micro.dev/v4/server
go: finding module for package gorm.io/gorm
go: finding module for package gorm.io/driver/mysql
go: finding module for package go-micro.dev/v4/api
go: finding module for package google.golang.org/protobuf/proto
go: finding module for package google.golang.org/protobuf/reflect/protoreflect
go: finding module for package google.golang.org/protobuf/runtime/protoimpl
go: found go-micro.dev/v4 in go-micro.dev/v4 v4.10.2
go: found go-micro.dev/v4/logger in go-micro.dev/v4/logger v1.18.0
go: found gopkg.in/ini.v1 in gopkg.in/ini.v1 v1.67.0
go: found gorm.io/driver/mysql in gorm.io/driver/mysql v1.5.1
go: found gorm.io/gorm in gorm.io/gorm v1.25.2
go: found go-micro.dev/v4/api in go-micro.dev/v4/api v1.18.0
go: found go-micro.dev/v4/client in go-micro.dev/v4/client v1.18.0
go: found go-micro.dev/v4/server in go-micro.dev/v4/server v1.18.0
go: found google.golang.org/protobuf/proto in google.golang.org/protobuf v1.31.0
go: found google.golang.org/protobuf/reflect/protoreflect in google.golang.org/protobuf v1.31.0
go: found google.golang.org/protobuf/runtime/protoimpl in google.golang.org/protobuf v1.31.0
go: rbac/proto/rbac importsgo-micro.dev/v4/api: go-micro.dev/v4/api@v1.18.0: parsing go.mod:module declares its path as: github.com/micro/go-microbut was required as: go-micro.dev/v4/api

说明下载"go-micro.dev/v4"这个包失败了,这时则需运行命令 go get go-micro.dev/v4,如图:

然后再次执行go mod tidy就可以了,这样项目中import包就会引入,不会变红

(8).编写handler/rbac.go用户登录微服务逻辑

因为要实现

//2.查询数据库,判断用户以及密码是否正确
userinfo := []models.Manager{}
password = models.Md5(password)
models.DB.Where("username = ? and password = ? ", username, password).Find(&userinfo)

所以在handler/rbac.go 中需要返回用户相关信息,实现用户登录逻辑代码如下:

package handlerimport ("context""rbac/models"pb "rbac/proto/rbac"
)type Rbac struct{}//后台用户登录的微服务
func (e *Rbac) Login(ctx context.Context, req *pb.LoginRequest, res *pb.LoginResponse) error {managerList := []models.Manager{}err := models.DB.Where("username=? AND password=?", req.Username, req.Password).Find(&managerList).Error//处理数据var templist []*pb.ManagerModelfor i := 0; i < len(managerList); i++ {templist = append(templist, &pb.ManagerModel{Id:       int64(managerList[i].Id),Username: managerList[i].Username,Password: managerList[i].Password,Mobile:   managerList[i].Mobile,Email:    managerList[i].Email,Status:   int64(managerList[i].Status),RoleId:   int64(managerList[i].RoleId),AddTime:  int64(managerList[i].AddTime),IsSuper:  int64(managerList[i].IsSuper),})}if len(managerList) > 0 {res.IsLogin = true} else {res.IsLogin = false}res.Userlist = templistreturn err
}

(9).配置consul服务发现

在main.go配置consul服务发现

package mainimport ("rbac/handler""rbac/models"pb "rbac/proto/rbac""go-micro.dev/v4""go-micro.dev/v4/logger""github.com/go-micro/plugins/v4/registry/consul"
)var (service = "rbac"version = "latest"
)func main() {//集成consulconsulReg := consul.NewRegistry()// Create service//读取.ini里面的配置addr := models.Config.Section("consul").Key("addr").String()srv := micro.NewService(micro.Address(addr),  //指定微服务的ip:  选择注册服务器地址,也可以不配置,默认为本机,也可以选择consul集群中的clientmicro.Name(service),micro.Version(version),//注册consulmicro.Registry(consulReg),)srv.Init(micro.Name(service),micro.Version(version),)// Register handlerif err := pb.RegisterRbacHandler(srv.Server(), new(handler.Rbac)); err != nil {logger.Fatal(err)}// Run serviceif err := srv.Run(); err != nil {logger.Fatal(err)}
}

3.启动consul服务发现

在cmd中运行命令: consul agent -dev,具体参考: [golang 微服务] 5. 微服务服务发现介绍,安装以及consul的使用,Consul集群

4.注册验证码微服务服务端到服务发现(consul)

rbac目录下运行go run main.go,然后在consul UI 查看,是否注册成功

注册成功了

三.创建权限管理Rbac微服务客户端

1.首先把server/rbac/proto文件夹复制到项目中

2.配置consul服务发现

在models下创建initRbacConsul.go,配置consul服务发现,以便调用,代码如下:

package models//Rbac consul(服务发现)初始化配置
//微服务客户端配置: 初始化consul配置,当一个项目中多个微服务时,就很方便了
//建议:一个微服务对应一个客户端,这样好管理import ("github.com/go-micro/plugins/v4/registry/consul""go-micro.dev/v4""go-micro.dev/v4/client""go-micro.dev/v4/registry"
)//RbacClient: 全局变量 在外部的包中可以调用
var RbacClient client.Client//init 方法: 当程序运行时就会自动执行
func init() {consulRegistry := consul.NewRegistry(//指定微服务的ip:  选择注册服务器地址,默认为本机,也可以选择consul集群中的client,建议一个微服务对应一个consul集群的clientregistry.Addrs("127.0.0.1:8500"),)// Create servicesrv := micro.NewService(micro.Registry(consulRegistry),)srv.Init()RbacClient = srv.Client()
}

在这里,也许没有引入,执行命令:go mod tidy引入对应的包

3.调用Rbac微服务

在controllers/login.go的DoIndex方法中调用Rbac微服务

原DoIndex方法代码如下:


//执行登录操作
func (con LoginController) DoIndex(c *gin.Context) {//获取表单中的数据captchaId := c.PostForm("captchaId")     // 验证码idverifyValue := c.PostForm("verifyValue") //验证码的值//获取用户名以及密码username := c.PostForm("username")password := c.PostForm("password")// 1.判断验证码是否验证成功if flag := models.VerifyCaptcha(captchaId, verifyValue); flag {//2.查询数据库,判断用户以及密码是否正确userinfo := []models.Manager{}password = models.Md5(password)models.DB.Where("username = ? and password = ? ", username, password).Find(&userinfo)if len(userinfo) > 0 {//3.执行登录,保存用户信息,执行跳转操作session := sessions.Default(c)//注意: session.Set没法保存结构体对应的切片,所以需要把结构体转换成json字符串userinfoSlice, _ := json.Marshal(userinfo)session.Set("userinfo_admin", string(userinfoSlice))session.Save()con.Success(c, "登录成功", "/admin")} else {con.Error(c, "用户名或密码错误", "/admin/login")}} else {con.Error(c, "验证码验证失败", "/admin/login")}
}

然后在//2.查询数据库,判断用户以及密码是否正确这一步调用Rbac微服务,代码修改如下:

//这里需要import (
//     "context"
//	 pbRbac "goshop/proto/rbac"
//    )//执行登录操作
func (con LoginController) DoIndex(c *gin.Context) {//获取表单中的数据captchaId := c.PostForm("captchaId")     // 验证码idverifyValue := c.PostForm("verifyValue") //验证码的值//获取用户名以及密码username := c.PostForm("username")password := c.PostForm("password")// 1.判断验证码是否验证成功if flag := models.VerifyCaptcha(captchaId, verifyValue); flag {//2.查询数据库,判断用户以及密码是否正确//userinfo := []models.Manager{}//password = models.Md5(password)//models.DB.Where("username = ? and password = ? ", username, password).Find(&userinfo)//调用Rbac微服务rbacClient := pbRbac.NewRbacService("rbac", models.RbacClient)res, _ := rbacClient.Login(context.Background(), &pbRbac.LoginRequest{Username: username,Password: models.Md5(password),})//通过Rbac微服务结果,判断是否登录if res.IsLogin {//3.执行登录,保存用户信息,执行跳转操作session := sessions.Default(c)//注意: session.Set没法保存结构体对应的切片,所以需要把结构体转换成json字符串userinfoSlice, _ := json.Marshal(res.Userlist)  //res.Userlist:微服务返回的用户信息session.Set("userinfo_admin", string(userinfoSlice))session.Save()con.Success(c, "登录成功", "/admin")} else {con.Error(c, "用户名或密码错误", "/admin/login")}} else {con.Error(c, "验证码验证失败", "/admin/login")}
}

好了,权限管理Rbac微服务服客户端代码就ok了,接下来验证Rbac微服务是否成功