resourceCenter/test/resource.go

package main

import (
	"context"
	"flag"
	"log"

	"bp.jydev.jianyu360.cn/BaseService/resourceCenter/rpc/pb"
	"bp.jydev.jianyu360.cn/BaseService/resourceCenter/rpc/resource"
	"github.com/zeromicro/go-zero/core/discov"
	"github.com/zeromicro/go-zero/zrpc"
)

var c = resource.NewResource(zrpc.MustNewClient(zrpc.RpcClientConf{
	Etcd: discov.EtcdConf{
		Hosts: []string{"192.168.3.206:2379"},
		Key:   "resource.rpc",
	},
}))

func main() {
	m := flag.Int("m", 0, "")
	flag.Parse()
	if *m == -1 {
		CancelPowerHandle()
	} else if *m == 1 {
		OpenPowerHandle()
	} else if *m == 2 {
		CheckPower()
	} else if *m == 3 {
		Deduction()
	} else if *m == 4 {
		ReEmpower()
	}
}

//开通或者取消用户/企业权益
func OpenPowerHandle() {
	r, err := c.PowerHandle(context.Background(), &pb.PowerReq{
		Appid:     "10000",
		GoodsCode: "xxfb_gyxx",
		EntId:     55,
		Type:      1,
	})
	log.Println("----------", r.Status, err)
}

//开通或者取消用户/企业权益
func CancelPowerHandle() {
	r, err := c.PowerHandle(context.Background(), &pb.PowerReq{
		Appid:     "10000",
		GoodsCode: "xxfb_gyxx",
		EntId:     55,
		Type:      -1,
	})
	log.Println("----------", r.Status, err)
}

//检查用户/企业权益
func CheckPower() {
	r, err := c.CheckPower(context.Background(), &pb.CheckPowerReq{
		Appid:        "10000",
		FunctionCode: "znsj_kf_use",
		EntId:        14544,
		EntUserId:    3536,
	})
	log.Println("----------", r.Status, err)
}

//资源扣减
func Deduction() {
	r, err := c.Deduction(context.Background(), &pb.DeductionReq{
		Appid:        "10000",
		FunctionCode: "xxfb_gyxx_add",
		EntId:        55,
		Count:        1,
	})
	log.Println("----------", r.Status, err)
}

//资源扣减
func ReEmpower() {
	r, err := c.ReEmpower(context.Background(), &pb.EmpowerReq{
		Appid:        "10000",
		FunctionCode: "znsj_kf_use",
		EntId:        14184,
		EntUserId:    []int64{4271},
	})
	log.Println("----------", r.Status, err)
}