msgrpc/client.go

package msgrpc

import (
    "encoding/json"
    "errors"
    "log"
    "strconv"
    "time"

    "git.tetele.net/tgo/crypter"

    "github.com/golang/protobuf/proto"
)

func SendMsg(site_id, dbname, business_id, domainname, msg_type, order_sn, memo string, data interface{}, url ...string) (*MsgSend, error) {

    conn, err := rpc_server_conn(url...)
    if err != nil {
        return nil, err
    }
    defer conn.Close()

    arg_data, err := json.Marshal(data)
    if err != nil {
        log.Println("json marshal error:", err)
    }

    arg := MsgBody{}
    arg.SiteId = site_id
    arg.Dbname = dbname
    arg.Platform = "1"
    arg.BusinessId = business_id
    arg.Domainname = domainname
    arg.Type = msg_type
    arg.Data = string(arg_data)
    arg.OrderSn = order_sn
    arg.Memo = memo

    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.SendMsg(req, res)

    if err != nil {
        return nil, err
    }

    res_data, err := GetOrgData(res)

    var res_arr MsgSend

    err = json.Unmarshal([]byte(res_data), &res_arr)

    if err != nil {
        return nil, err
    }
    return &res_arr, nil

}

// target_type:目标类型，商品或文章   target_id：目标ID
func SendNotcie(site_id, dbname, manager_id, target_type, target_id string, template_Data map[string]string, url ...string) (bool, error) {
    if dbname == "" || dbname == "" || manager_id == "" {
        return false, errors.New("参数错误")
    }

    conn, err := rpc_server_conn(url...)

    if err != nil {
        return false, err
    }
    defer conn.Close()

    arg := NoticeSend{}
    arg.SiteId = site_id
    arg.Dbname = dbname
    arg.ManagerId = manager_id
    arg.TargetType = target_type
    arg.TargetId = target_id
    arg.TemplateData = template_Data

    data_json, err := json.Marshal(arg)

    if err != nil {
        return false, 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.SendNotcie(req, res)

    if err != nil {
        log.Println("msgrpc SendNotcie error:", err)
        return false, err
    }

    res_data, err := GetOrgData(res)

    if err != nil {
        return false, err
    }
    if res_data == "" {
        return false, nil
    }

    var notice_res bool

    err = json.Unmarshal([]byte(res_data), &notice_res)

    if err != nil {
        return false, err
    }
    return notice_res, nil

}