dyDigitalPrintersAuthorize/conf/script/0/business/BE_System_Init_ERPClient.groovy

import com.yinjie.printerAuthorize.common.api.BusinessExecutor
import com.yinjie.printerAuthorize.common.api.ERPService
import com.yinjie.printerAuthorize.common.api.NoSqlKeysService
import com.yinjie.printerAuthorize.common.api.SupplierInitService
import com.yinjie.printerAuthorize.common.datas.ERPModule
import com.yinjie.printerAuthorize.common.datas.RedisKeys
import com.yinjie.printerAuthorize.common.entity.base.DataBaseMultiItemEx
import com.yinjie.printerAuthorize.common.entity.base.ProcessERPClientItem
import com.yinjie.printerAuthorize.common.entity.base.ProcessStringItem
import com.yinjie.printerAuthorize.common.tool.CodeGenerate
import com.yinjie.printerAuthorize.common.tool.ERPUtils
import com.dySweetFishPlugin.redis.RedisService
import com.dySweetFishPlugin.sql.DBService
import com.dySweetFishPlugin.sql.TableIdService
import com.sweetfish.service.RetResult
import org.apache.commons.io.FileUtils
import org.apache.logging.log4j.LogManager
import org.apache.logging.log4j.Logger
import org.rex.DB
import org.rex.RMap
import org.rex.db.exception.DBException

import javax.annotation.Resource

@SuppressWarnings("unused")
class BE_System_Init_ERPClient implements BusinessExecutor<ProcessERPClientItem, ProcessStringItem> {

    protected final Logger logger = LogManager.getLogger(this.getClass().getSimpleName())

    @Resource
    private SupplierInitService supplierService

    @Resource
    private ERPService erpService

    @Resource
    private DBService dbService

    @Resource
    private RedisService redisService

    @Resource
    private NoSqlKeysService keysService

    @Resource
    private TableIdService tableIdService

    @Resource
    private CodeGenerate codeGenerate

    @Resource(name = "APP_HOME")
    private String appHome

    @Resource(name = "property.sysRunMode")
    private String sysRunMode

    @Override
    String scriptName() {
        return "初始化SAAS商户"
    }

    @Override
    ERPModule module() {
        return ERPModule.SYSTEM
    }

    void start(long supplierCode) {

    }

