其他分享
首页 > 其他分享> > go搭建简单grpc服务(登录服务)2-增加proto字段,md5验签

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

作者:互联网

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

 

当前文件目录结构如下:

 

 

修改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
	}

 

标签:nk,string,proto,grpc,pb,go,pMap,验签,md5
来源: https://www.cnblogs.com/kala00k/p/16582864.html