go搭建简单grpc服务(登录服务）2-增加proto字段，md5验签

上一篇随笔搭建一个简单的登录grpc服务，但还有一些需要完善，这篇随笔主要介绍

• proto文件增加字段流程
• md5签名

 

当前文件目录结构如下：

 

 

修改pb/users/login.proto，直接在pwd参数下增加一个字段stime，

 

 

 

 

 

使用如下命令重新生成pb/users/login.pb.go文件

protoc --proto_path=./pb/users/ ./pb/users/*.proto --go_out=./pb/users/

 重新生成的文件，多了增加的参数

 

 

 

server/pkg/common/common.go 定义md5加密方法 (key 先写个简单的123）

//md5 加解密参数
func CheckSign(params map[string]string, key string, filter string) (sign string) {

	nk := make([]string, 0)
	for k, _ := range params {
		if k == filter {
			continue
		}
		nk = append(nk, k)
	}

	sort.Strings(nk)

	var originStr string
	for _, v := range nk {
		originStr += fmt.Sprintf("%v=%v&", v, params[v])
	}

	originStr += fmt.Sprintf("key=%v", key)
	h := md5.New()
	h.Write([]byte(originStr))
	sign = hex.EncodeToString(h.Sum(nil)[:])
	return
}

 

server/v1/service.go Login 方法中增加参数校验逻辑

	var pMap = make(map[string]string)
	pMap["name"] = req.Name
	pMap["pwd"] = req.Pwd
	pMap["stime"] = req.Stime

	calSign := common.CheckSign(pMap, "123", "1")

	if req.Sign != calSign {
		return &login.Response{
			Code:  "0",
			Msg:   "login fail,sign error",
			Data:  "",
			Token: "",
		}, nil
	}