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{"127.0.0.1:2379"},
		Key:   "resource.rpc",
	},
}))

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

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

//检查用户/企业权益
func CheckPower() {
	r, err := c.CheckPower(context.Background(), &pb.CheckPowerReq{
		Appid:        "10000",
		FunctionCode: "xxfb_gyxx_add",
		EntId:        55,
	})
	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)
}