Fabric区块链浏览器(3)

打印 上一主题 下一主题

主题 850|帖子 850|积分 2550

本文是区块链浏览器系列的第五篇,项目完整代码在这里
上一篇文章中给浏览器增加了简单的用户认证,至此浏览器的基本功能就已经大致完成了。
在这片文章中,我将使用kratos对区块链浏览器器进行重构,使之同时支持http和gRPC。
1. 项目结构

项目结构如下:
  1. $ tree
  2. .
  3. ├── api                                    
  4. │   └── browser
  5. │       └── v1
  6. │           ├── block_grpc.pb.go
  7. │           ├── block_http.pb.go
  8. │           ├── block.pb.go
  9. │           ├── block.proto
  10. │           ├── browser_grpc.pb.go
  11. │           ├── browser_http.pb.go
  12. │           ├── browser.pb.go
  13. │           └── browser.proto
  14. ├── cmd
  15. │   └── example
  16. │       ├── main.go
  17. │       ├── wire_gen.go
  18. │       └── wire.go
  19. ├── configs
  20. │   └── config.yaml
  21. ├── Dockerfile
  22. ├── go.mod
  23. ├── go.sum
  24. ├── internal
  25. │   ├── biz
  26. │   │   ├── account.go
  27. │   │   ├── biz.go
  28. │   │   ├── README.md
  29. │   │   └── repo.go
  30. │   ├── conf
  31. │   │   ├── conf.pb.go
  32. │   │   └── conf.proto
  33. │   ├── data
  34. │   │   ├── account.go
  35. │   │   ├── data.go
  36. │   │   └── README.md
  37. │   ├── db
  38. │   │   ├── db.go
  39. │   │   ├── mysql
  40. │   │   │   └── mysql.go
  41. │   │   ├── pg
  42. │   │   │   └── pg.go
  43. │   │   └── sqlite3
  44. │   │       └── sqlite3.go
  45. │   ├── server
  46. │   │   ├── grpc.go
  47. │   │   ├── http.go
  48. │   │   └── server.go
  49. │   ├── service
  50. │   │   ├── block.go
  51. │   │   ├── browser.go
  52. │   │   ├── global.go
  53. │   │   ├── README.md
  54. │   │   ├── service.go
  55. │   │   └── utils.go
  56. │   └── utils
  57. │       ├── putils.go
  58. │       └── utils.go
  59. ├── LICENSE
  60. ├── Makefile
  61. ├── openapi.yaml
  62. ├── README.md
  63. └── third_party
  64.     ├── errors
  65.     │   └── errors.proto
  66.     ├── google
  67.     │   ├── api
  68.     │   │   ├── annotations.proto
  69.     │   │   ├── client.proto
  70.     │   │   ├── field_behavior.proto
  71.     │   │   ├── httpbody.proto
  72.     │   │   └── http.proto
  73.     │   └── protobuf
  74.     │       ├── any.proto
  75.     │       ├── api.proto
  76.     │       ├── compiler
  77.     │       │   └── plugin.proto
  78.     │       ├── descriptor.proto
  79.     │       ├── duration.proto
  80.     │       ├── empty.proto
  81.     │       ├── field_mask.proto
  82.     │       ├── source_context.proto
  83.     │       ├── struct.proto
  84.     │       ├── timestamp.proto
  85.     │       ├── type.proto
  86.     │       └── wrappers.proto
  87.     ├── openapi
  88.     │   └── v3
  89.     │       ├── annotations.proto
  90.     │       └── openapi.proto
  91.     ├── README.md
  92.     └── validate
  93.         ├── README.md
  94.         └── validate.proto
  95. 26 directories, 66 files
复制代码

  • api 定义接口

    • block.prot:区块上传和区块解析的接口
    • browser:用户认证相关接口,包括用户注册、用户登录以及刷新token

  • cmd:本地调试使用,基于wire实现
  • configs:存放项目的配置文件
  • internal:业务逻辑实现

    • biz:类似DTO,封装数据库操作
    • conf:定义项目配置项
    • data:数据处理,目前主要是处理用户认证相关数据
    • db:数据库操作实现
    • server:注册http和gRPC服务
    • service:实现浏览器业务逻辑
    • utils:工具类

  • third_party:依赖的第三方库,主要是protobuf相关的第三方库
2. api接口定义

