项目优化

mizhewei · Mar 4, 2018 · 4550d21 · 4550d21
1 parent 0115369
commit 4550d21
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/src/main/java/com/example/ShadowSocksShare/service/impl/Free_ssCrawlerServiceImpl.java b/src/main/java/com/example/ShadowSocksShare/service/impl/Free_ssCrawlerServiceImpl.java
@@ -92,7 +92,7 @@ public ShadowSocksEntity getShadowSocks() {
 				options.addArguments("--headless");
 				options.addArguments("--disable-gpu");
 				options.addArguments("--no-sandbox");
-				options.addArguments("--remote-debugging-port=9222");
+				// options.addArguments("--remote-debugging-port=9222");
 			}
 
 			if (ssProxyEnable) {

diff --git a/src/main/java/com/example/ShadowSocksShare/web/MainController.java b/src/main/java/com/example/ShadowSocksShare/web/MainController.java
@@ -4,11 +4,14 @@
 import com.example.ShadowSocksShare.domain.ShadowSocksDetailsEntity;
 import com.example.ShadowSocksShare.domain.ShadowSocksEntity;
 import com.example.ShadowSocksShare.service.CountSerivce;
+import com.example.ShadowSocksShare.service.ShadowSocksCrawlerService;
 import com.example.ShadowSocksShare.service.ShadowSocksSerivce;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.google.zxing.WriterException;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationContext;
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.domain.Sort;
 import org.springframework.data.web.PageableDefault;
@@ -34,6 +37,10 @@ public class MainController {
 	private ShadowSocksSerivce shadowSocksSerivceImpl;
 	@Autowired
 	private CountSerivce countSerivce;
+	@Autowired
+	private ShadowSocksSerivce shadowSocksSerivce;
+	@Autowired
+	private ApplicationContext appContext;
 
 	/**
 	 * 首页
@@ -101,4 +108,17 @@ public ResponseEntity<byte[]> createQRCode(long id, String text, int width, int
 	public ResponseEntity<String> count() {
 		return ResponseEntity.ok().body(String.valueOf(countSerivce.get()));
 	}
+
+	@RequestMapping(value = "/run")
+	@ResponseBody
+	public ResponseEntity<String> run(String name) {
+		if (StringUtils.isNotBlank(name)) {
+			Object bean = appContext.getBean(name);
+
+			if (bean instanceof ShadowSocksCrawlerService) {
+				shadowSocksSerivce.crawlerAndSave(ShadowSocksCrawlerService.class.cast(bean));
+			}
+		}
+		return ResponseEntity.ok().body("OK...");
+	}
 }