go_gameshield/internal/service/tcplimit.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"
	"strconv"
)

type TcpLimitService interface {
	GetTcpLimit(ctx context.Context, id int64) (*model.TcpLimit, error)
	AddTcpLimit(ctx context.Context, req *v1.TcpLimitRequest) (string, error)
	UpdateTcpLimit(ctx context.Context, req *v1.TcpLimitRequest) (string, error)
	DeleteTcpLimit(ctx context.Context, TcpWebLimitId int) (string, error)
}

func NewTcpLimitService(
	service *Service,
	tcpLimitRepository repository.TcpLimitRepository,
	required RequiredService,
	parser ParserService,
	crawler CrawlerService,
) TcpLimitService {
	return &tcpLimitService{
		Service:            service,
		tcpLimitRepository: tcpLimitRepository,
		required:           required,
		parser:             parser,
		crawler:            crawler,
	}
}

type tcpLimitService struct {
	*Service
	tcpLimitRepository repository.TcpLimitRepository
	required           RequiredService
	parser             ParserService
	crawler            CrawlerService
}

func (s *tcpLimitService) GetTcpLimit(ctx context.Context, id int64) (*model.TcpLimit, error) {
	return s.tcpLimitRepository.GetTcpLimit(ctx, id)
}

func (s *tcpLimitService) AddTcpLimit(ctx context.Context, req *v1.TcpLimitRequest) (string, error) {
	formData := map[string]interface{}{
		"tag":             req.Tag,
		"conn_count":      req.ConnCount,
		"conn_duration":   req.ConnDuration,
		"bps":             req.Bps,
		"max_bytes_month": req.MaxBytesMonth,
		"expired_at":      "",
	}
	respBody, err := s.required.SendForm(ctx, "admin/info/waf_tcp_limit/new", "admin/new/waf_tcp_limit", formData)
	if err != nil {
		return "", err
	}
	// 解析响应内容中的 alert 消息
	res, err := s.parser.ParseAlert(string(respBody))
	if err != nil {
		return "", err
	}
	if res != "" {
		return "", fmt.Errorf(res)
	}
	tcpLimitId, err := s.parser.GetRuleId(ctx, respBody)
	if err != nil {
		return "", err
	}
	fmt.Println("=================================================", tcpLimitId)
	return "", nil
}

func (s *tcpLimitService) UpdateTcpLimit(ctx context.Context, req *v1.TcpLimitRequest) (string, error) {
	formData := map[string]interface{}{
		"tag":              req.Tag,
		"waf_tcp_limit_id": req.TcpWebLimitId,
		"conn_count":       req.ConnCount,
		"conn_duration":    req.ConnDuration,
		"bps":              req.Bps,
		"max_bytes_month":  req.MaxBytesMonth,
		"expired_at":       "",
	}
	respBody, err := s.required.SendForm(ctx, "admin/info/waf_tcp_limit/edit?&__goadmin_edit_pk="+strconv.Itoa(req.TcpWebLimitId), "admin/edit/waf_tcp_limit", formData)
	if err != nil {
		return "", err
	}
	// 解析响应内容中的 alert 消息
	res, err := s.parser.ParseAlert(string(respBody))
	if err != nil {
		return "", err
	}
	if res != "" {
		return "", fmt.Errorf(res)
	}
	return "", nil
}

func (s *tcpLimitService) DeleteTcpLimit(ctx context.Context, TcpWebLimitId int) (string, error) {
	res, err := s.crawler.DeleteRule(ctx, TcpWebLimitId, "admin/delete/waf_tcp_limit?page=1&__pageSize=10&__sort=waf_tcp_limit_id&__sort_type=desc")
	if err != nil {
		return "", err
	}
	return res, nil
}