block.proto中定义区块上传和解析操作:
  1. syntax = "proto3";
  2. package api.browser.v1;
  3. option go_package = "mengbin92/browser/api/browser/v1;v1";
  4. import "google/api/annotations.proto";
  5. service Block {
  6.   rpc UpChaincode(UploadRequest) returns (UploadResponse) {
  7.     option (google.api.http) = {
  8.       post : "/v1/block/up"
  9.       body : "*"
  10.     };
  11.   };
  12.   rpc ParseBlock(ParseRequest) returns (ParseResponse) {
  13.     option (google.api.http) = {
  14.       get : "/v1/block/parse"
  15.     };
  16.   }
  17. }
  18. message UploadRequest {
  19.   string name = 1;
  20.   bytes content = 2;
  21. }
  22. message UploadResponse {
  23.   bool result = 1;
  24.   string name = 2;
  25. }
  26. message ParseRequest {
  27.   enum BlockOperation {
  28.     BLOCK = 0;
  29.     HEADER = 1;
  30.     DATA = 2;
  31.     CONFIG = 3;
  32.     CHAINCODE = 4;
  33.     ACTIONS = 5;
  34.     TRANSACTION = 6;
  35.     INPUT = 7;
  36.     RWSET = 8;
  37.     CHANNEL = 9;
  38.     ENDORSEMENTS = 10;
  39.     CREATOR = 11;
  40.     METADATA = 12;
  41.   }
  42.   string name = 1;
  43.   BlockOperation operation = 2;
  44. }
  45. message ParseResponse { bytes data = 1; }
  46. message Endorser {
  47.   string MSP = 1;
  48.   string Name = 2;
  49. }
  50. message Endorsers { repeated Endorser endorsers = 1; }
复制代码
browser.proto定义用户注册、登录和token刷新操作:
  1. syntax = "proto3";
  2. package api.browser.v1;
  3. option go_package = "mengbin92/browser/api/browser/v1;v1";
  4. import "google/protobuf/timestamp.proto";
  5. import "google/api/annotations.proto";
  6. service Browser {
  7.   rpc GetToken(Login) returns (LoginResponse) {
  8.     option (google.api.http) = {
  9.       post : "/v1/user/login"
  10.       body : "*"
  11.     };
  12.   };
  13.   rpc Regisger(Login) returns (LoginResponse) {
  14.     option (google.api.http) = {
  15.       post : "/v1/user/register"
  16.       body : "*"
  17.     };
  18.   };
  19.   rpc RefreshToken(RefreshRequest) returns (LoginResponse) {
  20.     option (google.api.http) = {
  21.       get : "/v1/user/refresh/{id}"
  22.     };
  23.   };
  24. }
  25. message User {
  26.   uint64 id = 1;
  27.   string name = 2;
  28.   string password = 3;
  29.   string salt = 4;
  30.   google.protobuf.Timestamp create_at = 5;
  31.   google.protobuf.Timestamp update_at = 6;
  32.   google.protobuf.Timestamp delete_at = 7;
  33. }
  34. message LoginResponse {
  35.   string token = 1;
  36.   int64 expire = 2;
  37.   uint64 id = 3;
  38.   string username = 4;
  39. }
  40. message Login {
  41.   string username = 1;
  42.   string password = 2;
  43. }
  44. message RefreshRequest { uint64 id = 1; }
复制代码
在项目根目录下执行下面命令即可完成框架生成:
  1. # 生成对应的go文件
  2. $ make api
  3. # 生成service文件
  4. $ kratos proto server api/browser/v1/block.proto -t internal/service
  5. $ kratos proto server api/browser/v1/browser.proto -t internal/service
复制代码
3. biz介绍

biz定义了用户认证相关的接口,供service调用,接口内容如下:
  1. type AccountRepo interface {
  2.     // 用户注册
  3.         Register(context.Context, string, string) (*v1.LoginResponse, error)
  4.     // 用户登录,用于获取token
  5.         Login(context.Context, string, string) (*v1.LoginResponse, error)
  6.     // 刷新token
  7.         RefreshToken(context.Context, uint64) (*v1.LoginResponse, error)
  8. }
复制代码
4. conf介绍

