tijiao

rpc/integralclient/integral.go

@@ -14,15 +14,24 @@ import (
 )
 
 type (
-	AddReq    = integral.AddReq
-	AddResp   = integral.AddResp
 	CheckReq  = integral.CheckReq
 	CheckResp = integral.CheckResp
+	AddReq    = integral.AddReq
+	AddResp   = integral.AddResp
 
 	Integral interface {
-		Add(ctx context.Context, in *AddReq) (*AddResp, error)
-		Update(ctx context.Context, in *AddReq) (*AddResp, error)
-		Select(ctx context.Context, in *CheckReq) (*CheckResp, error)
+		// 新增积分
+		AddFlow(ctx context.Context, in *AddReq) (*AddResp, error)
+		// 调整结存
+		UpdateBalance(ctx context.Context, in *AddReq) (*AddResp, error)
+		// 调整余额
+		UpdateTotal(ctx context.Context, in *AddReq) (*AddResp, error)
+		// 消耗积分
+		IntegralConsume(ctx context.Context, in *AddReq) (*AddResp, error)
+		// 积分守护
+		IntegralGuard(ctx context.Context, in *AddReq) (*AddResp, error)
+		// 到期积分查询
+		IntegralCheck(ctx context.Context, in *AddReq) (*AddResp, error)
 	}
 
 	defaultIntegral struct {
@@ -36,17 +45,38 @@ func NewIntegral(cli zrpc.Client) Integral {
 	}
 }
 
-func (m *defaultIntegral) Add(ctx context.Context, in *AddReq) (*AddResp, error) {
+// 新增积分
+func (m *defaultIntegral) AddFlow(ctx context.Context, in *AddReq) (*AddResp, error) {
+	client := integral.NewIntegralClient(m.cli.Conn())
+	return client.AddFlow(ctx, in)
+}
+
+// 调整结存
+func (m *defaultIntegral) UpdateBalance(ctx context.Context, in *AddReq) (*AddResp, error) {
+	client := integral.NewIntegralClient(m.cli.Conn())
+	return client.UpdateBalance(ctx, in)
+}
+
+// 调整余额
+func (m *defaultIntegral) UpdateTotal(ctx context.Context, in *AddReq) (*AddResp, error) {
+	client := integral.NewIntegralClient(m.cli.Conn())
+	return client.UpdateTotal(ctx, in)
+}
+
+// 消耗积分
+func (m *defaultIntegral) IntegralConsume(ctx context.Context, in *AddReq) (*AddResp, error) {
 	client := integral.NewIntegralClient(m.cli.Conn())
-	return client.Add(ctx, in)
+	return client.IntegralConsume(ctx, in)
 }
 
-func (m *defaultIntegral) Update(ctx context.Context, in *AddReq) (*AddResp, error) {
+// 积分守护
+func (m *defaultIntegral) IntegralGuard(ctx context.Context, in *AddReq) (*AddResp, error) {
 	client := integral.NewIntegralClient(m.cli.Conn())
-	return client.Update(ctx, in)
+	return client.IntegralGuard(ctx, in)
 }
 
-func (m *defaultIntegral) Select(ctx context.Context, in *CheckReq) (*CheckResp, error) {
+// 到期积分查询
+func (m *defaultIntegral) IntegralCheck(ctx context.Context, in *AddReq) (*AddResp, error) {
 	client := integral.NewIntegralClient(m.cli.Conn())
-	return client.Select(ctx, in)
+	return client.IntegralCheck(ctx, in)
 }