生成登录用户的 proxy 日志

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

 package com.edgec.browserbackend.browser.repository;
 
-import com.edgec.browserbackend.account.repository.FileRepository;
 import com.edgec.browserbackend.account.repository.FileRepositoryImpl;
 import com.edgec.browserbackend.browser.domain.ProxyConfig;
-import com.edgec.browserbackend.browser.domain.UserShop;
-import com.mongodb.client.result.DeleteResult;
 import com.mongodb.client.result.UpdateResult;
 import org.bson.Document;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.data.mongodb.core.MongoTemplate;
 import org.springframework.data.mongodb.core.query.BasicQuery;
 import org.springframework.data.mongodb.core.query.Update;
 
 import java.io.File;
 
-import static org.springframework.data.mongodb.core.query.Criteria.where;
-
 public class ProxyConfigRepositoryCustomImpl implements ProxyConfigRepositoryCustom {
 
     @Autowired
@@ -33,10 +27,8 @@ public class ProxyConfigRepositoryCustomImpl implements ProxyConfigRepositoryCus
         Update update = new Update();
         update.set("configFileId", fileId).set("timestamp", time);
         UpdateResult operation = mongoTemplate.upsert(basicQuery, update, ProxyConfig.class);
-        if (operation.getModifiedCount() < 1)
-            return false;
-        else
-            return true;
+
+        return operation.getModifiedCount() >= 1;
 
     }
 }

src/main/java/com/edgec/browserbackend/browser/task/Set3proxyTask.java

 package com.edgec.browserbackend.browser.task;
 
 import com.edgec.browserbackend.account.repository.AccountRepository;
+import com.edgec.browserbackend.auth.domain.mongo.MongoOAuth2AccessToken;
 import com.edgec.browserbackend.auth.repository.mongo.MongoOAuth2AccessTokenRepository;
 import com.edgec.browserbackend.browser.domain.IpResource;
 import com.edgec.browserbackend.browser.domain.IpType;
-import com.edgec.browserbackend.browser.domain.ProxyConfig;
 import com.edgec.browserbackend.browser.repository.IpResourceRepository;
 import com.edgec.browserbackend.browser.repository.ProxyConfigRepository;
 import com.edgec.browserbackend.browser.repository.SpecialLineRepository;
@@ -19,8 +19,10 @@ import org.springframework.stereotype.Component;
 import java.io.BufferedWriter;
 import java.io.File;
 import java.io.FileWriter;
+import java.io.IOException;
 import java.time.Instant;
 import java.util.List;
