no message

张道锋
1 parent 762b1b1d
Showing 8 changed files with 59 additions and 36 deletions Show diff stats
.idea/modules.xml
app/libs/processor.jar
app/src/main/assets/migrations/2.sql
app/src/main/assets/migrations/3.sql
app/src/main/java/com/shunzhi/parent/annotation/VersionTest.java
processor/src/main/java/org/shunzhi/processorlibs/AddColumn.java
processor/src/main/java/org/shunzhi/processorlibs/VersionProcessor.java
sql.txt
@@ -4,8 +4,8 @@
     <modules>
       <module fileurl="file://$PROJECT_DIR$/app/app.iml" filepath="$PROJECT_DIR$/app/app.iml" />
       <module fileurl="file://$PROJECT_DIR$/mvpsdk/mvpsdk.iml" filepath="$PROJECT_DIR$/mvpsdk/mvpsdk.iml" />
+      <module fileurl="file://$PROJECT_DIR$/parentWorkHolper.iml" filepath="$PROJECT_DIR$/parentWorkHolper.iml" />
       <module fileurl="file://E:\parentwork\parentWorkHolper.iml" filepath="E:\parentwork\parentWorkHolper.iml" />
-      <module fileurl="file://$PROJECT_DIR$/parentwork.iml" filepath="$PROJECT_DIR$/parentwork.iml" />
       <module fileurl="file://F:\parentWorkHolper\parentwork.iml" filepath="F:\parentWorkHolper\parentwork.iml" />
       <module fileurl="file://$PROJECT_DIR$/processor/processor.iml" filepath="$PROJECT_DIR$/processor/processor.iml" />
       <module fileurl="file://$PROJECT_DIR$/roundedimageview-2.2.1/roundedimageview-2.2.1.iml" filepath="$PROJECT_DIR$/roundedimageview-2.2.1/roundedimageview-2.2.1.iml" />
@@ -1 +0,0 @@
-ALTER table test INSERT ADD COLUMN age INTEGER
@@ -1 +0,0 @@
-ALTER table test INSERT ADD COLUMN name TEXT
@@ -2,16 +2,13 @@ package com.shunzhi.parent.annotation;
 import org.shunzhi.processorlibs.AddColumn;
-/**
- * Created by 10501 on 2018/3/28.
- */
 public class VersionTest {
-    @AddColumn(version = 3,tableName = "test",type = "String",isDebug = false)
+    @AddColumn(version = 3, tableName = "test", type = "String", isDebug = true)
     private String name;
-    @AddColumn(version = 3,tableName = "test",type = "Integer",isDebug = false)
+    @AddColumn(version = 3, tableName = "test", type = "Integer", isDebug = true)
     Integer age;
-
-    String area;
+    @AddColumn(version = 3, tableName = "test", type = "Boolean", isDebug = true)
+    Boolean area;
 }
@@ -16,4 +16,9 @@ public @interface AddColumn {
     String type();
     boolean isDebug() default true;
+
+    String type_string = "string";
+    String type_boolean = "boolean";
+    String type_integer = "integer";
+
 }
 package org.shunzhi.processorlibs;
-import org.greenrobot.greendao.annotation.Entity;
-
 import java.io.BufferedWriter;
 import java.io.File;
-import java.io.FileOutputStream;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.io.OutputStreamWriter;
 import java.io.PrintStream;
-import java.util.ArrayList;
 import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.HashSet;
 import java.util.Set;
 import javax.annotation.processing.AbstractProcessor;
