mirror of
https://github.com/Hello-hao/Tbed.git
synced 2024-04-21 12:32:10 +00:00
167 lines
7.8 KiB
Java
167 lines
7.8 KiB
Java
package cn.hellohao.utils;
|
|
|
|
import cn.hellohao.config.GlobalConstant;
|
|
import org.springframework.beans.factory.InitializingBean;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.context.annotation.Configuration;
|
|
|
|
import java.io.File;
|
|
import java.util.Properties;
|
|
import java.util.UUID;
|
|
|
|
/**
|
|
* @author Hellohao
|
|
* @version 1.0
|
|
* @date 2019/11/29 14:32
|
|
*/
|
|
@Configuration
|
|
public class FirstRun implements InitializingBean {
|
|
|
|
@Value("${spring.datasource.username}")
|
|
private String jdbcusername;
|
|
|
|
@Value("${spring.datasource.password}")
|
|
private String jdbcpass;
|
|
|
|
@Value("${spring.datasource.url}")
|
|
private String jdbcurl;
|
|
|
|
@Override
|
|
public void afterPropertiesSet() {
|
|
isWindows();
|
|
RunSqlScript.USERNAME = jdbcusername;
|
|
RunSqlScript.PASSWORD = jdbcpass;
|
|
RunSqlScript.DBURL = jdbcurl;
|
|
|
|
Integer dataBaseName = RunSqlScript.RunSelectCount("SELECT count(*) FROM information_schema.SCHEMATA WHERE SCHEMA_NAME = 'tbed'");
|
|
if(dataBaseName<1){
|
|
Print.Normal("无法找到指定数据库，请先创建'tbed'数据库，并导入表结构后再试。");
|
|
System.exit(1);
|
|
}
|
|
|
|
Print.Normal("正在校验数据库参数...");
|
|
RunSqlScript.RunInsert(dynamic);
|
|
RunSqlScript.RunInsert(compressed);
|
|
|
|
String uid = UUID.randomUUID().toString().replace("-", "").toLowerCase();
|
|
RunSqlScript.RunInsert("update user set uid='"+uid+"' where id = 1");
|
|
|
|
Integer ret1 = RunSqlScript.RunSelectCount(sql1);
|
|
if(ret1==0){
|
|
Print.Normal("In execution...");
|
|
RunSqlScript.RunInsert(sql2);
|
|
}else{
|
|
if(ret1>0){
|
|
Print.Normal("Stage 1");
|
|
}else{
|
|
Print.Normal("Mysql 报了一个错");
|
|
}
|
|
}
|
|
Integer imgandalbum = RunSqlScript.RunSelectCount(judgeTable+"'imgandalbum'");
|
|
if(imgandalbum==0){
|
|
RunSqlScript.RunInsert(sql3);
|
|
Print.Normal("Stage 2");
|
|
}
|
|
Integer album = RunSqlScript.RunSelectCount(judgeTable+"'album'");
|
|
if(album==0){
|
|
RunSqlScript.RunInsert(sql4);
|
|
Print.Normal("Stage 3");
|
|
}
|
|
Integer ret2 = RunSqlScript.RunSelectCount(judgeTable+"'imgdata' and column_name = 'explains'");
|
|
if(ret2==0){
|
|
RunSqlScript.RunInsert(sql6);
|
|
Print.Normal("Stage 4");
|
|
}
|
|
Integer ret3 = RunSqlScript.RunSelectCount(judgeTable+"'album' and column_name = 'userid'");
|
|
if(ret3==0){
|
|
RunSqlScript.RunInsert(sql7);
|
|
Print.Normal("Stage 5");
|
|
}
|
|
Integer ret4 = RunSqlScript.RunSelectCount(judgeTable+"'imgdata' and column_name = 'md5key'");
|
|
if(ret4==0){
|
|
RunSqlScript.RunInsert(sql8);
|
|
Print.Normal("Stage 6");
|
|
}
|
|
Integer ret5 = RunSqlScript.RunSelectCount(judgeTable+"'config' and column_name = 'theme'");
|
|
if(ret5==0){
|
|
RunSqlScript.RunInsert(sql9);
|
|
Print.Normal("Stage 7");
|
|
}
|
|
Integer ret6 = RunSqlScript.RunSelectCount(judgeTable+" 'user' and column_name = 'token'");
|
|
if(ret6==0){
|
|
RunSqlScript.RunInsert(sql12);
|
|
Print.Normal("Add user.token");
|
|
}
|
|
Integer isappclient = RunSqlScript.RunSelectCount(isTableName+"'appclient'");
|
|
if(isappclient==0){
|
|
Integer integer = RunSqlScript.RunInsert(createAppclient);
|
|
RunSqlScript.RunInsert(instartAppclient);
|
|
Print.Normal("Add table.appclient");
|
|
}
|
|
Integer ret7 = RunSqlScript.RunSelectCount(judgeTable+" 'imgdata' and column_name = 'idname'");
|
|
if(ret7==0){
|
|
RunSqlScript.RunInsert(sql11);
|
|
Print.Normal("Add imgdata.idname");
|
|
}
|
|
|
|
RunSqlScript.RunInsert("alter table `config` modify column `explain` text,modify column `links` text,modify column `notice` text,modify column `baidu` text");
|
|
|
|
RunSqlScript.RunInsert(inddx_md5key);
|
|
RunSqlScript.RunInsert("UPDATE `keys` SET `Endpoint` = '0' WHERE `id` = 8");
|
|
Print.Normal("Stage success");
|
|
|
|
clears();
|
|
}
|
|
|
|
private String isTableName = "SELECT count(table_name) FROM information_schema.TABLES WHERE TABLE_SCHEMA='tbed' and table_name =";
|
|
private String judgeTable = "select count(*) from information_schema.columns where TABLE_SCHEMA='tbed' and table_name = ";
|
|
//创建blacklist 2019-11-29
|
|
private String sql1 = "select count(*) from information_schema.columns where table_name = 'uploadconfig' and column_name = 'blacklist'";
|
|
private String sql2 = "alter table uploadconfig add blacklist varchar(500);";
|
|
//创建imgandalbum和album 添加imgdata表字段explain 2019-12-20
|
|
private String sql3 ="CREATE TABLE `imgandalbum` (`imgname` varchar(5000) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL,`albumkey` varchar(5000) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL) ENGINE = InnoDB CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic";
|
|
private String sql4 ="CREATE TABLE `album` (`albumkey` varchar(5000) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL,`albumtitle` varchar(5000) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL,`createdate` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL, `password` varchar(5000) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL) ENGINE = InnoDB CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic";
|
|
private String sql6 = "alter table imgdata add explains varchar(5000)";
|
|
private String sql7 = "alter table album add userid int(10)";
|
|
private String sql8 = "alter table imgdata add md5key varchar(5000)";
|
|
private String sql9 = "ALTER TABLE config ADD theme int(4) DEFAULT '1' COMMENT '主题' ";
|
|
private String sql11 = "alter table `imgdata` add idname varchar(255) DEFAULT '未命名图像' ;";
|
|
private String sql12 = "alter table tbed.user add `token` varchar(255)";
|
|
private String createAppclient = "CREATE TABLE `appclient` (`id` varchar(10) NOT NULL,`isuse` varchar(10) NOT NULL,`winpackurl` varchar(255) NULL DEFAULT NULL,`macpackurl` varchar(255) NULL DEFAULT NULL,`appname` varchar(20) NULL,`applogo` varchar(255) NULL,`appupdate` varchar(10) NOT NULL) ";
|
|
private String instartAppclient = "INSERT INTO `appclient` VALUES ('app', 'on', NULL, NULL, 'Hellohao图像托管', 'https://hellohao.nos-eastchina1.126.net/TbedClient/app.png', '1.0.1');";
|
|
private String inddx_md5key = "ALTER TABLE imgdata ADD INDEX index_md5key_url ( md5key,imgurl)";
|
|
private String dynamic = "alter table imgdata row_format=dynamic";
|
|
private String compressed = "alter table imgdata row_format=compressed";
|
|
|
|
private void clears(){
|
|
File file1 = new File(GlobalConstant.LOCPATH+File.separator+"hellohaotempimg");
|
|
File file2 = new File(GlobalConstant.LOCPATH+File.separator+"hellohaotempwatermarimg");
|
|
|
|
//判断目录有没有创建
|
|
File file = new File(GlobalConstant.LOCPATH);
|
|
if(!file.exists()){
|
|
file.mkdirs();
|
|
file1.mkdirs();
|
|
file2.mkdirs();
|
|
}else{
|
|
if(!file1.exists()){
|
|
file1.mkdirs();
|
|
}else if(!file2.exists()){
|
|
file2.mkdirs();
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
public boolean isWindows() {
|
|
System.out.println("当前系统类型:"+System.getProperties().getProperty("os.name").toUpperCase());
|
|
if(System.getProperties().getProperty("os.name").toUpperCase().contains("MAC")){
|
|
GlobalConstant.SYSTYPE = "MAC";
|
|
Properties props=System.getProperties();
|
|
GlobalConstant.LOCPATH = props.getProperty("user.home")+File.separator+".HellohaoData";
|
|
}
|
|
return System.getProperties().getProperty("os.name").toUpperCase().indexOf("WINDOWS") != -1;
|
|
}
|
|
|
|
}
|