Browse Source

feat: swap

Wren 1 year ago
parent
commit
a0c7e2091c

+ 2 - 1
src/main/java/com/ichaoj/ams/controller/AirdropTaskController.java

@@ -8,6 +8,7 @@ import com.ichaoj.ams.request.task.PageTaskRequest;
 import com.ichaoj.ams.request.task.UpdateAirdropTask;
 import com.ichaoj.ams.request.task.UpdateTaskStatusRequest;
 import com.ichaoj.ams.response.task.TaskResponse;
+import com.ichaoj.ams.script.IScript;
 import com.ichaoj.ams.script.JavaScript;
 import com.ichaoj.ams.script.ScriptContext;
 import com.ichaoj.ams.script.annotation.Script;
@@ -101,7 +102,7 @@ public class AirdropTaskController {
     @AuthResource
     @Operation(summary = "查询任务参数")
     public PublicResult<List<AirdropParam>> getAirdropParam(@PathVariable("task-code") String taskCode) {
-        JavaScript script = ScriptContext.getScriptByCode(taskCode);
+        IScript script = ScriptContext.getScriptByCode(taskCode);
         Script annotation = script.getClass().getAnnotation(Script.class);
         ScriptParam[] params = annotation.params();
         if (params != null && params.length > 0) {

+ 3 - 3
src/main/java/com/ichaoj/ams/script/ScriptContext.java

@@ -13,9 +13,9 @@ import java.util.Optional;
 public class ScriptContext {
 
 
-    public static JavaScript getScriptByCode(String code) {
-        Map<String, JavaScript> beansOfType = SpringUtil.getBeansOfType(JavaScript.class);
-        Optional<JavaScript> optional = beansOfType
+    public static IScript getScriptByCode(String code) {
+        Map<String, IScript> beansOfType = SpringUtil.getBeansOfType(IScript.class);
+        Optional<IScript> optional = beansOfType
                 .values()
                 .stream()
                 .filter(script -> {

+ 2 - 1
src/main/java/com/ichaoj/ams/service/impl/AmsExecuteRecordServiceImpl.java

@@ -15,6 +15,7 @@ import com.ichaoj.ams.request.execute.CreateExecute;
 import com.ichaoj.ams.request.execute.PageExecuteRequest;
 import com.ichaoj.ams.request.execute.UpdateExecute;
 import com.ichaoj.ams.response.execute.ExecuteResponse;
+import com.ichaoj.ams.script.IScript;
 import com.ichaoj.ams.script.JavaScript;
 import com.ichaoj.ams.script.model.SResult;
 import com.ichaoj.ams.script.ScriptContext;
@@ -156,7 +157,7 @@ public class AmsExecuteRecordServiceImpl extends SuperWhaleServiceImpl<AmsExecut
 
                     }
                     i++;
-                    JavaScript script = ScriptContext.getScriptByCode(taskCode);
+                    IScript script = ScriptContext.getScriptByCode(taskCode);
                     SResult sResult = script.run(airdropParamMap, wallet);
                     AmsTradeRecord tradeRecord = new AmsTradeRecord();
                     tradeRecord.setExecuteId(execId);