package service

import (
	"context"
	"strings"
	"youngee_b_api/db"
	"youngee_b_api/model/http_model"

	"github.com/issue9/conv"

	"github.com/sirupsen/logrus"
)

var Number *number

type number struct {
}

func (*number) GetReviewNumberInfo(ctx context.Context, request http_model.GetReviewNumberInfoRequest) (*http_model.GetReviewNumberInfoData, error) {
	var StrategyIdList []int64
	StrategyIds := strings.Split(request.StrategyIds, ",")
	for _, strategyId := range StrategyIds {
		StrategyIdList = append(StrategyIdList, conv.MustInt64(strategyId))
	}
	NumberData, err := db.GetReviewNumberInfo(ctx, request.ProjectId, StrategyIdList)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call CreateData error,err:%+v", err)
		return nil, err
	}

	return NumberData, nil
}

func (*number) GetLinkNumberInfo(ctx context.Context, request http_model.GetLinkNumberInfoRequest) (*http_model.GetLinkNumberInfoData, error) {
	var StrategyIdList []int64
	StrategyIds := strings.Split(request.StrategyIds, ",")
	for _, strategyId := range StrategyIds {
		StrategyIdList = append(StrategyIdList, conv.MustInt64(strategyId))
	}
	NumberData, err := db.GetLinkNumberInfo(ctx, request.ProjectId, StrategyIdList)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call CreateData error,err:%+v", err)
		return nil, err
	}

	return NumberData, nil
}

func (*number) GetDataNumberInfo(ctx context.Context, request http_model.GetDataNumberInfoRequest) (*http_model.GetDataNumberInfoData, error) {
	var StrategyIdList []int64
	StrategyIds := strings.Split(request.StrategyIds, ",")
	for _, strategyId := range StrategyIds {
		StrategyIdList = append(StrategyIdList, conv.MustInt64(strategyId))
	}
	NumberData, err := db.GetDataNumberInfo(ctx, request.ProjectId, StrategyIdList)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call CreateData error,err:%+v", err)
		return nil, err
	}

	return NumberData, nil
}

func (*number) GetDefaultNumberInfo(ctx context.Context, request http_model.GetDefaultNumberInfoRequest) (*http_model.GetDefaultNumberInfoData, error) {
	var StrategyIdList []int64
	StrategyIds := strings.Split(request.StrategyIds, ",")
	for _, strategyId := range StrategyIds {
		StrategyIdList = append(StrategyIdList, conv.MustInt64(strategyId))
	}
	NumberData, err := db.GetDefaultNumberInfo(ctx, request.ProjectId, StrategyIdList)
	if err != nil {
		logrus.WithContext(ctx).Errorf("[Data service] call CreateData error,err:%+v", err)
		return nil, err
	}

	return NumberData, nil
}