|
@@ -9,7 +9,6 @@ import (
|
|
|
"github.com/spf13/viper"
|
|
|
"strconv"
|
|
|
"strings"
|
|
|
- "time"
|
|
|
)
|
|
|
|
|
|
type GameShieldBackendService interface {
|
|
@@ -80,8 +79,11 @@ func (s *gameShieldBackendService) GetGameShieldRequired(ctx context.Context, re
|
|
|
if err != nil {
|
|
|
return nil, 0, err
|
|
|
}
|
|
|
- OldBackend, err := s.formatter.OldFormat(ctx, oldBackend)
|
|
|
- res.Backend, count, err = s.formatter.FormatBackendData(ctx, req, OldBackend)
|
|
|
+ if len(oldBackend) != 0 {
|
|
|
+ count = oldBackend[0].KeySort
|
|
|
+ }
|
|
|
+ OldBackend, err := s.formatter.OldFormat(ctx, &oldBackend)
|
|
|
+ res.Backend, err = s.formatter.FormatBackendData(ctx, req, OldBackend, count)
|
|
|
if err != nil {
|
|
|
return nil, 0, err
|
|
|
}
|
|
@@ -89,7 +91,7 @@ func (s *gameShieldBackendService) GetGameShieldRequired(ctx context.Context, re
|
|
|
if err != nil {
|
|
|
return nil, 0, err
|
|
|
}
|
|
|
- res.Domain, err = s.gameShieldRepository.GetHostById(ctx, req.HostId)
|
|
|
+ res.DunName = gameShield.DunName
|
|
|
return &res, count, nil
|
|
|
}
|
|
|
func (s *gameShieldBackendService) GetGameShieldBackend(ctx context.Context, id int64) (*model.GameShieldBackend, error) {
|
|
@@ -105,14 +107,13 @@ func (s *gameShieldBackendService) GameShieldBackend(ctx context.Context, req *v
|
|
|
if err != nil {
|
|
|
return "", 0, err
|
|
|
}
|
|
|
- tokenUrl := s.Url + "admin/info/rule/edit?&__goadmin_edit_pk=" + strconv.Itoa(require.RuleId) + "_" + require.Domain
|
|
|
+ tokenUrl := s.Url + "admin/info/rule/edit?&__goadmin_edit_pk=" + strconv.Itoa(require.RuleId) + "_" + require.DunName
|
|
|
tokens, err := s.crawlerService.GetFormTokens(ctx, tokenUrl, require.Cookie)
|
|
|
if err != nil {
|
|
|
return "", 0, err
|
|
|
}
|
|
|
- dunName := strconv.Itoa(req.Uid) + "_" + strconv.FormatInt(time.Now().Unix(), 10) + "_" + require.Domain
|
|
|
formData := map[string]interface{}{
|
|
|
- "app_name": dunName,
|
|
|
+ "app_name": require.DunName,
|
|
|
"gateway_group_id": 4,
|
|
|
"backend": require.Backend,
|
|
|
"rule_id": require.RuleId,
|
|
@@ -134,11 +135,11 @@ func (s *gameShieldBackendService) GameShieldBackend(ctx context.Context, req *v
|
|
|
if res != "" {
|
|
|
return "", 0, fmt.Errorf(res)
|
|
|
}
|
|
|
- KeyAndField, err := s.required.GetKeyAndField(ctx, dunName, "rule_id")
|
|
|
+ KeyAndField, err := s.required.GetKeyAndField(ctx, require.DunName, "rule_id")
|
|
|
if err != nil {
|
|
|
return "", 0, err
|
|
|
}
|
|
|
- if err := s.gameShieldRepository.UpdateGameShieldByHostId(ctx, &model.GameShield{Id: req.HostId, Key: KeyAndField.Key, DunName: dunName}); err != nil {
|
|
|
+ if err := s.gameShieldRepository.UpdateGameShieldByHostId(ctx, &model.GameShield{Id: req.HostId, Key: KeyAndField.Key}); err != nil {
|
|
|
return "", 0, err
|
|
|
}
|
|
|
return res, count, nil
|