Fabric区块链浏览器（2）

本文是区块链浏览器系列的第四篇。
在上一篇文章介绍如何解析区块数据时，使用session对客户端上传的pb文件进行区分，到期后自动删除。
在这片文章中，会着重介绍下认证系统的实现，主要分为三部分：

• 添加数据库，存储用户信息
  
• 实现用户认证中间件
  
• 修改路由
  

1. 用户信息存储

我这里使用MySQL来存储数据，使用gorm来实现与数据库的交换。
首先需要创建用户表：

1. CREATE TABLE `users` (
2.   `id` bigint unsigned NOT NULL AUTO_INCREMENT,
3.   `name` varchar(100) NOT NULL,
4.   `password` varchar(100) DEFAULT NULL,
5.   `salt` longtext,
6.   `created_at` datetime(3) DEFAULT NULL,
7.   `updated_at` datetime(3) DEFAULT NULL,
8.   `deleted_at` datetime(3) DEFAULT NULL,
9.   PRIMARY KEY (`id`)
10. ) ENGINE=InnoDB AUTO_INCREMENT=9 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci

复制代码

创建MySQL链接句柄：

1. func InitDB(source string) (*gorm.DB, error) {
2.         dblog := logger.New(
3.                 log.New(os.Stdout, "\r\n", log.LstdFlags),
4.                 logger.Config{
5.                         LogLevel:                  logger.Error,
6.                         IgnoreRecordNotFoundError: true,
7.                         Colorful:                  true,
8.                         SlowThreshold:             time.Second,
9.                 },
10.         )
11.         return gorm.Open(mysql.Open(source), &gorm.Config{
12.                 SkipDefaultTransaction:                   true,
13.                 AllowGlobalUpdate:                        false,
14.                 DisableForeignKeyConstraintWhenMigrating: true,
15.                 Logger:                                   dblog,
16.         })
17. }

复制代码

表结构比较简单，实现两个查询接口：

1. func GetUserByName(name string) (*User, error) {
2.         var user User
3.         db.Get().First(&user, "name = ?", name)
4.         if user.ID == 0 {
5.                 return nil, fmt.Errorf("user with name: %s is not found", name)
6.         }
7.         return &user, nil
8. }
10. func GetUserByID(id uint) (*User, error) {
11.         var user User
12.         db.Get().First(&user, "id = ?", id)
13.         if user.ID == 0 {
14.                 return nil, fmt.Errorf("user with id: %d is not found", id)
15.         }
16.         return &user, nil
17. }

复制代码

除了查询接口外，还需要提供用户注册，这里直接使用Save()接口进行数据库写入操作：

1. func RegisterUser(name, password string) (*LoginResponse, error) {
2.         salt := genSalt()
3.         u := &User{
4.                 Name:     name,
5.                 Password: utils.CalcPassword(password, salt),
6.                 Salt:     salt,
7.         }
8.         if err := db.Get().Save(u).Error; err != nil {
9.                 return nil, errors.Wrap(err, "RegisterUser error")
10.         }
12.         now := time.Now()
13.         claims := &jwtv5.RegisteredClaims{
14.                 ExpiresAt: jwtv5.NewNumericDate(now.Add(30 * time.Minute)),
15.                 Issuer:    "browser",
16.                 Subject:   fmt.Sprintf("%d", u.ID),
17.         }
18.         token := jwtv5.NewWithClaims(jwtv5.SigningMethodHS256, claims)
19.         tokenString, err := token.SignedString(securityKey)
20.         if err != nil {
21.                 return nil, errors.Wrap(err, "create token error")
22.         }
24.         return &LoginResponse{
25.                 Token:    tokenString,
26.                 Expire:   now.Add(30 * time.Minute).Unix(),
27.                 ID:       u.ID,
28.                 Username: u.Name,
29.         }, nil
30. }

复制代码

用户认证采用的JWT（JSON Web Token），实现方法在JWT介绍有介绍，所以还需要提供两个接口：Login实现token获取，RefreshToken刷新token：

1. func Login(name, password string) (*LoginResponse, error) {
2.         user, err := GetUserByName(name)
3.         if err != nil {
4.                 return nil, errors.Wrap(err, "GetUserByName error")
5.         }
7.         if utils.CalcPassword(password, user.Salt) != user.Password {
8.                 return nil, errors.New("user name or password is incorrect")
9.         }
11.         now := time.Now()
12.         claims := &jwtv5.RegisteredClaims{
13.                 ExpiresAt: jwtv5.NewNumericDate(now.Add(30 * time.Minute)),
14.                 Issuer:    "browser",
15.                 Subject:   fmt.Sprintf("%d", user.ID),
16.         }
17.         token := jwtv5.NewWithClaims(jwtv5.SigningMethodHS256, claims)
18.         tokenString, err := token.SignedString(securityKey)
19.         if err != nil {
20.                 return nil, errors.Wrap(err, "create token error")
21.         }
23.         return &LoginResponse{
24.                 Token:    tokenString,
25.                 Expire:   now.Add(30 * time.Minute).Unix(),
26.                 ID:       user.ID,
27.                 Username: user.Name,
28.         }, nil
29. }
31. func RefreshToken(id uint) (*LoginResponse, error) {
32.         user, err := GetUserByID(id)
33.         if err != nil {
34.                 return nil, errors.Wrap(err, "GetUserByName error")
35.         }
37.         now := time.Now()
38.         claims := &jwtv5.RegisteredClaims{
39.                 ExpiresAt: jwtv5.NewNumericDate(now.Add(30 * time.Minute)),
40.                 Issuer:    "browser",
41.                 Subject:   fmt.Sprintf("%d", user.ID),
42.         }
43.         token := jwtv5.NewWithClaims(jwtv5.SigningMethodHS256, claims)
44.         tokenString, err := token.SignedString(securityKey)
45.         if err != nil {
46.                 return nil, errors.Wrap(err, "create token error")
47.         }
48.         return &LoginResponse{
49.                 Token:    tokenString,
50.                 Expire:   now.Add(30 * time.Minute).Unix(),
51.                 ID:       user.ID,
52.                 Username: user.Name,
53.         }, nil
54. }

