/
get_user_by_email.go
79 lines (73 loc) 路 1.83 KB
/
get_user_by_email.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
package utils
import (
"context"
"github.com/buger/jsonparser"
"github.com/dgraph-io/dgo/v230"
"github.com/dgraph-io/dgo/v230/protos/api"
"github.com/hongbo-miao/hongbomiao.com/api-go/internal/graphql_server/types"
"github.com/rs/zerolog/log"
"go.elastic.co/apm/module/apmgrpc/v2"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
)
func GetUserByEmail(email string) (*types.User, error) {
config := GetConfig()
conn, err := grpc.Dial(
config.DgraphHost+":"+config.DgraphGRPCPort,
grpc.WithUnaryInterceptor(apmgrpc.NewUnaryClientInterceptor()),
grpc.WithTransportCredentials(insecure.NewCredentials()))
if err != nil {
log.Error().Err(err).Msg("grpc.Dial")
return nil, err
}
defer func(conn *grpc.ClientConn) {
err := conn.Close()
if err != nil {
log.Error().Err(err).Msg("conn.Close")
}
}(conn)
dgraphClient := dgo.NewDgraphClient(api.NewDgraphClient(conn))
ctx := context.Background()
txn := dgraphClient.NewTxn()
defer func(txn *dgo.Txn, ctx context.Context) {
err := txn.Discard(ctx)
if err != nil {
log.Error().Err(err).Msg("txn.Discard")
}
}(txn, ctx)
q := `query user($email: string) {
user(func: eq(email, $email)) {
uid
name
}
}`
req := &api.Request{
Query: q,
Vars: map[string]string{"$email": email},
}
res, err := txn.Do(ctx, req)
if err != nil {
log.Error().Err(err).Msg("txn.Do")
return nil, err
}
uid, err := jsonparser.GetString(res.Json, "user", "[0]", "uid")
if err != nil {
log.Error().
Err(err).
Bytes("res.Json", res.Json).
Msg("jsonparser.GetString uid")
return nil, err
}
name, err := jsonparser.GetString(res.Json, "user", "[0]", "name")
if err != nil {
log.Error().
Err(err).
Bytes("res.Json", res.Json).
Msg("jsonparser.GetString name")
return nil, err
}
return &types.User{
ID: uid,
Name: name,
}, nil
}