package logic

import (
	"context"

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

	"github.com/tal-tech/go-zero/core/logx"
)

type GetEntCountLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewGetEntCountLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetEntCountLogic {
	return &GetEntCountLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

//  获取企业数据总量
func (l *GetEntCountLogic) GetEntCount(in *deduplication.GetEntCountRequest) (*deduplication.GetEntCountResponse, error) {
	count, err := deduplicationService.EntCount(in)
	code := 0
	if err != "" {
		code = -1
	}
	return &deduplication.GetEntCountResponse{
		Data:    count,
		Message: err,
		Code:    int64(code),
	}, nil
}