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