conf定义项目的配置项,也是通过protobuf定义的:
  1. syntax = "proto3";
  2. package browser.api;
  3. option go_package = "mengbin92/browser/internal/conf;conf";
  4. import "google/protobuf/duration.proto";
  5. message Bootstrap {
  6.   Server server = 1;
  7.   Log log = 2;
  8.   Data data = 3;
  9. }
  10. message Server {
  11.   message HTTP {
  12.     string network = 1;
  13.     string addr = 2;
  14.     google.protobuf.Duration timeout = 3;
  15.   }
  16.   message GRPC {
  17.     string network = 1;
  18.     string addr = 2;
  19.     google.protobuf.Duration timeout = 3;
  20.   }
  21.   enum AuthType{
  22.     TOKENAUTH = 0;
  23.     BASICAUTH = 1;
  24.     NOAUTH = 2;
  25.   }
  26.   HTTP http = 1;
  27.   GRPC grpc = 2;
  28.   TLS tls = 3;
  29.   // 2: no auth; 1: basic auth; default: 0, token auth
  30.   AuthType auth_type = 4;
  31.   Auth auth = 5;
  32. }
  33. message TLS {
  34.   // 是否启用tls
  35.   bool enbale = 1;
  36.   // 证书路径
  37.   string cert = 2;
  38.   // 对应私钥路径
  39.   string key = 3;
  40. }
  41. message Log {
  42.   // 日志级别设置
  43.   // 支持debug(-1)、info(0)、warn(1)、error(2)、dpanic(3)、panic(4)、fatal(5)
  44.   int32 level = 1;
  45.   // 日志输出格式,支持json or console
  46.   string format = 2;
  47. }
  48. message Database {
  49.   // 数据库类型:mysql、sqlite3、postgres
  50.   string driver = 1;
  51.   string source = 2;
  52.   // 最大空闲连接数
  53.   int64 max_idle_conn = 3;
  54.   // 最大连接数
  55.   int64 max_open_conn = 4;
  56.   // 连接空闲超时参数
  57.   google.protobuf.Duration conn_max_lifetime = 5;
  58. }
  59. message Redis {
  60.   string network = 1;
  61.   string addr = 2;
  62.   string password = 3;
  63.   int32 db = 4;
  64.   google.protobuf.Duration dial_timeout = 5;
  65.   google.protobuf.Duration read_timeout = 6;
  66.   google.protobuf.Duration write_timeout = 7;
  67. }
  68. message Data {
  69.   Database database = 1;
  70.   Redis redis = 2;
  71. }
  72. message Auth {
  73.   // jwt 秘钥,暂未启用
  74.   string jwt_secret = 1;
  75.   // token过期时间
  76.   google.protobuf.Duration expire = 2;
  77. }
复制代码
5. data介绍

