scheme2ddl-28 filter-sequence-values option

Author: qwazer

src/main/java/com/googlecode/scheme2ddl/DDLFormatter.java

@@ -1,5 +1,8 @@
 package com.googlecode.scheme2ddl;
 
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
 /**
  * @author A_Reshetnikov
  * @since Date: 18.10.2012
@@ -41,4 +44,21 @@ public void setStatementOnNewLine(Boolean statementOnNewLine) {
     public void setIsMorePrettyFormat(boolean isMorePrettyFormat) {
         this.isMorePrettyFormat = isMorePrettyFormat;
     }
+
+
+    public String replaceActualSequenceValueWithOne(String res) {
+
+        String output;
+        Pattern p = Pattern.compile("CREATE SEQUENCE (.*) START WITH (\\d+) (.*)");
+        Matcher m = p.matcher(res);
+        if (m.find()) {
+             output = m.replaceFirst("CREATE SEQUENCE " + m.group(1) + " START WITH 1 " + m.group(3) );
+            if (!"1".equals(m.group(2)))
+             output = output + newline + "/* -- actual sequence value was replaced by scheme2ddl to 1 */";
+        }
+        else {
+            output = res;
+        }
+        return output;
+    }
 }

src/main/java/com/googlecode/scheme2ddl/Main.java

@@ -33,6 +33,7 @@ public class Main {
     private static boolean justTestConnection = false;
     private static boolean skipPublicDbLinks = false;
     private static boolean stopOnWarning = false;
+    private static boolean filterSequenceValues = false;
     private static String customConfigLocation = null;
     private static String defaultConfigLocation = "scheme2ddl.config.xml";
     private static String dbUrl = null;
@@ -120,6 +121,10 @@ private static void modifyContext(ConfigurableApplicationContext context) {
             UserObjectProcessor processor = (UserObjectProcessor) context.getBean("processor");
             processor.setStopOnWarning(stopOnWarning);
         }
+        if (filterSequenceValues){
+            UserObjectProcessor processor = (UserObjectProcessor) context.getBean("processor");
+            processor.setFilterSequenceValues(filterSequenceValues);
+        }
 
     }
 
@@ -297,8 +302,10 @@ private static void collectArgs(String[] args) throws Exception {
                 i++;
             } else if (arg.equals("-tc") || arg.equals("--test-connection")) {
                 justTestConnection = true;
-            }else if (arg.equals("--stop-on-warning")) {
+            } else if (arg.equals("--stop-on-warning")) {
                 stopOnWarning = true;
+            }   else if ((arg.equals("-fsv") || arg.equals("--filter-sequence-values"))) {
+                filterSequenceValues = true;
             } else if (arg.equals("-c") || arg.equals("--config")) {
                 customConfigLocation = args[i + 1];
                 i++;

src/main/java/com/googlecode/scheme2ddl/UserObjectProcessor.java

@@ -26,6 +26,7 @@ public class UserObjectProcessor implements ItemProcessor<UserObject, UserObject
     private Map<String, Set<String>> excludes;
     private Map<String, Set<String>> dependencies;
     private boolean stopOnWarning;
+    private boolean filterSequenceValues;
 
     public UserObject process(UserObject userObject) throws Exception {
 
@@ -75,6 +76,9 @@ private String map2Ddl(UserObject userObject) throws CannotGetDDLException, NonS
                 return userObjectDao.findRefGroupDDL(userObject.getType(), userObject.getName());
             }
             String res = userObjectDao.findPrimaryDDL(map2TypeForDBMS(userObject.getType()), userObject.getName());
+            if (userObject.getType().equals("SEQUENCE") && filterSequenceValues) {
+                res = ddlFormatter.replaceActualSequenceValueWithOne(res);
+            }
             Set<String> dependedTypes = dependencies.get(userObject.getType());
             if (dependedTypes != null) {
                 for (String dependedType : dependedTypes) {
@@ -112,6 +116,10 @@ public void setFileNameConstructor(FileNameConstructor fileNameConstructor) {
         this.fileNameConstructor = fileNameConstructor;
     }
 
+    public void setFilterSequenceValues(boolean filterSequenceValues) {
+        this.filterSequenceValues = filterSequenceValues;
+    }
+
     public void setStopOnWarning(boolean stopOnWarning) {
         this.stopOnWarning = stopOnWarning;
     }

src/main/resources/applicationContext.xml

@@ -9,7 +9,7 @@
 		http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-3.0.xsd">
 
     <context:annotation-config/>
-    <aop:aspectj-autoproxy/>
+    <aop:aspectj-autoproxy proxy-target-class="true"/>
 
     <job id="job1" xmlns="http://www.springframework.org/schema/batch">
         <step id="step1" parent="simpleStep">
@@ -51,6 +51,7 @@
         <property name="excludes" ref="excludes"/>
         <property name="dependencies" ref="dependencies"/>
         <property name="stopOnWarning" value="false"/>
+        <property name="filterSequenceValues" value="false"/>
     </bean>
 
 

src/test/java/com/googlecode/scheme2ddl/DDLFormatterTest.java

@@ -0,0 +1,36 @@
+package com.googlecode.scheme2ddl;
+
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+
+/**
+ * @author ar
+ * @since Date: 18.04.2015
+ */
+public class DDLFormatterTest {
+
+    private DDLFormatter ddlFormatter = new DDLFormatter();
+
+    @Test
+    public void testReplaceActualSequenceValueWithOne() throws Exception {
+
+        String s = "CREATE SEQUENCE  \"TEST01\".\"SEQ_01\"  MINVALUE 1 MAXVALUE 9999999999999999999999999999 INCREMENT BY 1 START WITH 2122 CACHE 20 NOORDER  NOCYCLE ;\n";
+        String res = ddlFormatter.replaceActualSequenceValueWithOne(s);
+        assertEquals(
+                "CREATE SEQUENCE  \"TEST01\".\"SEQ_01\"  MINVALUE 1 MAXVALUE 9999999999999999999999999999 INCREMENT BY 1 START WITH 1 CACHE 20 NOORDER  NOCYCLE ;\n"
+                , res);
+
+    }
+
+    @Test
+    public void testReplaceActualSequenceValueWithOneOnWrongDDL() throws Exception {
+
+        String s = "CREATE TABLE  \"TEST01\".\"SEQ_01\"  MINVALUE 1 MAXVALUE 9999999999999999999999999999 INCREMENT BY 1 START WITH 2122 CACHE 20 NOORDER  NOCYCLE ;\n";
+        String res = ddlFormatter.replaceActualSequenceValueWithOne(s);
+        assertNotEquals(
+                "CREATE TABLE  \"TEST01\".\"SEQ_01\"  MINVALUE 1 MAXVALUE 9999999999999999999999999999 INCREMENT BY 1 START WITH 1 CACHE 20 NOORDER  NOCYCLE ;\n"
+                , res);
+
+    }
+}