-
Notifications
You must be signed in to change notification settings - Fork 1
/
server.go
110 lines (100 loc) · 2.47 KB
/
server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package main
import (
"context"
"log"
"net"
"os"
_ "github.com/CodyGuo/godaemon"
"github.com/chaossat/tiktak/middleware"
"github.com/chaossat/tiktak/service/register/model"
"github.com/chaossat/tiktak/service/register/pb"
"github.com/chaossat/tiktak/util"
"github.com/spf13/viper"
"google.golang.org/grpc"
)
type Register struct{}
func (r *Register) Register(ctx context.Context, req *pb.DouyinUserRegisterRequest) (*pb.DouyinUserRegisterResponse, error) {
username := req.GetUsername()
password := req.GetPassword()
password_hash := util.MD5V(password)
user, err := model.SaveUser(username, password_hash)
var statuscode int32
var statusmsg string
var userid int64
var token string
if err != nil {
if err.Error() == "用户名已存在" {
log.Println("发现用户名已存在错误...")
statuscode = int32(1)
statusmsg = "用户名已存在"
userid = int64(0)
token = ""
resp := pb.DouyinUserRegisterResponse{
StatusCode: &statuscode,
StatusMsg: &statusmsg,
UserId: &userid,
Token: &token,
}
log.Println("用户已存在 resp:", resp)
return &resp, nil
}
statusmsg = "数据库存储失败"
return &pb.DouyinUserRegisterResponse{
StatusCode: &statuscode,
StatusMsg: &statusmsg,
UserId: &userid,
Token: &token,
}, nil
}
token, err = middleware.CreateToken(user.ID)
if err != nil {
log.Println("生成token失败", err.Error())
statuscode = 1
statusmsg = "token生成失败"
userid = 0
token = ""
return &pb.DouyinUserRegisterResponse{
StatusCode: &statuscode,
StatusMsg: &statusmsg,
UserId: &userid,
Token: &token,
}, nil
}
statuscode = 0
statusmsg = "注册成功"
userid = user.ID
resp := pb.DouyinUserRegisterResponse{
StatusCode: &statuscode,
StatusMsg: &statusmsg,
UserId: &userid,
Token: &token,
}
return &resp, nil
}
func InitConfig() {
workDir, _ := os.Getwd()
viper.SetConfigName("config")
viper.SetConfigType("yml")
viper.AddConfigPath(workDir + "/config")
err := viper.ReadInConfig()
if err != nil {
panic(err)
}
}
func main() {
log.Println("正在启动Register服务......")
InitConfig()
model.InitDB()
//初始化grpc实例
grpcServer := grpc.NewServer()
//注册服务
pb.RegisterRegisterServer(grpcServer, new(Register))
//设置监听
listen, err := net.Listen("tcp", ":12345")
if err != nil {
log.Println("注册服务启动监听失败")
}
defer listen.Close()
//启动服务
grpcServer.Serve(listen)
}