新增应用安装类，自动检测数据库，自动创建数据库表

hsweb-web-core/pom.xml

@@ -21,7 +21,10 @@
             <artifactId>spring-boot-starter-web</artifactId>
             <optional>true</optional>
         </dependency>
-
+        <dependency>
+            <groupId>org.webbuilder</groupId>
+            <artifactId>wb-sql-util</artifactId>
+        </dependency>
     </dependencies>
 
 </project>

hsweb-web-core/src/main/java/org/hsweb/web/Install.java

@@ -0,0 +1,94 @@
+package org.hsweb.web;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.AutoConfigureAfter;
+import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
+import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.util.Assert;
+import org.webbuilder.sql.param.SqlAppender;
+import org.webbuilder.sql.support.common.CommonSql;
+import org.webbuilder.sql.support.executor.SqlExecutor;
+import org.webbuilder.utils.file.FileUtils;
+
+import javax.annotation.PostConstruct;
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.Reader;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Created by zhouhao on 16-4-23.
+ */
+@Configuration
+@EnableConfigurationProperties({DataSourceProperties.class})
+@AutoConfigureAfter({DataSourceAutoConfiguration.class})
+@ConfigurationProperties(
+        prefix = "spring.datasource"
+)
+public class Install {
+    private static String DATABASE_TYPE = "h2";
+
+    /**
+     * 获取当前数据库类型
+     *
+     * @return
+     */
+    public static String getDatabaseType() {
+        return DATABASE_TYPE;
+    }
+
+    @Autowired
+    private DataSourceProperties properties;
+
+    @Autowired
+    private SqlExecutor sqlExecutor;
+
+    private Logger logger = LoggerFactory.getLogger(this.getClass());
+
+    @PostConstruct
+    public void install() {
+        String dc = properties.getDriverClassName();
+        String dbType = dc.contains("mysql") ? "mysql" : dc.contains("oracle") ? "oracle" : dc.contains("h2") ? "h2" : null;
+        DATABASE_TYPE = dbType;
+        Assert.notNull(dbType, "不支持的数据库类型");
+        try {
+            boolean firstInstall = false;
+            try {
+                sqlExecutor.exec(new CommonSql("select * from s_user where 1=2"));
+            } catch (Exception e) {
+                firstInstall = true;
+            }
+            if (firstInstall) {
+                Reader reader = FileUtils.getResourceAsReader("system/install/sql/" + dbType + "/install.sql");
+                BufferedReader bufferedReader = new BufferedReader(reader);
+                List<String> sqlList = new ArrayList<>();
+                SqlAppender tmp = new SqlAppender();
+                bufferedReader.lines().forEach((line) -> {
+                    if (line.startsWith("--")) return;
+                    tmp.add(line);
+                    tmp.add("\n");
+                    if (line.endsWith(";")) {
+                        sqlList.add(tmp.toString());
+                        tmp.clear();
+                    }
+                });
+                sqlList.forEach((sql) -> {
+                    try {
+                        sqlExecutor.exec(new CommonSql(sql));
+                    } catch (Exception e) {
+                        logger.warn("install sql fail", e.getMessage());
+                    }
+                });
+            }
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+}