手动专线回落

src/main/java/com/edgec/browserbackend/browser/controller/TempController.java

@@ -3,10 +3,7 @@ package com.edgec.browserbackend.browser.controller;
 import com.edgec.browserbackend.browser.domain.UserCode;
 import com.edgec.browserbackend.browser.service.TempService;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 /**
  * 一些临时执行的代码可以放在这里，方便删除
@@ -18,8 +15,19 @@ public class TempController {
     @Autowired
     private TempService tempService;
 
+    /**
+     * 记住用户明文密码
+     */
     @PostMapping(value = "/usercode")
     public void addUserCode(@RequestBody UserCode userCode) {
         tempService.save(userCode);
     }
+
+    /**
+     * 专线回落
+     */
+    @PutMapping(value = "/updateip/{addr}/{flag}")
+    public void updateIp(@PathVariable String addr, @PathVariable String flag) {
+        tempService.saveIpResource(addr, flag);
+    }
 }

src/main/java/com/edgec/browserbackend/browser/repository/IpResourceRepositoryCustomImpl.java

@@ -66,8 +66,10 @@ public class IpResourceRepositoryCustomImpl implements IpResourceRepositoryCusto
     @Override
     public List<IpResource> sampleTasks(int status, long timestamp) {
         Criteria matchCriteria = new Criteria();
-        matchCriteria.orOperator(where("status").is(status).and("isLocked").is(false).and("isDeleted").is(false),
-                where("status").is(status).and("isLocked").is(true).and("lockTimestamp").lte(timestamp).and("isDeleted").is(false));
+        matchCriteria.orOperator(
+                where("status").is(status).and("isLocked").is(false).and("isDeleted").is(false),
+                where("status").is(status).and("isLocked").is(true).and("lockTimestamp").lte(timestamp).and("isDeleted").is(false)
+        );
 
         MatchOperation match = Aggregation.match(matchCriteria);
 
@@ -82,8 +84,8 @@ public class IpResourceRepositoryCustomImpl implements IpResourceRepositoryCusto
     public List<IpResource> sampleTasks(List<Integer> status) {
         Criteria matchCriteria = new Criteria();
         matchCriteria.orOperator(
-                where("status").in(status).and("isLocked").is(false).and("validTime").gt(Instant.now().toEpochMilli()).and("isDeleted").is(false).and("healthLockTimestamp").lte(Instant.now().minusSeconds(60 * 30).toEpochMilli()),
-                where("status").in(status).and("isLocked").is(true).and("healthLockTimestamp").lte(Instant.now().minusSeconds(600).toEpochMilli()).and("isDeleted").is(false)
+                where("status").in(status).and("isLocked").is(false).and("isDeleted").is(false).and("healthLockTimestamp").lte(Instant.now().minusSeconds(60 * 30).toEpochMilli()).and("validTime").gt(Instant.now().toEpochMilli()),
+                where("status").in(status).and("isLocked").is(true).and("isDeleted").is(false).and("healthLockTimestamp").lte(Instant.now().minusSeconds(60 * 10).toEpochMilli())
         );
 
         MatchOperation match = Aggregation.match(matchCriteria);

src/main/java/com/edgec/browserbackend/browser/service/Impl/TempServiceImpl.java

 package com.edgec.browserbackend.browser.service.Impl;
 
+import com.edgec.browserbackend.browser.domain.IpResource;
 import com.edgec.browserbackend.browser.domain.UserCode;
+import com.edgec.browserbackend.browser.repository.IpResourceRepository;
 import com.edgec.browserbackend.browser.repository.UserCodeRepository;
 import com.edgec.browserbackend.browser.service.TempService;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -14,8 +16,26 @@ public class TempServiceImpl implements TempService {
     @Autowired
     private UserCodeRepository userCodeRepository;
 
+    @Autowired
+    private IpResourceRepository ipResourceRepository;
+
     @Override
     public void save(UserCode userCode) {
         userCodeRepository.save(userCode);
     }
+
+    @Override
+    public void saveIpResource(String addr, String flag) {
+        IpResource ipResource = ipResourceRepository.findByAddr(addr);
+
+        if ("true".equals(flag)) {
+            ipResource.setSpecialLine(true);
+        }
+
+        if ("false".equals(flag)) {
+            ipResource.setSpecialLine(false);
+        }
+
+        ipResourceRepository.save(ipResource);
+    }
 }

src/main/java/com/edgec/browserbackend/browser/service/TempService.java

@@ -7,4 +7,6 @@ import com.edgec.browserbackend.browser.domain.UserCode;
  */
 public interface TempService {
     void save(UserCode userCode);
+
+    void saveIpResource(String addr, String flag);
 }