forked from hwholiday/learning_tools
/
service.go
42 lines (37 loc) · 1.06 KB
/
service.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
package src
import (
"context"
"errors"
"github.com/go-kit/kit/metrics"
"github.com/hwholiday/learning_tools/go-kit/v7/user_agent/pb"
"github.com/hwholiday/learning_tools/go-kit/v7/utils"
"go.uber.org/zap"
"math/rand"
"time"
)
type Service interface {
Login(ctx context.Context, in *pb.Login) (ack *pb.LoginAck, err error)
}
type baseServer struct {
logger *zap.Logger
}
func NewService(log *zap.Logger, counter metrics.Counter, histogram metrics.Histogram) Service {
var server Service
server = &baseServer{log}
server = NewLogMiddlewareServer(log)(server)
server = NewMetricsMiddlewareServer(counter, histogram)(server)
return server
}
func (s baseServer) Login(ctx context.Context, in *pb.Login) (ack *pb.LoginAck, err error) {
if in.Account != "hwholiday" || in.Password != "123456" {
err = errors.New("用户信息错误")
return
}
//模拟耗时
rand.Seed(time.Now().UnixNano())
sl := rand.Int31n(10-1) + 1
time.Sleep(time.Duration(sl) * time.Millisecond * 100)
ack = &pb.LoginAck{}
ack.Token, err = utils.CreateJwtToken(in.Account, 1)
return
}