复制代码

2. 用户认证中间件

关于Gin中间件的开发，可以参照gin中间件开发，这里增加三种认证方式：noAuth，不使用认证；basicAuth，用户名密码方式认证；tokenAuth，使用token进行认证：

1. func noAuth(ctx *gin.Context) {
2.         ctx.Next()
3. }
5. func basicAuth(ctx *gin.Context) {
6.         name, pwd, ok := ctx.Request.BasicAuth()
7.         if !ok {
8.                 srvLogger.Error("basic auth failed")
9.                 ctx.JSON(http.StatusForbidden, gin.H{"code": http.StatusForbidden, "msg": "basic auth failed"})
10.                 ctx.Abort()
11.                 return
12.         }
13.         user, err := data.GetUserByName(name)
14.         if err != nil {
15.                 srvLogger.Errorf("GetUserByName error: %s", err.Error())
16.                 ctx.JSON(http.StatusForbidden, gin.H{"code": http.StatusForbidden, "msg": err.Error()})
17.                 ctx.Abort()
18.                 return
19.         }
20.         if utils.CalcPassword(pwd, user.Salt) != user.Password {
21.                 srvLogger.Error("user name or password is incorrect")
22.                 ctx.JSON(http.StatusForbidden, gin.H{"code": http.StatusForbidden, "msg": "user name or password is incorrect"})
23.                 ctx.Abort()
24.                 return
25.         }
26.         ctx.Next()
27. }
29. func tokenAuth(ctx *gin.Context) {
30.         if err := data.ParseJWT(strings.Split(ctx.Request.Header.Get("Authorization"), " ")[1]); err != nil {
31.                 srvLogger.Errorf("tokenAuth error: %s", err.Error())
32.                 ctx.JSON(http.StatusForbidden, gin.H{"code": http.StatusForbidden, "msg": "token auth failed"})
33.                 ctx.Abort()
34.                 return
35.         }
36.         ctx.Next()
37. }

复制代码

3. 注册路由

在上篇中，注册的路由是这样的：

1. engine.POST("/login", login)
2. engine.GET("/hi/:name", sayHi)
3. engine.POST("/block/upload", upload)
4. engine.GET("/block/parse/:msgType", parse)
5. engine.POST("/block/update/:channel", updateConfig)

复制代码

现在需要对/block/upload、/block/parse/:msgType、/block/update/:channel接口增加认证，这就需要用到我们上面实现的三个中间件。
由于中间件会按照它们的注册顺利来执行，所以需要认证中间件需要在相应的处理接口前执行，针对noAuth的情况，上面的代码并不需要进行修改，但对于basicAuth、tokenAuth，上面的代码就需要修改了：

1. engine.POST("/block/upload", basicAuth, upload)
2. engine.GET("/block/parse/:msgType", basicAuth, parse)
3. engine.POST("/block/update/:channel", basicAuth, updateConfig)

复制代码

或

1. engine.POST("/block/upload", tokenAuth, upload)
2. engine.GET("/block/parse/:msgType", tokenAuth, parse)
3. engine.POST("/block/update/:channel", tokenAuth, updateConfig)

复制代码

当然我们也可以使用Handle(httpMethod, relativePath string, handlers ...HandlerFunc)来进行路由注册：

1. for _, router := range server.Routers() {
2.         var handlers []gin.HandlerFunc
3.         if router.AuthType == 0 {
4.                 router.AuthType = conf.AuthType
5.         }
6.         switch router.AuthType {
7.         case config.Server_BASICAUTH:
8.                 handlers = append(handlers, basicAuth)
9.         case config.Server_TOKENAUTH:
10.                 handlers = append(handlers, tokenAuth)
11.         default:
12.                 handlers = append(handlers, noAuth)
13.         }
14.         handlers = append(handlers, router.Handler)
15.         engine.Handle(router.Method, router.Path, handlers...)
16. }

复制代码

项目完整代码可以从Github上查看。
  

声明：本作品采用署名-非商业性使用-相同方式共享 4.0 国际 (CC BY-NC-SA 4.0)进行许可，使用时请注明出处。
Author: mengbin
blog: mengbin
Github: mengbin92
cnblogs: 恋水无意

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！