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