dataDeduplication/rpc/deduplicationclient/deduplication.go

// Code generated by goctl. DO NOT EDIT!
// Source: deduplication.proto

//go:generate mockgen -destination ./deduplication_mock.go -package deduplicationclient -source $GOFILE

package deduplicationclient

import (
	"context"

	"app.yhyue.com/moapp/dataDeduplication/rpc/deduplication"

	"github.com/tal-tech/go-zero/zrpc"
)

type (
	Request             = deduplication.Request
	Info                = deduplication.Info
	GetEntCountRequest  = deduplication.GetEntCountRequest
	Response            = deduplication.Response
	GetEntCountResponse = deduplication.GetEntCountResponse
	ByAccountRequest    = deduplication.ByAccountRequest

	Deduplication interface {
		//  数据判重
		DataDeduplication(ctx context.Context, in *Request) (*Response, error)
		//  获取企业数据总量
		GetEntCount(ctx context.Context, in *GetEntCountRequest) (*GetEntCountResponse, error)
		//   根据账户id进行判重
		DataDeduplicateByAccount(ctx context.Context, in *ByAccountRequest) (*Response, error)
		//   根据账户id判重并存入数据
		DataDeduplicateAndSave(ctx context.Context, in *ByAccountRequest) (*Response, error)
	}

	defaultDeduplication struct {
		cli zrpc.Client
	}
)

func NewDeduplication(cli zrpc.Client) Deduplication {
	return &defaultDeduplication{
		cli: cli,
	}
}

//  数据判重
func (m *defaultDeduplication) DataDeduplication(ctx context.Context, in *Request) (*Response, error) {
	client := deduplication.NewDeduplicationClient(m.cli.Conn())
	return client.DataDeduplication(ctx, in)
}

//  获取企业数据总量
func (m *defaultDeduplication) GetEntCount(ctx context.Context, in *GetEntCountRequest) (*GetEntCountResponse, error) {
	client := deduplication.NewDeduplicationClient(m.cli.Conn())
	return client.GetEntCount(ctx, in)
}

//   根据账户id进行判重
func (m *defaultDeduplication) DataDeduplicateByAccount(ctx context.Context, in *ByAccountRequest) (*Response, error) {
	client := deduplication.NewDeduplicationClient(m.cli.Conn())
	return client.DataDeduplicateByAccount(ctx, in)
}

//   根据账户id判重并存入数据
func (m *defaultDeduplication) DataDeduplicateAndSave(ctx context.Context, in *ByAccountRequest) (*Response, error) {
	client := deduplication.NewDeduplicationClient(m.cli.Conn())
	return client.DataDeduplicateAndSave(ctx, in)
}