add DynamicScript

Author: xlorne

pom.xml

@@ -35,11 +35,13 @@
         <jsonwebtoken.jjwt.version>0.12.3</jsonwebtoken.jjwt.version>
         <commons-io.version>2.15.0</commons-io.version>
         <commons-dbutils.version>1.8.1</commons-dbutils.version>
+        <commons-text.version>1.11.0</commons-text.version>
         <org.reflections.version>0.10.2</org.reflections.version>
         <perf4j.version>0.9.16</perf4j.version>
         <bcprov-jdk15on.version>1.70</bcprov-jdk15on.version>
         <commons-crypto.version>1.2.0</commons-crypto.version>
         <snakeyaml.version>2.2</snakeyaml.version>
+        <apache-groovy.version>4.0.15</apache-groovy.version>
     </properties>
 
     <dependencies>
@@ -157,6 +159,30 @@
                 <version>${commons-dbutils.version}</version>
             </dependency>
 
+            <dependency>
+                <groupId>org.apache.commons</groupId>
+                <artifactId>commons-text</artifactId>
+                <version>${commons-text.version}</version>
+            </dependency>
+
+            <dependency>
+                <groupId>org.apache.groovy</groupId>
+                <artifactId>groovy</artifactId>
+                <version>${apache-groovy.version}</version>
+            </dependency>
+
+            <dependency>
+                <groupId>org.apache.groovy</groupId>
+                <artifactId>groovy-json</artifactId>
+                <version>${apache-groovy.version}</version>
+            </dependency>
+
+            <dependency>
+                <groupId>org.apache.groovy</groupId>
+                <artifactId>groovy-xml</artifactId>
+                <version>${apache-groovy.version}</version>
+            </dependency>
+
         </dependencies>
     </dependencyManagement>
 

springboot-starter-data-fast/pom.xml

@@ -42,7 +42,21 @@
         <dependency>
             <groupId>org.apache.commons</groupId>
             <artifactId>commons-text</artifactId>
-            <version>1.11.0</version>
+        </dependency>
+
+        <dependency>
+            <groupId>org.apache.groovy</groupId>
+            <artifactId>groovy</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.apache.groovy</groupId>
+            <artifactId>groovy-json</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.apache.groovy</groupId>
+            <artifactId>groovy-xml</artifactId>
         </dependency>
 
     </dependencies>

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/DataFastConfiguration.java

@@ -4,6 +4,7 @@
 import com.codingapi.springboot.fast.executor.JpaExecutor;
 import com.codingapi.springboot.fast.manager.EntityManagerInitializer;
 import com.codingapi.springboot.fast.mapping.DynamicMappingRegister;
+import com.codingapi.springboot.fast.mapping.DynamicScriptRegister;
 import com.codingapi.springboot.fast.mapping.MvcEndpointMapping;
 import com.codingapi.springboot.fast.registrar.MvcMappingRegistrar;
 import org.springframework.aop.Advisor;
@@ -35,6 +36,12 @@ public DynamicMappingRegister dynamicMapping(MvcEndpointMapping mvcEndpointMappi
         return new DynamicMappingRegister(mvcEndpointMapping, dynamicQuery, jdbcTemplate);
     }
 
+    @Bean
+    @ConditionalOnMissingBean
+    public DynamicScriptRegister dynamicScriptRegister(MvcEndpointMapping mvcEndpointMapping, DynamicQuery dynamicQuery, JdbcTemplate jdbcTemplate) {
+        return new DynamicScriptRegister(mvcEndpointMapping, dynamicQuery, jdbcTemplate);
+    }
+
 
     @Bean(initMethod = "registerMvcMapping")
     @ConditionalOnMissingBean

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/mapping/DynamicMappingRegister.java

@@ -48,6 +48,27 @@ private void addHqlMapping(String mapping, RequestMethod requestMethod,String hq
         mvcEndpointMapping.addMapping(mapping, requestMethod, handler, method);
     }
 
