Browse Source

refactor(internal): 优化代码结构和错误处理

- 移除了不必要的打印语句
- 优化了错误处理逻辑,提高了代码的健壮性
- 调整了代码格式,提高了可读性
fusu 3 tháng trước cách đây
mục cha
commit
8df3646f87

+ 2 - 3
internal/service/formatter.go

@@ -119,17 +119,16 @@ func (service *formatterService) TidyFormatBackendData(ctx context.Context, req
 	if err != nil {
 		return nil, err
 	}
-	fmt.Println("=====================================req.item", req.Items)
 	for _, item := range req.Items {
 		// 提取必要字段
 		sourceIP := item.SourceMachineIP // 假设结构体中有这个字段
 		if sourceIP == "" {
-			continue // 跳过没有有效源IP的配置
+			return nil, fmt.Errorf("没有有效源IP的配置") // 跳过没有有效源IP的配置
 		}
 
 		protocol := item.Protocol // 假设结构体中有这个字段
 		if protocol == "" {
-			continue // 跳过没有有效协议的配置
+			return nil, fmt.Errorf("没有有效协议的配置") // 跳过没有有效协议的配置
 		}
 		// 获取端口数组
 		conPorts := service.FormatPort(ctx, item.ConnectPort)

+ 1 - 0
internal/service/gameshield.go

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

+ 0 - 4
internal/service/gameshieldbackend.go

@@ -146,13 +146,11 @@ func (s *gameShieldBackendService) GameShieldBackend(ctx context.Context, req *v
 }
 
 func (s *gameShieldBackendService) AddGameShieldBackend(ctx context.Context, req *v1.GameShieldBackendArrayRequest) (string, error) {
-	fmt.Println("=====================================", req)
 	res, count, err := s.GameShieldBackend(ctx, req)
 	if err != nil {
 		return "", err
 	}
 	saveData, err := s.formatter.TidyFormatBackendData(ctx, req, count)
-	fmt.Println("=====================================saveData", saveData)
 	if err != nil {
 		return "", err
 	}
@@ -162,7 +160,6 @@ func (s *gameShieldBackendService) AddGameShieldBackend(ctx context.Context, req
 	return res, nil
 }
 func (s *gameShieldBackendService) EditGameShieldBackend(ctx context.Context, req *v1.GameShieldBackendArrayRequest) (string, error) {
-	fmt.Println("=====================================", req)
 
 	for _, v := range req.Items {
 		if v.Id == 0 {
@@ -193,7 +190,6 @@ func (s *gameShieldBackendService) DeleteGameShieldBackend(ctx context.Context,
 }
 
 func (s *gameShieldBackendService) SaveGameShieldBackend(ctx context.Context, req map[string]v1.SendGameShieldBackend, hostId int) error {
-	fmt.Println("=====================================", req)
 	for k, v := range req {
 		parts := strings.Split(v.Addr[0], ":")
 		keyName := strings.Split(k, "key")[1]