table_create.js

var fs = require('fs');
require('./util/extension')
const log = require("./config/log");
const db = require('./config/db')
const Sequelize = db.sequelize;
var stat = fs.stat;
var dateString = process.argv[2]
/**
 * 创建表
 */
function createTable() {
    const BehaviorInfo = Sequelize.import('./schema_temp/behaviorInfo');
    BehaviorInfo.sync({force: false});
    
    const CustomerPV = Sequelize.import('./schema_temp/customerPV');
    CustomerPV.sync({force: false});
    
    const DailyActivity = Sequelize.import('./schema_temp/dailyActivity');
    DailyActivity.sync({force: false});
    
    const EmailCode = Sequelize.import('./schema_temp/emailCode');
    EmailCode.sync({force: false});
    
    const ExtendBehaviorInfo = Sequelize.import('./schema_temp/extendBehaviorInfo');
    ExtendBehaviorInfo.sync({force: false});
    
    const HttpErrorInfo = Sequelize.import('./schema_temp/HttpErrorInfo');
    HttpErrorInfo.sync({force: false});
    
    const HttpLogInfo = Sequelize.import('./schema_temp/HttpLogInfo');
    HttpLogInfo.sync({force: false});
    
    const JavascriptErrorInfo = Sequelize.import('./schema_temp/javascriptErrorInfo');
    JavascriptErrorInfo.sync({force: false});
    
    const LoadPageInfo = Sequelize.import('./schema_temp/loadPageInfo');
    LoadPageInfo.sync({force: false});
    
    const ResourceLoadInfo = Sequelize.import('./schema_temp/resourceLoadInfo');
    ResourceLoadInfo.sync({force: false});
    
    const ScreenShotInfo = Sequelize.import('./schema_temp/ScreenShotInfo');
    ScreenShotInfo.sync({force: false});
    
    const VideosInfo = Sequelize.import('./schema_temp/videosInfo');
    VideosInfo.sync({force: false});
    setTimeout(function() {
        log.printInfo("即将启动下一轮创建程序，请不要关闭，等待程序自动停止...")
    }, 2000)
}
/*
 * 复制目录中的所有文件包括子目录
 * @param{ String } 需要复制的目录
 * @param{ String } 复制到指定的目录
 */

var copy = function( src, dst ){
    // 读取目录中的所有文件/目录
    fs.readdir( src, function( err, paths ){
        if( err ){ throw err; }
        paths.forEach(function( path ){
            var _src = src + '/' + path,
                _dst = dst + '/' + path,
                readable, writable;      
            stat( _src, function( err, st ){
                if( err ){ throw err; }
                // 判断是否为文件
                if( st.isFile() ){
                    // 创建读取流
                    readable = fs.createReadStream( _src );
                    // 创建写入流
                    writable = fs.createWriteStream( _dst ); 
                    // 通过管道来传输流
                    readable.pipe( writable );
                }
                // 如果是目录则递归调用自身
                else if( st.isDirectory() ){
                    exists( _src, _dst, copy );
                }
            });
        });
    });
};

/**
 * 替换日期方法
 */
function replaceDate(dateString) {
    let path = './schema_temp';
    let files = fs.readdirSync(path);
    let fileCount = files.length
    let replaceCount = 0
    for(let i = 0; i < files.length; i++){
      fs.readFile(`${path}/${files[i]}`,function(err, data){
          if (data.indexOf("date-tail") >= 0) {
            let newString = data.toString().replace(/date-tail/g, dateString)
            fs.writeFile(`${path}/${files[i]}`, newString, (err) => {
              if (err) throw err;
              log.printInfo(files[i] + "  重置数据库表名！" + i);
              replaceCount ++
            });
          }
      })
    }
    let timeout = setInterval(function() {
        // 因为有个baseInfo.js不用替换，所以需要减一
        if (replaceCount == fileCount - 1) {
            createTable()
            clearInterval(timeout)
        }
    }, 200)
}

/**
 * 启动程序
 */
function startProgram() {
    log.printInfo("===============================================")
    log.printInfo("= 即将启动创建程序，时间比较长，请耐心等待... =")
    log.printInfo("===============================================")
    fs.mkdir( "./schema_temp", function(err){
        if ( err ) { 
            log.printInfo("文件夹 /schema_temp 已经存在")
        } else {
            log.printInfo("新建文件夹 /schema_temp")
        }
        copy("./schema_base/", "./schema_temp/")
    });
    setTimeout(function() {
        replaceDate(dateString)
    }, 3000)

    // 将package.json恢复成原样
    fs.readFile('./package.json', function(err, data){
        let newString = data.toString().replace(/node .* \d{8}/g, "table_create_command")
        fs.writeFile('./package.json', newString, (err) => {
            if (err) throw err;
            log.printInfo("命令配置已经恢复");
        });
    })
}

startProgram()