Emulate user queries in a script

Author: rahulgurnani

scripts/emulate_queries.go

@@ -0,0 +1,84 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"net/http"
+)
+
+const BASE_URL = "http://localhost:8080"
+
+func post(contentType string, suffixURL string, data []byte) string {
+	req, err := http.NewRequest("POST", BASE_URL+suffixURL, bytes.NewBuffer(data))
+	//req.Header.Set("X-Custom-Header", "myvalue")
+	req.Header.Set("Content-Type", contentType)
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		panic(err)
+	}
+	defer resp.Body.Close()
+
+	fmt.Println("response Status:", resp.Status)
+	fmt.Println("response Headers:", resp.Header)
+	body, _ := ioutil.ReadAll(resp.Body)
+	fmt.Println("response Body:", string(body))
+
+	return string(body)
+}
+
+func read_query(res chan string) {
+	data, err := ioutil.ReadFile("read_nodes.graphql")
+	if err != nil {
+		log.Println("Error")
+	}
+
+	body := post("application/graphql+-", "/query", data)
+	res <- body
+}
+
+func mutate_schema_query(res chan string) {
+	data, err := ioutil.ReadFile("mutate_schema.graphql")
+	if err != nil {
+		log.Println("Error")
+	}
+	print(string(data))
+	body := post("application/json", "/alter", data)
+	res <- body
+}
+
+func write_query(res chan string) {
+	data, err := ioutil.ReadFile("mutate_data.graphql")
+	if err != nil {
+		log.Println("Error")
+	}
+
+	body := post("application/rdf", "/mutate", data)
+	res <- body
+}
+
+func export_query(res chan string) {
+	data, err := ioutil.ReadFile("export.graphql")
+	if err != nil {
+		log.Println("Error")
+	}
+
+	body := post("application/graphql", "/admin", data)
+	res <- body
+}
+
+func main() {
+	res1 := make(chan string)
+	res2 := make(chan string)
+	res3 := make(chan string)
+	go read_query(res1)
+	go mutate_schema_query(res2)
+	go write_query(res3)
+	//go export_query(res2)
+	<-res1
+	<-res2
+	<-res3
+}

scripts/export.graphql

@@ -0,0 +1,8 @@
+mutation {
+  export(input: {format: "rdf"}) {
+    response {
+      message
+      code
+    }
+  }
+}

scripts/mutate_data.graphql

@@ -0,0 +1,13 @@
+{
+  set {
+    _:company1 <name> "alpha" .
+    _:company1 <dgraph.type> "Company" .
+    _:company2 <name> "beta" .
+    _:company2 <dgraph.type> "Company" .
+
+    _:company1 <industry> "Machinery" .
+
+    _:company2 <industry> "High Tech" .
+
+  }
+}

scripts/mutate_schema.graphql

@@ -0,0 +1 @@
+{"drop_op": "TYPE", "drop_value": "Company"}

scripts/read_nodes.graphql

@@ -0,0 +1,7 @@
+{
+  data(func: eq(name, "Alice")) {
+     name
+     mobile @facets
+     car @facets
+  }
+}