data中定义了AccountRepo结构,主要实现了biz中与数据库交互的业务逻辑:
  1. type AccountRepo struct {
  2.         data      *Data
  3.         log       *log.Helper
  4.         jwtSecret string
  5.         expire    time.Duration
  6. }
  7. func NewAccountRepo(data *Data, logger log.Logger, auth *conf.Auth) biz.AccountRepo {
  8.         return &AccountRepo{
  9.                 data:      data,
  10.                 log:       log.NewHelper(logger),
  11.                 jwtSecret: auth.JwtSecret,
  12.                 expire:    auth.Expire.AsDuration(),
  13.         }
  14. }
  15. func genSalt() string {
  16.         uid, _ := uuid.NewRandom()
  17.         return uid.String()
  18. }
  19. func (ar *AccountRepo) Register(ctx context.Context, username, password string) (*v1.LoginResponse, error) {
  20.         salt := genSalt()
  21.         u := &User{
  22.                 Name:     username,
  23.                 Password: utils.CalcPassword(password, salt),
  24.                 Salt:     salt,
  25.         }
  26.         if err := ar.data.db.Save(u).Error; err != nil {
  27.                 ar.log.Errorf("save user data error: %s", err.Error())
  28.                 return nil, errors.Wrap(err, "save user error")
  29.         }
  30.         now := time.Now()
  31.         tokenString, err := ar.genToken(uint64(u.ID), now)
  32.         if err != nil {
  33.                 ar.log.Errorf("create token error: %s", err.Error())
  34.                 return nil, errors.Wrap(err, "create token error")
  35.         }
  36.         return &v1.LoginResponse{
  37.                 Token:    tokenString,
  38.                 Expire:   now.Add(ar.expire).Unix(),
  39.                 Id:       uint64(u.ID),
  40.                 Username: u.Name,
  41.         }, nil
  42. }
  43. func (ar *AccountRepo) Login(ctx context.Context, username, password string) (*v1.LoginResponse, error) {
  44.         user, err := ar.getUserByName(ctx, username)
  45.         if err != nil {
  46.                 ar.log.Errorf("get user from data error: %s", err.Error())
  47.                 return nil, errors.Wrap(err, "GetUserByName error")
  48.         }
  49.         if utils.CalcPassword(password, user.Salt) != user.Password {
  50.                 ar.log.Error("user name or password is incorrect")
  51.                 return nil, errors.New("user name or password is incorrect")
  52.         }
  53.         now := time.Now()
  54.         tokenString, err := ar.genToken(user.Id, now)
  55.         if err != nil {
  56.                 ar.log.Errorf("create token error: %s", err.Error())
  57.                 return nil, errors.Wrap(err, "create token error")
  58.         }
  59.         return &v1.LoginResponse{
  60.                 Token:    tokenString,
  61.                 Expire:   now.Add(ar.expire).Unix(),
  62.                 Id:       user.Id,
  63.                 Username: user.Name,
  64.         }, nil
  65. }
  66. func (ar *AccountRepo) RefreshToken(ctx context.Context, id uint64) (*v1.LoginResponse, error) {
  67.         user, err := ar.getUserById(ctx, id)
  68.         if err != nil {
  69.                 ar.log.Errorf("get user from data error: %s", err.Error())
  70.                 return nil, errors.Wrap(err, "GetUserByName error")
  71.         }
  72.         now := time.Now()
  73.         tokenString, err := ar.genToken(user.Id, now)
  74.         if err != nil {
  75.                 ar.log.Errorf("create token error: %s", err.Error())
  76.                 return nil, errors.Wrap(err, "create token error")
  77.         }
  78.         return &v1.LoginResponse{
  79.                 Token:    tokenString,
  80.                 Expire:   now.Add(ar.expire).Unix(),
  81.                 Id:       user.Id,
  82.                 Username: user.Name,
  83.         }, nil
  84. }
复制代码
6. service介绍

service主要实现api中定义的接口:

  • block.go:
    1. type BlockService struct {
    2.         pb.UnimplementedBlockServer
    3. }
    4. func NewBlockService() *BlockService {
    5.         return &BlockService{}
    6. }
    7. func (s *BlockService) UpChaincode(ctx context.Context, req *pb.UploadRequest) (*pb.UploadResponse, error) {
    8.     // 与gin实现版本类似,客户端上传区块数据存储到服务端
    9.         return &pb.UploadResponse{
    10.                 Result: true,
    11.                 Name:   req.Name,
    12.         }, nil
    13. }
    14. func (s *BlockService) ParseBlock(ctx context.Context, req *pb.ParseRequest) (*pb.ParseResponse, error) {
    15.     // 与gin实现版本逻辑类似,这里就不再赘述了
    16.         return &pb.ParseResponse{}, nil
    17. }
    复制代码
  • browser.go:
    1. type BrowserService struct {
    2.         account *biz.AccountUsecase
    3.         pb.UnimplementedBrowserServer
    4. }
    5. func NewBrowserService(repo *biz.AccountUsecase, logger klog.Logger) *BrowserService {
    6.         log = klog.NewHelper(logger)
    7.         return &BrowserService{
    8.                 account: repo,
    9.         }
    10. }
    11. func (s *BrowserService) GetToken(ctx context.Context, req *pb.Login) (*pb.LoginResponse, error) {
    12.         return s.account.Login(ctx, req.Username, req.Password)
    13. }
    14. func (s *BrowserService) Regisger(ctx context.Context, req *pb.Login) (*pb.LoginResponse, error) {
    15.         return s.account.Register(ctx, req.Username, req.Password)
    16. }
    17. func (s *BrowserService) RefreshToken(ctx context.Context, req *pb.RefreshRequest) (*pb.LoginResponse, error) {
    18.         return s.account.RefreshToken(ctx, req.Id)
    19. }
    复制代码
  
声明:本作品采用署名-非商业性使用-相同方式共享 4.0 国际 (CC BY-NC-SA 4.0)进行许可,使用时请注明出处。
Author: mengbin
blog: mengbin
Github: mengbin92
cnblogs: 恋水无意

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

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

正序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

渣渣兔

金牌会员
这个人很懒什么都没写!
快速回复 返回顶部 返回列表