dev add monitoring basic

2026-03-03 15:43:47 +07:00
parent 404e1a6bee
commit 38e7a382c7
6 changed files with 155 additions and 20 deletions
@@ -0,0 +1,86 @@
+package ru.soune.nocopy.handler;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+import ru.soune.nocopy.dto.BaseRequest;
+import ru.soune.nocopy.dto.BaseResponse;
+import ru.soune.nocopy.dto.MessageCode;
+import ru.soune.nocopy.dto.cost.CostRequest;
+import ru.soune.nocopy.dto.cost.CostResponse;
+import ru.soune.nocopy.entity.user.AuthToken;
+import ru.soune.nocopy.entity.user.User;
+import ru.soune.nocopy.repository.AuthTokenRepository;
+import ru.soune.nocopy.service.cost.CostService;
+
+import java.util.Optional;
+
+@Slf4j
+@Component
+@RequiredArgsConstructor
+public class CostHandler implements RequestHandler {
+
+    private final CostService costService;
+
+    private final AuthTokenRepository authTokenRepository;
+
+    private final ObjectMapper objectMapper;
+
+    @Override
+    public BaseResponse handle(BaseRequest request) throws Exception {
+        CostRequest costRequest = objectMapper.convertValue(request.getMessageBody(), CostRequest.class);
+        String action = costRequest.getAction();
+
+        return switch (action) {
+            case "costFileProtect" -> handleFileProtectCost(costRequest, request);
+            case "checkTokensForProtect" -> handleCheckTokensForFilesProtect(costRequest, request);
+            default -> new BaseResponse(
+                    request.getMsgId(),
+                    MessageCode.INVALID_ACTION.getCode(),
+                    "Invalid action: " + action,
+                    null);
+        };
+    }
+
+    private BaseResponse handleFileProtectCost(CostRequest costRequest, BaseRequest request) {
+        int cost = costService.protectFileCost(costRequest.getFileType());
+
+        return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
+                MessageCode.SUCCESS.getDescription(),
+                CostResponse.builder()
+                        .cost(cost)
+                        .build());
+    }
+
+    private BaseResponse handleCheckTokensForFilesProtect(CostRequest costRequest, BaseRequest request) {
+        String authToken = costRequest.getAuthToken();
+        Optional<AuthToken> token = authTokenRepository.findByToken(authToken);
+
+        if (token.isEmpty()) {
+           return new BaseResponse(
+                    request.getMsgId(),
+                    MessageCode.AUTH_TOKEN_NOT_FOUND.getCode(),
+                    MessageCode.AUTH_TOKEN_NOT_FOUND.getDescription(),
+                    null);
+        }
+
+        User user = token.get().getUser();
+        Integer tokens = user.getActiveTariffInfo().getTokens();
+
+        int cost = costService.protectFileCost(costRequest.getFileType());
+        int count = costRequest.getCountFilesForProtect();
+        int totalCost = count * cost;
+        boolean success = totalCost <= tokens;
+        int needToken = success ? 0: tokens - totalCost;
+
+        CostResponse costResponse = CostResponse.builder()
+                .cost(totalCost)
+                .success(totalCost <= tokens)
+                .needTokensCount(Math.abs(needToken))
+                .build();
+
+        return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
+                MessageCode.SUCCESS.getDescription(), costResponse);
+    }
+}