    RetResult<ProcessStringItem> execute(ProcessERPClientItem source) {
        DataBaseMultiItemEx dataItem = source.dataItem

        DataBaseMultiItemEx srcItem = erpService.getERPSupplierInfoByCode(dataItem.getItemCode())
        if (srcItem != null) {
            return RetResult.<ProcessStringItem> errorT().retinfo("重复的itemCode")
        }

        srcItem = erpService.getERPSupplierInfoBySharding(Long.parseLong(dataItem.getShardingKey()))
        if (srcItem != null) {
            return RetResult.<ProcessStringItem> errorT().retinfo("重复的shardingKey")
        }

//        dataItem.setMd5ParamDigest(codeGenerate.generateMD5Key());

        dataItem.itemType = 1
        dataItem.voidFlag = 0
        dataItem.dataSchema = dataItem.itemCode + "_" + dataItem.shardingKey

        //1、创建数据库
        dbService.update("create dataBase IF NOT EXISTS " + dataItem.dataSchema + " CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;")

        //2、sys_DataBaseMultiItem表中插入记录
        RMap dataParams = new RMap()
        dataParams.put("itemName", dataItem.itemName)
        dataParams.put("dataBaseAlias", dataItem.dataBaseAlias)
        dataParams.put("shardingKey", dataItem.shardingKey)
        dataParams.put("itemType", dataItem.itemType)
        dataParams.put("contactUrl", dataItem.contactUrl)
        dataParams.put("voidFlag", dataItem.voidFlag)
        dataParams.put("itemCode", dataItem.itemCode)
        dataParams.put("contactPerson", dataItem.contactPerson)
        dataParams.put("contactPhone", dataItem.contactPhone)
        dataParams.put("md5ParamDigest", dataItem.md5ParamDigest)

        //先不执行，免得启动的时候找脚本
        DB.update("insert into sys_DataBaseMultiItem " +
                "(itemName, dataBaseAlias, shardingKey, itemType, contactUrl, voidFlag, itemCode, contactPerson, contactPhone, md5ParamDigest) " +
                "values(" +
                "#{itemName}, #{dataBaseAlias}, #{shardingKey}, #{itemType}, #{contactUrl}, #{voidFlag}, #{itemCode}, #{contactPerson}, #{contactPhone}, #{md5ParamDigest})",
                dataParams)

        //需要上面的id,
        long idItem = dbService.getLong("select id from sys_DataBaseMultiItem where shardingKey = " + dataItem.shardingKey)

        //3、增加数据源，这个必须执行，否则下面执行sql无法带上数据源
        dbService.addDataSource(dataItem.dataBaseAlias,
                "com.mysql.jdbc.Driver",
                "jdbc:mysql://192.168.1.22:3306/" + dataItem.dataSchema + "?useUnicode=true&characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull&useSSL=false&rewriteBatchedStatements=true",
                "dbadmin", "Admin@123")

        //4、创建表和视图
        String initSqlRoot = appHome + File.separator +
                "conf" + File.separator +
                "sqlInit" + File.separator +
                (("ERP".equalsIgnoreCase(sysRunMode)) ? "erp" : "mall")

        //获取脚本目录下的所有文件，并按照文件名排序
        File dir = new File(initSqlRoot)
        File[] scriptFiles = dir.listFiles().sort { file ->
            return Integer.parseInt(file.name.split("_")[1])
        }


        scriptFiles.each { file ->
            //如果有存储过程的语句，则需要单独处理，最好每一个存储过程一个sql，因为存储过程的sql包含有 ; 号，整个文件一起执行最好
            def fileContent = file.text
            // 替换${supplyCode}为1000
            def replacedContent = fileContent.replaceAll('\\$\\{supplyCode}', dataItem.shardingKey)

            try {
                def content = ERPUtils.replaceEscape(replacedContent)
                def sqlArray = content.split(";")
                sqlArray.each {sql ->
                    try {
                        dbService.update(dataItem.dataBaseAlias, sql)
                    } catch (DBException e1) {
                        logger.error(sql)
                        logger.error("执行初始化sql失败: " + e1.getMessage(), e1)
                    }

                }
                logger.info("初始化" + dataItem.itemName + "完成")
                //好像文件太大了直接批量执行不行：（
                //dbService.batchUpdate(dataItem.dataBaseAlias, content.split(";"))
            } catch (DBException e) {
                logger.error("执行初始化sql失败: " + e.getMessage(), e)
            }
        }

        //用于产品中心对接,此处将上面生成的itemCode与Md5ParamDigest更新进数据库
        String sql = "UPDATE tbconfigvalue" + dataItem.shardingKey + " SET configValue1 = '" + dataItem.itemCode + "', " +
                "configValue2 = '" + dataItem.md5ParamDigest + "' WHERE configKey = 'productCenterCallClientInfo';"
        dbService.update(dataItem.dataBaseAlias, sql)

        //5、拷贝脚本目录，1000是保留的，不论是mall还erp，都保留一个1000的目录，不然此脚本也不方便调用，其它的都从1000拷贝出去
        FileUtils.copyDirectory(
                new File(appHome + File.separator +
                        "conf" + File.separator +
                        "script" + File.separator +
                        "1000"),
                new File(appHome + File.separator +
                        "conf" + File.separator +
                        "script" + File.separator +
                        dataItem.shardingKey)
        )

        //6、分表分库写入到redis
        dataItem.id = idItem

        redisService.hset(keysService.getRedisKey(RedisKeys.KEY_SUPPLIER_PLATFORM), String.valueOf(idItem), dataItem)
        redisService.hset(keysService.getRedisKey(RedisKeys.KEY_SUPPLIER_SHARDING_KEY_PLATFORM), dataItem.shardingKey, dataItem)
        redisService.hset(keysService.getRedisKey(RedisKeys.KEY_SUPPLIER_ITEM_CODE_PLATFORM), dataItem.itemCode, dataItem)

        //7、id、code初始化
        tableIdService.initSupplierId(dataItem)
        tableIdService.initSupplierCode(dataItem)

        //8、调用初始化
        supplierService.addSupplier(dataItem)


        return RetResult.<ProcessStringItem> successT()
    }
}