shell脚本批量执行指定路径下sql脚本

柔情只为你懂 2024-03-31 09:39 112阅读 0赞

#### shell脚本批量执行指定路径下sql脚本 ####

*  1. 场景描述
 *  2. 创建sql
 *  3. 创建脚本
 *   *  3.1 方式一
     *  3.2 方式二

## 1. 场景描述 ##

Linux环境下通过shell脚本批量执行指定目录下所有sql语句，用来建表建库，初始化项目sql等。  
Linux shell在线格式化：[https://tool.lu/shell/][https_tool.lu_shell]

## 2. 创建sql ##

创建contract\_ddl.sql

-- 创建数据库contract_user
    CREATE DATABASE `contract_user` DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci;
    
    -- 创建合同表contract
    DROP TABLE IF EXISTS `contract`;
    CREATE TABLE `contract`  (
      `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT COMMENT 'ID主键',
      `name` varchar(64) NOT NULL COMMENT '合同名称',
      `code` varchar(64) NOT NULL COMMENT '合同编号',
    
      `deleted` tinyint NOT NULL DEFAULT 0 COMMENT '是否删除 0 未删除 1 删除 默认是0',
      `create_id` bigint(20) NOT NULL DEFAULT 0 COMMENT '创建人账号id',
      `create_time` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
      `update_id` bigint(20) NOT NULL DEFAULT 0 COMMENT '更新人账号id',
      `update_time` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP	COMMENT '更新时间',
    	
      PRIMARY KEY (`id`) USING BTREE,
      index `idx_code_name`(`code`,`name`) USING BTREE
    ) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '合同表' ROW_FORMAT = Dynamic;

创建template\_ddl.sql

-- 创建数据库template_user
    CREATE DATABASE `template_user` DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci;
    
    -- 模板设置表template
    DROP TABLE IF EXISTS `template`;
    CREATE TABLE `template`  (
      `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT COMMENT 'ID主键',
      `name` bigint(20) NOT NULL COMMENT '模板名称',
      `code` bigint(20) NOT NULL COMMENT '模板编码',
    
      `deleted` tinyint NOT NULL DEFAULT 0 COMMENT '是否删除 0 未删除 1 删除 默认是0',
      `create_id` bigint(20) NOT NULL DEFAULT 0 COMMENT '创建人账号id',
      `create_time` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
      `update_id` bigint(20) NOT NULL DEFAULT 0 COMMENT '更新人账号id',
      `update_time` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP	COMMENT '更新时间',
    	
      PRIMARY KEY (`id`) USING BTREE,
      index `idx_code_name`(`code`, `name`) USING BTREE
    ) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '模板表' ROW_FORMAT = Dynamic;

## 3. 创建脚本 ##

### 3.1 方式一 ###

方式一需要手动指定sql全路径名称，相对比较麻烦，仅供参考。

#!/bin/sh
    
    start_date=`date '+%Y%m%d-%H%M%S'`
    echo $start_date ${
        USER} "execute ddl start..."
    
    # mysql
    shost=127.0.0.1
    sport=3306
    suser=root
    spwd=123456
    
    # path，sql放入ddl下
    sqlpath="source /home/ddl/"
    sqlsource="${sqlpath}contract_ddl.sql;${sqlpath}template_ddl.sql;"
    
    # 执行sql脚本，这里会有告警，不影响执行，提示直接输入密码不安全，去掉-p后的spwd，执行时输入安全
    # Warning: Using a password on the command line interface can be insecure.
    mysql -h$shost -p$sport -u$suser -p$spwd -e"$sqlsource"
    
    # end
    end_date=`date '+%Y%m%d-%H%M%S'`
    echo $end_date ${
        USER} "execute ddl end..."

### 3.2 方式二 ###

方式二只需要指定sql路径即可，通过shell遍历，相对方便。

#!/bin/bash
    
    #execute all script in specified directory
    
    MYDATE=$(date +%F'-'%T'-'%w)
    
    MYSQL_PATH=/tmp/scripts #指定的目录
    
    LOG_FILE=/tmp/scripts/exec_${MYDATE}.log
    
    confirm=
    
    db_name=
    
    db_pass=
    
    for file in ${MYSQL_PATH}/*; do
    
    	if [ -f "$file" ]; then
    
    		postfix=$(echo $file | awk -F'.' '{print "."$NF}')
    
    		if [ $postfix = ".sql" ]; then
    
    			if [ ! $db_name ]; then #如果没有指定数据库
    
    				read -p "请输入数据库名：" db_name
    
    				read -p "你输入的数据名是【$db_name】，确认继续请输入--yes--: " confirm
    
    			fi
    
    			if [ "$confirm" = "yes" ] && [ -n $confirm ]; then
    
    				if [ ! $db_pass ]; then #如果没有设置密码
    
    					stty -echo #密码输入保护关闭显示
    
    					read -p "请输入数据库密码：" db_pass
    
    					echo -e "\n"
    
    					stty echo
    
    				fi
    
    				mysql -uroot -p$db_pass -P3306 --default-character-set=utf8 ${db_name} <$file >&error.log
    
    				echo $file
    
    				echo -e "\n===========$file=============\n" >>${LOG_FILE}
    
    				cat error.log >>${LOG_FILE}   #输出执行日志
    
    				error=$(grep ERROR error.log) #读取错误日志信息
    
    				if [ -n "$error" ]; then #如果有错误就退出程序
    
    					echo $error
    
    					exit
    
    				fi
    
    			else
    
    				echo "您已经取消操作!"
    
    				exit
    
    			fi
    
    		fi
    
    	fi
    
    done

[https_tool.lu_shell]: https://tool.lu/shell/