import "github.com/dgraph-io/dgo"
import "github.com/dgraph-io/dgo/protos/api"
import "google.golang.org/grpc"
我在这篇博客将会做以下的工作:
初始化Node
添加Node
查询数据库
为数据库添加Edge , 即 添加predicate
更新数据库
删除Node
之前我已经实现了用dgraph的http客户端实现 dgraph 的各种操作, 但是在 go 客户端实现和 http 客户端实现还是存在着比较大的区别. 因此, 我就写了这篇博客用于记录.
哦, 对了, 另外还有一个关键就是, 在我写这篇博客的时候 dgraph 在GODOC和GOWalker的文档都还没补全, 有很多方法都还只是写了个名, 甚至都没介绍, 对新手非常不友好.
初始化数据库此处, 我将创建五个用户作为基础数据库
初始化Person的结构体, 表的属性从中也可以看出
type Person struct{ Uid string `json:"uid,omitempty"` Name string `json:"name,omitempty"` From string `json:"from,omitempty"` NameOFcn string `json:"nameOFcn,omitempty"` NameOFjp string `json:"nameOFjp,omitempty"` NameOFen string `json:"nameOFen,omitempty"` Age int `json:"age,omitempty"` Friend []Person `json:"friend,omitempty"` }插入五个人的数据
var ( dgraph = flag.String("d", "127.0.0.1:9080", "Dgraph server address") ) func main() { flag.Parse() conn, err := grpc.Dial(*dgraph, grpc.WithInsecure()) if err != nil { log.Fatal(err) } defer conn.Close() dg := dgo.NewDgraphClient(api.NewDgraphClient(conn)) p1 := Person{ Name: "wangha", Age: 17, From: "China", NameOFen: "wangha", NameOFcn: "王哈", NameOFjp: "王ハ", } p2 := Person{ Name: "chenchao", Age: 22, From: "China", NameOFen: "ChaoChen", NameOFcn: "陈超", } p3 := Person{ Name: "xhe", Age: 18, From: "Japan", NameOFen: "wanghe", NameOFcn: "x鹤", } p4 := Person{ Name: "changyang", Age: 19, From: "England", NameOFcn: "常飏", } p5 := Person{ Name: "yetao", Age: 18, From: "Russian", NameOFen: "TaoYe", NameOFcn: "叶掏", } op := &api.Operation{} op.Schema = ` name: string . age: int . from: string . nameOFcn: string @index(term) . nameOFjp: string @index(term) . nameOFen: string @index(term) . ` ctx := context.Background() if err := dg.Alter(ctx, op); err != nil { log.Fatal(err) } mu := &api.Mutation{ CommitNow: true, } var p = [5]Person{p1,p2,p3,p4,p5} for _,x := range p { pb, err := json.Marshal(x) if err != nil { log.Println(err) } mu.SetJson = pb _,err = dg.NewTxn().Mutate(ctx, mu) if err != nil { log.Println(err) } } }我们可以从http客户端查看验证是否插入成功
可以看出, 插入成功了
添加Node此处新注册了一位朋友, 我们需要将他添加到 dgprah 中
和前面的步骤基本一样, 只是少了初始化 schema 的步骤
func main() { flag.Parse() conn, err := grpc.Dial(*dgraph, grpc.WithInsecure()) if err != nil { log.Fatal(err) } defer conn.Close() dg := dgo.NewDgraphClient(api.NewDgraphClient(conn)) ctx := context.Background() mu := &api.Mutation{ CommitNow: true, } type arrays struct{ Uids []Person `json:"info"` } t := Person{ Name : "yaozhao", Age : 24, From : "M78Star", NameOFcn : "姚X", NameOFjp : "姚飞机", NameOFen : "ZhaoYao", } pb, err := json.Marshal(t) if err != nil { log.Println(err) } mu.SetJson = pb assign,err := dg.NewTxn().Mutate(ctx,mu) if err != nil{ log.Println(err) } fmt.Printf("assign: %v \n",assign) } 查询此时,我们查询有关于Name="yaozhao"的信息
func main() { flag.Parse() conn, err := grpc.Dial(*dgraph, grpc.WithInsecure()) if err != nil { log.Fatal(err) } defer conn.Close() dg := dgo.NewDgraphClient(api.NewDgraphClient(conn)) ctx := context.Background() vars := make(map[string]string) vars["$enname"] = "ZhaoYao" q := `query wanghainfo($enname: string){ info(func: eq(nameOFen, $enname)){ uid expand(_all_) } }` resp, err := dg.NewTxn().QueryWithVars(ctx,q,vars) if err != nil { log.Println(err) } type arrays struct{ Uids []Person `json:"info"` } var r arrays err = json.Unmarshal(resp.Json, &r) if err != nil{ log.Println(err) } log.Println(string(resp.Json)) log.Println(r.Uids[0].Uid) }可以看到结果如下:
我在 json 格式进行Mutate时, 没有找到相关的添加Edge的方法, 因此我改为用NQuad格式.
NQuad介绍: RDF/NQuad
func main() { flag.Parse() conn, err := grpc.Dial(*dgraph, grpc.WithInsecure()) if err != nil { log.Fatal(err) } defer conn.Close() dg := dgo.NewDgraphClient(api.NewDgraphClient(conn)) ctx := context.Background() mu := &api.Mutation{ CommitNow: true, } type arrays struct{ Uids []Person `json:"info"` } var1 := make(map[string]string) var1["$enname"] = "wangha" q1 := `query wanghainfo($enname: string){ info(func: eq(nameOFen, $enname)){ uid } }` resp, err := dg.NewTxn().QueryWithVars(ctx,q1,var1) if err != nil { log.Println(err) } var r1 arrays var r2 arrays err = json.Unmarshal(resp.Json, &r1) if err != nil{ log.Println(err) } Uid_wangha := r1.Uids[0].Uid var1["$enname"] = "TaoYe" q2 := `query wanghainfo($enname: string){ info(func: eq(nameOFen, $enname)){ uid } }` resp, err = dg.NewTxn().QueryWithVars(ctx,q2,var1) if err != nil{ log.Println(err) } err = json.Unmarshal(resp.Json, &r2) if err != nil{ log.Println(err) } Uid_TaoYe := r2.Uids[0].Uid t :=fmt.Sprintf("<%s> <friend> <%s> .",Uid_wangha,Uid_TaoYe) mu.SetNquads = []byte(t) _,err = dg.NewTxn().Mutate(ctx,mu) if err != nil{ log.Println(err) } }验证: