message/rpc/message.go

package rpc

import (
	"fmt"
	"log"

	"app.yhyue.com/moapp/MessageCenter/rpc/messageclient"
	"app.yhyue.com/moapp/MessageCenter/rpc/type/message"
	"app.yhyue.com/moapp/jybase/go-logger/logger"
	"github.com/gogf/gf/v2/os/gcfg"
	"github.com/gogf/gf/v2/os/gctx"
	"github.com/zeromicro/go-zero/core/discov"
	"github.com/zeromicro/go-zero/zrpc"
)

var messageClient zrpc.Client

func init() {
	var err error
	messageClient, err = zrpc.NewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: gcfg.Instance().MustGet(gctx.New(), "etcd.address", nil).Strings(),
			Key:   gcfg.Instance().MustGet(gctx.New(), "etcd.messageKey", nil).String(),
		},
	})
	if err != nil {
		log.Fatalln(err)
	}
}

//新用户注册，发送增送剑鱼币消息
func SendMsg(source string, req *message.MultipleSaveMsgReq) error {
	resp, err := messageclient.NewMessage(messageClient).MultipleSaveMsg(gctx.New(), req)
	if err != nil {
		return fmt.Errorf(fmt.Sprintf("%+v", req), source, "发送消息 error", err)
	}
	if resp.Code == 1 {
		logger.Info(fmt.Sprintf("%+v", req), source, "已成功发送消息")
		return nil
	} else {
		return fmt.Errorf(fmt.Sprintf("%+v", req), source, "发送消息失败")
	}
}