@ -0,0 +1,50 @@ | |||
package supplierrpc | |||
import ( | |||
"encoding/json" | |||
"errors" | |||
) | |||
func Get(site_id, dbname, id string, url ...string) (map[string]string, error) { | |||
if dbname == "" || id == "" { | |||
return nil, errors.New("参数错误") | |||
} | |||
conn, err := rpc_server_conn(url...) | |||
if err != nil { | |||
return nil, err | |||
} | |||
defer conn.Close() | |||
arg := GetParam{site_id, dbname, id} | |||
req, err := SetReqData(arg) | |||
if err != nil { | |||
return nil, err | |||
} | |||
res := &Response{} | |||
err = conn.Get(req, res) | |||
if err != nil { | |||
return nil, err | |||
} | |||
res_data_de, err := GetResData(res) | |||
if err != nil { | |||
return nil, err | |||
} | |||
if res_data_de == "" { | |||
return nil, nil | |||
} | |||
var res_arr map[string]string | |||
err = json.Unmarshal([]byte(res_data_de), &res_arr) | |||
if err != nil { | |||
return nil, err | |||
} | |||
return res_arr, nil | |||
} |
@ -0,0 +1,16 @@ | |||
package supplierrpc | |||
import ( | |||
"testing" | |||
) | |||
func Test_Get(t *testing.T) { | |||
siteid := "1056475" | |||
dbname := "shop_v2" | |||
key := "2" | |||
res, err := Get(siteid, dbname, key) | |||
t.Log(res) | |||
t.Log(err) | |||
} |
@ -0,0 +1,50 @@ | |||
package supplierrpc | |||
import ( | |||
"encoding/json" | |||
"errors" | |||
) | |||
func IsOpen(site_id, dbname, id string, url ...string) (*BoolRet, error) { | |||
if dbname == "" || id == "" { | |||
return nil, errors.New("参数错误") | |||
} | |||
conn, err := rpc_server_conn(url...) | |||
if err != nil { | |||
return nil, err | |||
} | |||
defer conn.Close() | |||
arg := GetParam{site_id, dbname, id} | |||
req, err := SetReqData(arg) | |||
if err != nil { | |||
return nil, err | |||
} | |||
res := &Response{} | |||
err = conn.IsOpen(req, res) | |||
if err != nil { | |||
return nil, err | |||
} | |||
res_data_de, err := GetResData(res) | |||
if err != nil { | |||
return nil, err | |||
} | |||
if res_data_de == "" { | |||
return nil, nil | |||
} | |||
var res_arr BoolRet | |||
err = json.Unmarshal([]byte(res_data_de), &res_arr) | |||
if err != nil { | |||
return nil, err | |||
} | |||
return &res_arr, nil | |||
} |
@ -0,0 +1,17 @@ | |||
package supplierrpc | |||
import ( | |||
"testing" | |||
) | |||
func Test_IsOpen(t *testing.T) { | |||
siteid := "1056475" | |||
dbname := "shop_v2" | |||
key := "1" | |||
res, err := IsOpen(siteid, dbname, key) | |||
t.Log(res) | |||
t.Log(res.Value) | |||
t.Log(err) | |||
} |
@ -0,0 +1,25 @@ | |||
package supplierrpc | |||
import ( | |||
"git.tetele.net/tgo/conf" | |||
) | |||
func rpc_server_conn(url ...string) (*SupplierServiceClient, error) { | |||
var rpc_url string | |||
if len(url) > 0 && url[0] != "" { | |||
rpc_url = url[0] | |||
} else if conf.SUPPLIER_RPC_URL != "" { | |||
rpc_url = conf.SUPPLIER_RPC_URL | |||
} else { | |||
rpc_url = "127.0.0.1:" + conf.SUPPLIER_RPC_PORT | |||
} | |||
conn, _, err := DialSupplierService("tcp", rpc_url) | |||
if err != nil { | |||
return nil, err | |||
} | |||
return conn, nil | |||
} |