Hulk did something

[bytom/vapor.git] / util / util.go

diff --git a/util/util.go b/util/util.go
new file mode 100644 (file)
index 0000000..f26da96
--- /dev/null
+++ b/util/util.go
@@ -0,0 +1,59 @@
+package util
+
+import (
+       "context"
+
+       jww "github.com/spf13/jwalterweatherman"
+       "github.com/vapor/api"
+       "github.com/vapor/blockchain/rpc"
+       "github.com/vapor/env"
+)
+
+const (
+       // Success indicates the rpc calling is successful.
+       Success = iota
+       // ErrLocalExe indicates error occurs before the rpc calling.
+       ErrLocalExe
+       // ErrConnect indicates error occurs connecting to the bytomd, e.g.,
+       // bytomd can't parse the received arguments.
+       ErrConnect
+       // ErrLocalParse indicates error occurs locally when parsing the response.
+       ErrLocalParse
+       // ErrRemote indicates error occurs in bytomd.
+       ErrRemote
+)
+
+var (
+       coreURL = env.String("BYTOM_URL", "http://127.0.0.1:9888")
+)
+
+// Wraper rpc's client
+func MustRPCClient() *rpc.Client {
+       env.Parse()
+       return &rpc.Client{BaseURL: *coreURL}
+}
+
+// Wrapper rpc call api.
+func ClientCall(path string, req ...interface{}) (interface{}, int) {
+
+       var response = &api.Response{}
+       var request interface{}
+
+       if req != nil {
+               request = req[0]
+       }
+
+       client := MustRPCClient()
+       client.Call(context.Background(), path, request, response)
+
+       switch response.Status {
+       case api.FAIL:
+               jww.ERROR.Println(response.Msg)
+               return nil, ErrRemote
+       case "":
+               jww.ERROR.Println("Unable to connect to the bytomd")
+               return nil, ErrConnect
+       }
+
+       return response.Data, Success
+}