feat: add get all inner operations by reflactions (DataLinkDC#1477)

Signed-off-by: Licho <chinabhsun@gmail.com> Signed-off-by: Licho <chinabhsun@gmail.com>
aiwenmo · Dec 28, 2022 · 245f3b3 · 245f3b3
1 parent 87d5483
commit 245f3b3
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 10 deletions.
diff --git a/dlink-executor/pom.xml b/dlink-executor/pom.xml
@@ -58,6 +58,10 @@
             <artifactId>dlink-metadata-mysql</artifactId>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.reflections</groupId>
+            <artifactId>reflections</artifactId>
+        </dependency>
         <dependency>
             <groupId>org.apache.hadoop</groupId>
             <artifactId>hadoop-common</artifactId>

diff --git a/dlink-executor/src/main/java/com/dlink/trans/Operations.java b/dlink-executor/src/main/java/com/dlink/trans/Operations.java
@@ -20,30 +20,42 @@
 package com.dlink.trans;
 
 import com.dlink.parser.SqlType;
-import com.dlink.trans.ddl.AddJarOperation;
-import com.dlink.trans.ddl.CreateAggTableOperation;
-import com.dlink.trans.ddl.CreateCDCSourceOperation;
-import com.dlink.trans.ddl.SetOperation;
-import com.dlink.trans.ddl.ShowFragmentOperation;
-import com.dlink.trans.ddl.ShowFragmentsOperation;
+import lombok.extern.slf4j.Slf4j;
+import org.reflections.Reflections;
+import org.reflections.scanners.Scanners;
 
+import java.lang.reflect.InvocationTargetException;
 import java.util.Arrays;
+import java.util.Set;
+
 
 /**
  * Operations
  *
  * @author wenmo
  * @since 2021/5/25 15:50
  **/
+@Slf4j
 public class Operations {
 
     private Operations() {
     }
 
-    private static final Operation[] ALL_OPERATIONS = {new CreateAggTableOperation(), new SetOperation(),
-            new CreateCDCSourceOperation(), new ShowFragmentsOperation(), new ShowFragmentOperation(),
-            new AddJarOperation()
-    };
+    private static final Operation[] ALL_OPERATIONS = getAllOperations();
+
+    private static Operation[] getAllOperations() {
+        Reflections reflections = new Reflections(Operation.class.getPackage().getName());
+        Set<Class<?>> operations = reflections.get(Scanners.SubTypes.of(Operation.class).asClass());
+        return operations.stream().filter(t -> !t.isInterface()).map(t -> {
+            try {
+                return t.getConstructor().newInstance();
+            } catch (InstantiationException | IllegalAccessException | InvocationTargetException |
+                     NoSuchMethodException e) {
+                log.error(String.format("getAllOperations error, class %s, err: %s", t, e));
+                throw new RuntimeException(e);
+            }
+        }).toArray(Operation[]::new);
+    }
 
     public static SqlType getSqlTypeFromStatements(String statement) {
         String[] statements = statement.split(";");

diff --git a/pom.xml b/pom.xml
@@ -94,10 +94,16 @@
         <javax.annotation-api.version>1.3.2</javax.annotation-api.version>
         <log4j.version>2.19.0</log4j.version>
         <docker.java.version>3.2.13</docker.java.version>
+        <reflections.version>0.10.2</reflections.version>
     </properties>
 
     <dependencyManagement>
         <dependencies>
+            <dependency>
+                <groupId>org.reflections</groupId>
+                <artifactId>reflections</artifactId>
+                <version>${reflections.version}</version>
+            </dependency>
             <dependency>
                 <groupId>jakarta.ws.rs</groupId>
                 <artifactId>jakarta.ws.rs-api</artifactId>