+import java.util.stream.Collectors;
 
 @Component
 public class Set3proxyTask {
@@ -42,22 +44,21 @@ public class Set3proxyTask {
     @Autowired
     private ProxyConfigRepository proxyConfigRepository;
 
-    @Scheduled(cron = "0 0/10 * * * ?")
+    @Scheduled(cron = "0 0/2 * * * ?")
     @SchedulerLock(name = "proxyTask", lockAtLeastFor = 60 * 1000 * 5, lockAtMostFor = 60 * 1000 * 9)
     public void set3proxy() {
-        long nowtime = Instant.now().toEpochMilli();
-//        List<String> tokenUsernames = mongoOAuth2AccessTokenRepository.findByCreatedAtGreaterThan(validTime).stream().map(MongoOAuth2AccessToken::getUsername).collect(Collectors.toList());
-//        List<String> accountParents = accountRepository.findByNameIn(tokenUsernames).stream().map(x -> x.getParent() == null ? x.getName() : x.getParent()).distinct().collect(Collectors.toList());
-//        List<IpResource> ipResources = ipResourceRepository.findByOwnerInAndSpecialLine(accountParents, true);
-        List<IpResource> ipResources = ipResourceRepository.findBySpecialLineAndIsDeleted(true, false);
+        long validTime = Instant.now().minusSeconds(43200).toEpochMilli();
+
+        List<String> tokenUsernames = mongoOAuth2AccessTokenRepository.findByCreatedAtGreaterThan(validTime).stream().map(MongoOAuth2AccessToken::getUsername).collect(Collectors.toList());
+        List<String> accountParents = accountRepository.findByNameIn(tokenUsernames).stream().map(x -> x.getParent() == null ? x.getName() : x.getParent()).distinct().collect(Collectors.toList());
+        List<IpResource> ipResources = ipResourceRepository.findByOwnerInAndSpecialLine(accountParents, true);
+
+        // List<IpResource> ipResources = ipResourceRepository.findBySpecialLineAndIsDeleted(true, false);
         File file = new File("3proxy_" + Instant.now().toEpochMilli());
         try {
-            try {
-                file.delete();
-                file.createNewFile();
-            } catch (Exception t1) {
-                log.error(t1.getMessage(), t1);
-            }
+
+            file.delete();
+            file.createNewFile();
 
             FileWriter fileWriter = new FileWriter(file, true);
             BufferedWriter bw = new BufferedWriter(fileWriter);
@@ -66,53 +67,53 @@ public class Set3proxyTask {
             bw.write("config /root/3proxy.cfg\n");
             bw.write("monitor /root/3proxy.cfg\n");
 
-            for (IpResource ipResource : ipResources) {
-                if (StringUtils.isNotBlank(ipResource.getAddr()))
-                    bw.write("users \"" + ipResource.getProxyUsername() + ":CL:" + ipResource.getProxyPassword() + "\"\n");
-                ipResource.setUsingSpecialLine(true);
-            }
+            // 写入 ipResources 相关信息
+            bwWriteIpResources(ipResources, bw);
 
             bw.write("\nauth strong\n");
-            a:
-            for (IpResource ipResource : ipResources) {
-                if (StringUtils.isNotBlank(ipResource.getAddr())) {
-                    if (ipResource.getPort() == null || ipResource.getPort().size() == 0)
-                        continue a;
-
-                    bw.write("allow " + ipResource.getProxyUsername() + "\n");
-                    if (ipResource.getIpType() != IpType.OWN) {
-                        bw.write("parent 1000 http " + ipResource.getAddr() + " " +
-                                (ipResource.getPort().size() > 1 ? ipResource.getPort().get(1) : ipResource.getPort().get(0)) + " fangguanlianbrowser " + ipResource.getPassword() + "\n");
-                    } else {
-                        bw.write("parent 1000 http " + ipResource.getAddr() + " " +
-                                (ipResource.getPort().size() > 1 ? ipResource.getPort().get(1) : ipResource.getPort().get(0)) + " " + ipResource.getUsername() + " " + ipResource.getPassword() + "\n");
-
-                    }
-                }
-            }
             bw.write("\nallow none\nproxy -p20004\nsocks -p20005\n");
             bw.flush();
             bw.close();
-            log.error("成功写入文件");
+            log.info("成功写入文件");
 
-            if (proxyConfigRepository.count() > 0) {
-                ProxyConfig proxyConfig = proxyConfigRepository.findAll().get(0);
-                if (nowtime < proxyConfig.getTimestamp()) {
-                    file.delete();
-                    return;
-                } else {
-                    proxyConfigRepository.updateProxy(file, nowtime);
-                    file.delete();
-                }
+            long nowtime = Instant.now().toEpochMilli();
+            long proxyConfigTimestamp = proxyConfigRepository.findAll().get(0).getTimestamp();
+            if (proxyConfigRepository.count() > 0 && nowtime < proxyConfigTimestamp) {
+                file.delete();
             } else {
                 proxyConfigRepository.updateProxy(file, nowtime);
                 file.delete();
             }
 
         } catch (Exception e) {
-            log.error("出错了");
             log.error(e.getMessage(), e);
         }
     }
 
+    private void bwWriteIpResources(List<IpResource> ipResources, BufferedWriter bw) throws IOException {
+        for (IpResource ipResource : ipResources) {
+            ipResource.setUsingSpecialLine(true);
+            if (StringUtils.isNotBlank(ipResource.getAddr())) {
+
+                bw.write("users \"" + ipResource.getProxyUsername() + ":CL:" + ipResource.getProxyPassword() + "\"\n");
+
+                if (ipResource.getPort() == null || ipResource.getPort().size() == 0) {
+                    continue;
+                }
+
+                bw.write("allow " + ipResource.getProxyUsername() + "\n");
+
+                String textStart = "parent 1000 http " + ipResource.getAddr() + " ";
+                String textMiddle = ipResource.getPort().size() > 1 ? ipResource.getPort().get(1) : ipResource.getPort().get(0);
+                if (ipResource.getIpType() != IpType.OWN) {
+                    bw.write(textStart + textMiddle
+                            + " fangguanlianbrowser " + ipResource.getPassword() + "\n");
+                } else {
+                    bw.write(textStart + textMiddle
+                            + " " + ipResource.getUsername() + " " + ipResource.getPassword() + "\n");
+                }
+            }
+        }
+    }
+
 }