@@ -26,19 +21,27 @@ import javax.lang.model.element.TypeElement;
 @SupportedAnnotationTypes({"org.shunzhi.processorlibs.AddColumn"})
 @SupportedSourceVersion(SourceVersion.RELEASE_7)
 public class VersionProcessor extends AbstractProcessor {
-    private final String sql_insert = "ALTER table %s INSERT ADD COLUMN %s %s";
-    private final String sql_update = "UPDATE table message SET %s = '%s' WHERE %s = '%s'";
+    //private final String sql_update = "UPDATE table message SET %s = '%s' WHERE %s = '%s'";
-    String path = "app\src\main\assets\migrations".replace("\", File.separator);
+    //String path = "app\src\main\assets\migrations".replace("\", File.separator);
     @Override
     public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
 //        File file = new File("test");
 //        if (!file.exists())file.mkdir();
-        int i = 0;
+//        File countFile = new File("测试数据.txt");
+//        try {
+//            BufferedWriter writer = new BufferedWriter(new FileWriter(countFile));
+//            writer.write(System.currentTimeMillis() + "");
+//            writer.newLine();
+//            writer.flush();writer.close();
+//        } catch (IOException e) {
+//            e.printStackTrace();
+//        }
+
         BufferedWriter testBw = null;
-        HashMap<Integer, List<String>> map = new HashMap<>();
+        HashMap<Integer, Set<String>> map = new HashMap<>();
         try {
             for (Element element : roundEnv.getElementsAnnotatedWith(AddColumn.class)) {
@@ -49,21 +52,32 @@ public class VersionProcessor extends AbstractProcessor {
                 int version = addColumn.version();
                 String tableName = addColumn.tableName();
                 String type = addColumn.type();
-                List<String> list = map.get(version);
+                Set<String> set = map.get(version);
                 boolean isDebug = addColumn.isDebug();
-                if (list == null) list = new ArrayList<>();
+                if (set == null) set = new HashSet<>();
                 String sql = "null";
+                String sql_insert = "ALTER table %s ADD COLUMN %s %s";
                 if (type.equalsIgnoreCase("String")) {
                     sql = String.format(sql_insert, tableName, element.getSimpleName().toString(), "TEXT");
-                    if (!isDebug) list.add(sql);
+                    if (!isDebug)
+                        set.add(sql);
                 } else if (type.equalsIgnoreCase("Integer")) {
                     sql = String.format(sql_insert, tableName, element.getSimpleName().toString(), "INTEGER");
-                    if (!isDebug) list.add(sql);
+                    if (!isDebug)
+                        set.add(sql);
+                } else if (type.equalsIgnoreCase("boolean")){
+                    sql = String.format(sql_insert, tableName, element.getSimpleName().toString(), "INTEGER");
+                    if (!isDebug)
+                        set.add(sql);
+                }
+                if (isDebug) {
+                    testBw.write(version + "   ");
+                    testBw.write(sql);
+                    testBw.newLine();
+                    testBw.flush();
+                } else {
+                    map.put(version, set);
                 }
-                testBw.write(version + "   ");
-                testBw.write(sql);
-                testBw.newLine();
-                testBw.flush();
             }
         } catch (Exception e) {
             e.printStackTrace();
@@ -74,18 +88,27 @@ public class VersionProcessor extends AbstractProcessor {
                 e.printStackTrace();
             }
         }
-
+        File tmp = new File("tmp");
+//        File migrations = new File(tmp.getParentFile(), "app\\src\\main\\assets\\migrations\\".replace("\\", File.separator));
+//        if (migrations.exists() || migrations.mkdirs()) {
+//            File[] files = migrations.listFiles();
+//            if (files != null)
+//                for (File f : files) {
+//                    if (!f.getName().equals("1.sql") && f.getName().endsWith(".sql")) f.delete();
+//                }
+//        }
         for (Integer version : map.keySet()) {
             try {
-                File tmp = new File("tmp");
                 String path = "app\\src\\main\\assets\\migrations\\".replace("\\", File.separator) + version + ".sql";
                 File file = new File(tmp.getParentFile(), path);
                 BufferedWriter bw = new BufferedWriter(new FileWriter(file));
+                StringBuilder builder = new StringBuilder();
                 for (String str : map.get(version)) {
-                    bw.write(str);
-                    bw.newLine();
-                    bw.flush();
+                    builder.append(str);
+                    builder.append("\r\n");
                 }
+                bw.write(builder.toString());
+                bw.flush();
                 bw.close();
             } catch (IOException e) {
                 e.printStackTrace();
@@ -93,4 +116,4 @@ public class VersionProcessor extends AbstractProcessor {
         }
         return true;
     }
 -}
+}
 \ No newline at end of file
-3   ALTER table test INSERT ADD COLUMN age INTEGER
+3   ALTER table test ADD COLUMN area INTEGER
	@@ -1 +0,0 @@		@@ -1 +0,0 @@
1	-ALTER table test INSERT ADD COLUMN age INTEGER
	@@ -1 +0,0 @@		@@ -1 +0,0 @@
1	-ALTER table test INSERT ADD COLUMN name TEXT
	@@ -16,4 +16,9 @@ public @interface AddColumn {		@@ -16,4 +16,9 @@ public @interface AddColumn {
16	String type();	16	String type();
17		17
18	boolean isDebug() default true;	18	boolean isDebug() default true;
		19	+
		20	+ String type_string = "string";
		21	+ String type_boolean = "boolean";
		22	+ String type_integer = "integer";
		23	+
19	}	24	}