/
get_user.go
67 lines (61 loc) · 1.59 KB
/
get_user.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
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 GetUser(id 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($uid: string) {
user(func: uid($uid)) {
name
}
}`
req := &api.Request{
Query: q,
Vars: map[string]string{"$uid": id},
}
res, err := txn.Do(ctx, req)
if err != nil {
log.Error().Err(err).Msg("txn.Do")
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")
return nil, err
}
return &types.User{ID: id, Name: name}, nil
}