浏览代码

feat: powerjob-worker-agent support use custom protocl

tjq 2 年之前
父节点
当前提交
3f7d4328e9
共有 1 个文件被更改,包括 7 次插入2 次删除
  1. 7 2
      powerjob-worker-agent/src/main/java/tech/powerjob/agent/MainApplication.java

+ 7 - 2
powerjob-worker-agent/src/main/java/tech/powerjob/agent/MainApplication.java

@@ -1,6 +1,7 @@
 package tech.powerjob.agent;
 package tech.powerjob.agent;
 
 
 import tech.powerjob.common.RemoteConstant;
 import tech.powerjob.common.RemoteConstant;
+import tech.powerjob.common.enums.Protocol;
 import tech.powerjob.worker.PowerJobWorker;
 import tech.powerjob.worker.PowerJobWorker;
 import tech.powerjob.worker.common.PowerJobWorkerConfig;
 import tech.powerjob.worker.common.PowerJobWorkerConfig;
 import tech.powerjob.worker.common.constants.StoreStrategy;
 import tech.powerjob.worker.common.constants.StoreStrategy;
@@ -18,15 +19,18 @@ import picocli.CommandLine.Option;
  * @since 2020/5/20
  * @since 2020/5/20
  */
  */
 @Slf4j
 @Slf4j
-@Command(name = "PowerJobAgent", mixinStandardHelpOptions = true, version = "4.0.0", description = "powerjob-worker agent")
+@Command(name = "PowerJobAgent", mixinStandardHelpOptions = true, version = "4.3.0", description = "powerjob-worker agent")
 public class MainApplication implements Runnable {
 public class MainApplication implements Runnable {
 
 
     @Option(names = {"-a", "--app"}, description = "worker-agent's name", required = true)
     @Option(names = {"-a", "--app"}, description = "worker-agent's name", required = true)
     private String appName;
     private String appName;
 
 
-    @Option(names = {"-p", "--port"}, description = "akka ActorSystem working port, not recommended to change")
+    @Option(names = {"-p", "--port"}, description = "transporter working port, not recommended to change")
     private Integer port = RemoteConstant.DEFAULT_WORKER_PORT;
     private Integer port = RemoteConstant.DEFAULT_WORKER_PORT;
 
 
+    @Option(names = {"-o", "--protocol"}, description = "transporter protocol, AKKA or HTTP")
+    private String protocol = Protocol.AKKA.name();
+
     @Option(names = {"-e", "--persistence"}, description = "storage strategy, DISK or MEMORY")
     @Option(names = {"-e", "--persistence"}, description = "storage strategy, DISK or MEMORY")
     private String storeStrategy = "DISK";
     private String storeStrategy = "DISK";
 
 
@@ -56,6 +60,7 @@ public class MainApplication implements Runnable {
             cfg.setStoreStrategy(StoreStrategy.MEMORY.name().equals(storeStrategy) ? StoreStrategy.MEMORY : StoreStrategy.DISK);
             cfg.setStoreStrategy(StoreStrategy.MEMORY.name().equals(storeStrategy) ? StoreStrategy.MEMORY : StoreStrategy.DISK);
             cfg.setMaxResultLength(length);
             cfg.setMaxResultLength(length);
             cfg.setTag(tag);
             cfg.setTag(tag);
+            cfg.setProtocol(Protocol.of(protocol));
 
 
             PowerJobWorker worker = new PowerJobWorker(cfg);
             PowerJobWorker worker = new PowerJobWorker(cfg);