+    /**
+     * test dynamic mapping
+     * @param dynamicMapping dynamic mapping
+     * @return result
+     */
+    @ResponseBody
+    public Object test(DynamicMapping dynamicMapping)  {
+        return switch (dynamicMapping.getType()) {
+            case JDBC ->
+                    new JdbcMapping(jdbcTemplate, dynamicMapping.getSql(), dynamicMapping.getClazz(), dynamicMapping.getParams()).execute();
+            case HQL ->
+                    new HqlMapping(dynamicQuery, dynamicMapping.getSql(), dynamicMapping.getClazz(), dynamicMapping.getParams()).execute();
+            case JDBC_MAP ->
+                    new JdbcMapMapping(jdbcTemplate, dynamicMapping.getSql(), dynamicMapping.getParams()).execute();
+        };
+    }
+
+    /**
+     * add dynamic mapping
+     * @param dynamicMapping dynamic mapping
+     */
     public void addMapping(DynamicMapping dynamicMapping)  {
         switch (dynamicMapping.getType()) {
             case JDBC:
@@ -64,6 +85,14 @@ public void addMapping(DynamicMapping dynamicMapping)  {
         }
     }
 
+    /**
+     * remove mapping
+     * @param mapping mapping
+     * @param requestMethod request method
+     */
+    public void removeMapping(String mapping,RequestMethod requestMethod)  {
+        mvcEndpointMapping.removeMapping(mapping,requestMethod);
+    }
 
 
     @SneakyThrows

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/mapping/DynamicScriptRegister.java

@@ -0,0 +1,49 @@
+package com.codingapi.springboot.fast.mapping;
+
+import com.codingapi.springboot.fast.dynamic.DynamicQuery;
+import com.codingapi.springboot.fast.script.DynamicScript;
+import com.codingapi.springboot.fast.script.ScriptContext;
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.ResponseBody;
+
+public class DynamicScriptRegister {
+
+    private final MvcEndpointMapping mvcEndpointMapping;
+
+
+    public DynamicScriptRegister(MvcEndpointMapping mvcEndpointMapping, DynamicQuery dynamicQuery, JdbcTemplate jdbcTemplate) {
+        this.mvcEndpointMapping = mvcEndpointMapping;
+        ScriptContext.getInstance().init(dynamicQuery, jdbcTemplate);
+    }
+
+
+    /**
+     * test dynamic mapping
+     * @param dynamicScript dynamic mapping
+     **/
+    public void addMapping(DynamicScript dynamicScript) {
+        mvcEndpointMapping.addMapping(dynamicScript.getMapping(), dynamicScript.getRequestMethod(),
+                dynamicScript, dynamicScript.getExecuteMethod());
+    }
+
+    /**
+     *  remove mapping
+     * @param mapping mapping
+     * @param requestMethod requestMethod
+     */
+    public void removeMapping(String mapping, RequestMethod requestMethod) {
+        mvcEndpointMapping.removeMapping(mapping, requestMethod);
+    }
+
+    /**
+     * test dynamic mapping
+     * @param dynamicScript dynamic mapping
+     * @return result
+     */
+    @ResponseBody
+    public Object test(DynamicScript dynamicScript) {
+        return dynamicScript.execute();
+    }
+
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/mapping/MvcEndpointMapping.java

@@ -37,4 +37,23 @@ public void addMapping(String url, RequestMethod requestMethod, Object handler,
         handlerMapping.registerMapping(mappingInfo, handler, method);
     }
 
+    /**
+     * remove mvc mapping
+     * @param url           mapping url
+     * @param requestMethod request method
+     */
+    public void removeMapping(String url, RequestMethod requestMethod) {
+        RequestMappingInfo.BuilderConfiguration options = new RequestMappingInfo.BuilderConfiguration();
+        options.setPatternParser(new PathPatternParser());
+
+        RequestMappingInfo mappingInfo = RequestMappingInfo
+                .paths(url)
+                .methods(requestMethod)
+                .produces(MediaType.APPLICATION_JSON_VALUE)
+                .options(options)
+                .build();
+
+        handlerMapping.unregisterMapping(mappingInfo);
+    }
+
 }

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/script/DynamicScript.java

@@ -0,0 +1,34 @@
+package com.codingapi.springboot.fast.script;
+
+import lombok.Getter;
+import lombok.Setter;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.ResponseBody;
+
+import java.lang.reflect.Method;
+
+@Setter
+@Getter
+public class DynamicScript {
+
+    private String script;
+    private String mapping;
+    private RequestMethod requestMethod;
+
+
+    @ResponseBody
+    public Object execute() {
+        ScriptContext context = ScriptContext.getInstance();
+        return ScriptRuntime.running(script,context);
+    }
+
+    public Method getExecuteMethod() {
+        try {
+            return this.getClass().getDeclaredMethod("execute");
+        } catch (NoSuchMethodException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/script/ScriptContext.java

@@ -0,0 +1,29 @@
+package com.codingapi.springboot.fast.script;
+
+import com.codingapi.springboot.fast.dynamic.DynamicQuery;
+import lombok.Getter;
+import org.springframework.jdbc.core.JdbcTemplate;
+
+public class ScriptContext {
+
+    @Getter
+    private final static ScriptContext instance = new ScriptContext();
+
+
+    private ScriptContext() {
+
+    }
+
+
+    @Getter
+    private DynamicQuery dynamicQuery;
+    @Getter
+    private JdbcTemplate jdbcTemplate;
+
+
+    public void init(DynamicQuery dynamicQuery, JdbcTemplate jdbcTemplate) {
+        this.dynamicQuery = dynamicQuery;
+        this.jdbcTemplate = jdbcTemplate;
+    }
+
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/script/ScriptRuntime.java

@@ -0,0 +1,18 @@
+package com.codingapi.springboot.fast.script;
+
+import groovy.lang.Binding;
+import groovy.lang.GroovyShell;
+import groovy.lang.Script;
+
+
+public class ScriptRuntime {
+
+    public static Object running(String script, ScriptContext context) {
+        Binding binding = new Binding();
+        binding.setVariable("$hql", context.getDynamicQuery());
+        binding.setVariable("$jdbc", context.getJdbcTemplate());
+        GroovyShell groovyShell = new GroovyShell(binding);
+        Script userScript = groovyShell.parse(script);
+        return userScript.run();
+    }
+}

springboot-starter-data-fast/src/test/java/com/codingapi/springboot/fast/script/ScriptRuntimeTest.java

@@ -0,0 +1,16 @@
+package com.codingapi.springboot.fast.script;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.boot.test.context.SpringBootTest;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+@SpringBootTest
+class ScriptRuntimeTest {
+
+    @Test
+    void running() {
+        Object res = ScriptRuntime.running("return 1",ScriptContext.getInstance());
+        assertEquals(1,res);
+    }
+}