jy-docs/jydocs-back/rpc/partnerRpc.go

package rpc

import (
	"app.yhyue.com/moapp/jy_docs/rpc/partnerlib/partner"
	"context"
	"github.com/zeromicro/go-zero/core/discov"
	"github.com/zeromicro/go-zero/zrpc"
	"jy-docs/config"
	"jy-docs/public"
	"log"
)

// 标准库RPC接口
var jyParterLib partner.Partner

func init() {
	jyParterLib = partner.NewPartner(zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Key:   config.JyDocsAppConfig.RpcServers.Partner.Key,
			Hosts: config.JyDocsAppConfig.RpcServers.Partner.Address,
		}, Timeout: config.JyDocsAppConfig.RpcServers.Partner.Timeout,
	}))
}

// PartDocDownload 下载
func PartDocDownload(docId, mgoUserId, phone string, positionId int64) (*partner.UDRes, error) {
	resp, err := jyParterLib.DocDownload(context.Background(), &partner.UserDownloadRequest{
		MgoUserId:  mgoUserId,
		Phone:      phone,
		PositionId: positionId,
		AppId:      public.AppID,
		State:      public.PartnerState,
		DocId:      docId,
	})
	if err != nil || resp == nil {
		log.Printf("%s %d  jyParterLib.DocDownload call error %v\n", mgoUserId, positionId, err)
		return resp, err
	}
	if resp.Code < 0 {
		log.Printf("%s %d  jyParterLib.DocDownload call code error %v\n", mgoUserId, positionId, resp)
		return resp, err
	}
	return resp, nil
}

// PartUserBuy 购买
func PartUserBuy(docId, mgoUserId, phone string, positionId int64, buyType int64) (*partner.UBVRes, error) {
	resp, err := jyParterLib.UserBuy(context.Background(), &partner.UserBuyVipRequest{
		MgoUserId:  mgoUserId,
		Phone:      phone,
		PositionId: positionId,
		AppId:      public.AppID,
		State:      public.PartnerState,
		DocId:      docId,
		Type:       buyType,
	})
	if err != nil || resp == nil {
		log.Printf("%s %d  jyParterLib.PartUserBuy call error %v\n", mgoUserId, positionId, err)
		return resp, err
	}
	if resp.Code < 0 {
		log.Printf("%s %d  jyParterLib.PartUserBuy call code error %v\n", mgoUserId, positionId, resp)
		return resp, err
	}
	return resp, nil
}