站点RPC方法
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

95 lines
1.6 KiB

package siterpc
import (
"encoding/json"
"log"
"strconv"
"time"
"git.tetele.net/tgo/conf"
"git.tetele.net/tgo/crypter"
"github.com/golang/protobuf/proto"
)
type SiteInfoReq struct {
Master string
Host string
}
type SiteInfoRes struct {
SiteId string `json:"site_id"`
Dbname string `json:"database"`
Domainname string `json:"domainname"`
}
/**
* 由host取对应站点关键信息
* 请求及回均加密验签
* 2021/06/24
* GZ
*/
func GetSiteInfoByHost(host string, params ...string) (*SiteInfoRes, error) {
var res_arr SiteInfoRes
if conf.IS_PRIVATE {
res_arr.Dbname = conf.DBNAME
res_arr.SiteId = conf.SITE_ID
res_arr.Domainname = conf.DOMAIN
return &res_arr, nil
}
var url string = ""
var master string = ""
if len(params) > 1 {
master = params[1]
}
if len(params) > 0 {
url = params[0]
}
conn, err := rpc_server_conn(url)
if err != nil {
return nil, err
}
defer conn.Close()
arg := SiteInfoReq{master, host}
data_json, err := json.Marshal(arg)
if err != nil {
return nil, err
}
now_int64 := time.Now().Unix()
encryData := crypter.DesEn(string(data_json), DES_KEY)
now := strconv.FormatInt(now_int64, 10)
sign := Sign(encryData, now)
req := &Request{proto.String(encryData), proto.String(now), proto.String(sign), nil}
res := &Response{}
err = conn.GetSiteInfoByHost(req, res)
if err != nil {
return nil, err
}
res_data, err := GetOrgData(res)
log.Println(res_data, err)
if err != nil {
return nil, err
}
err = json.Unmarshal([]byte(res_data), &res_arr)
if err != nil {
return nil, err
}
return &res_arr, nil
}