package util;

import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;

/**
 * mysql数据库操作工具
 *
 * @author yzc  2019/12/12 上午9:55.
 */
public class MysqlDatabaseUtil {

    /**
     * 数据库导入
     *
     * @param filePath     导入文件存储位置
     * @param host         mysql IP
     * @param port         mysql port
     * @param userName     mysql 用户名
     * @param password     mysql 密码
     * @param dataBaseName 数据库名称
     * @return
     */
    public static ControlResult importSql(String filePath, String host, String port, String userName, String password, String dataBaseName) {
        try {
            StringBuffer sbr = new StringBuffer();
            sbr.append("gunzip < ");
            sbr.append(filePath);
            sbr.append(" | mysql");
            sbr.append(" -u").append(userName);
            sbr.append(" -p").append(password);
            sbr.append(" -h").append(host);
            sbr.append(" -P").append(port).append(" ");
            sbr.append(dataBaseName);
            Runtime rt = Runtime.getRuntime();
            // 调用命令
            rt.exec(new String[]{"/bin/sh", "-c", sbr.toString()});

        } catch (Exception e) {
            e.printStackTrace();
            return ControlResult.error("操作异常，请检查数据库是否存在,或联系管理员！");
        }
        return ControlResult.success();
    }

    /**
     * 数据库导出
     *
     * @param outputPath   导出文件存储位置
     * @param host         mysql IP
     * @param port         mysql port
     * @param userName     mysql 用户名
     * @param password     mysql 密码
     * @param dataBaseName 数据库名称
     * @param tableNames   表名称数组（如果有表名称，则只导出对应的数据表）
     * @return
     */
    public static ControlResult exportSql(String outputPath, String host, String port, String userName, String password,
                                          String dataBaseName, List<String> tableNames) {
        //拼接导出的文件名
        StringBuffer nameSbf = new StringBuffer();
        try {
            Runtime rt = Runtime.getRuntime();
            //拼接数据库导出命令
            StringBuffer sbr = new StringBuffer();
            sbr.append("mysqldump");
            sbr.append(" -u").append(userName);
            sbr.append(" -p").append(password);
            sbr.append(" -h").append(host);
            sbr.append(" -P").append(port);
            nameSbf.append(dataBaseName);
            if (null != tableNames && tableNames.size() > 0) {
                sbr.append(" -BR ");
                sbr.append(dataBaseName).append(" ");
                tableNames.stream().forEach(name -> {
                    sbr.append(name).append(" ");
                    nameSbf.append("_").append(name);
                });
            } else {
                // -B 创建数据库 -R 导出函数和存储过程 sbr.append(" -BR ")
                sbr.append(" -BR ").append(dataBaseName);
            }
            sbr.append(" | gzip ");
            Date date = new Date();
            SimpleDateFormat sdf = new SimpleDateFormat("yyMMddHHmmssSSS");
            String now = sdf.format(date);
            nameSbf.append("_" + now + ".sql.gz");
            sbr.append(" > ").append(outputPath).append(nameSbf);
            // 调用命令
            rt.exec(new String[]{"/bin/sh", "-c", sbr.toString()});
            System.out.println("================");
            System.out.println(nameSbf.toString());
            return ControlResult.success(nameSbf.toString());
        } catch (Exception e) {
            e.printStackTrace();
            return ControlResult.error("操作异常，请检查数据库是否存在或联系管理员！");
        }
    }

    public static void main(String[] args) throws InterruptedException {
        //importSql("/Users/yzc/Desktop/keyManage_20191212181852073.sql.gz","127.0.0.1","3306","root","root","");
        ControlResult res = exportSql("/Users/yzc/Desktop/", "localhost", "3306", "root", "root", "keyManage", null);

    }

}