@ -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 | |||||
} |