go_gameshield/internal/service/gameshield.go

package service

import (
	"context"
	"fmt"
	v1 "github.com/go-nunu/nunu-layout-advanced/api/v1"
	"github.com/go-nunu/nunu-layout-advanced/internal/model"
	"github.com/go-nunu/nunu-layout-advanced/internal/repository"
	"github.com/spf13/viper"
	"strconv"
	"time"
)

type GameShieldService interface {
	SubmitGameShield(ctx context.Context, req *v1.GameShieldSubmitRequest) (string, error)
	EditGameShield(ctx context.Context, req *v1.GameShieldSubmitRequest) (string, error)
	DeleteGameShield(ctx context.Context, req int) (string, error)
	GetGameShieldKey(ctx context.Context, id int) (string, error)
}

func NewGameShieldService(
	service *Service,
	gameShieldRepository repository.GameShieldRepository,
	crawlerService CrawlerService,
	gameShieldPublicIpService GameShieldPublicIpService,
	duedate DuedateService,
	formatter FormatterService,
	parser ParserService,
	required RequiredService,
	conf *viper.Viper,
) GameShieldService {
	return &gameShieldService{
		Service:                   service,
		gameShieldRepository:      gameShieldRepository,
		crawlerService:            crawlerService,
		gameShieldPublicIpService: gameShieldPublicIpService,
		duedate:                   duedate,
		formatter:                 formatter,
		parser:                    parser,
		required:                  required,
		Url:                       conf.GetString("crawler.Url"),
	}
}

type gameShieldService struct {
	*Service
	config                    *viper.Viper
	crawlerService            CrawlerService
	gameShieldRepository      repository.GameShieldRepository
	gameShieldPublicIpService GameShieldPublicIpService
	duedate                   DuedateService
	formatter                 FormatterService
	Url                       string
	parser                    ParserService
	required                  RequiredService
}

func (service *gameShieldService) SubmitGameShield(ctx context.Context, req *v1.GameShieldSubmitRequest) (string, error) {
	nameCount, err := service.gameShieldRepository.GetGameShieldDuplicateName(ctx, req.AppName, req.Uid)
	if err != nil {
		return "", err
	}
	if nameCount > 0 {
		return "", fmt.Errorf("应用名称已存在")
	}
	dunName := strconv.Itoa(req.Uid) + "_" + strconv.FormatInt(time.Now().Unix(), 10) + "_hostId" + req.HostId
	formData := map[string]interface{}{
		"app_name":         dunName,
		"gateway_group_id": 4,
		"backend":          "{}",
		"expired_at":       "",
		"max_device_count": 0,
	}
	respBody, err := service.required.SendForm(ctx, "admin/info/rule/new", "admin/new/rule", formData)
	if err != nil {
		return "", err
	}
	// 解析响应内容中的 alert 消息
	res, err := service.parser.ParseAlert(string(respBody))
	if err != nil {
		return "", err
	}
	if res != "" {
		return "", fmt.Errorf(res)
	}
	KeyAndField, err := service.required.GetKeyAndField(ctx, dunName, "rule_id")
	if err != nil {
		return "", err
	}
	if err := service.gameShieldRepository.AddGameShield(ctx, &model.GameShield{
		AppName:        req.AppName,
		GatewayGroupId: 4,
		AddTime:        time.Now(),
		Uid:            req.Uid,
		HostId:         req.HostId,
		DunName:        dunName,
		RuleId:         KeyAndField.FieldId,
	}); err != nil {
		return "", err
	}
	return "", nil
}

func (service *gameShieldService) EditGameShield(ctx context.Context, req *v1.GameShieldSubmitRequest) (string, error) {
	if err := service.gameShieldRepository.UpdateGameShield(ctx, &model.GameShield{
		AppName: req.AppName,
	}); err != nil {
		return "", err
	}
	return "", nil
}

func (service *gameShieldService) DeleteGameShield(ctx context.Context, id int) (string, error) {
	shieldInfo, err := service.gameShieldRepository.GetGameShieldById(ctx, int64(id))
	if err != nil {
		return "", err
	}
	if err := service.gameShieldRepository.DeleteGameShield(ctx, shieldInfo.RuleId); err != nil {
		return "", err
	}
	res, err := service.crawlerService.DeleteRule(ctx, shieldInfo.RuleId, "admin/delete/rule?page=1&__pageSize=10&__sort=rule_id&__sort_type=desc")
	if err != nil {
		return "", err
	}

	return res, nil
}

func (service *gameShieldService) GetGameShieldKey(ctx context.Context, id int) (string, error) {
	dunName, err := service.gameShieldRepository.GetGameShieldIdByDunName(ctx, int64(id))
	if err != nil {
		return "", err
	}
	if dunName == "" {
		return "", fmt.Errorf("暂无数据")
	}
	res, err := service.crawlerService.GetKey(ctx, dunName)
	if err != nil {
		return "", err
	}
	return res, nil
}