shell脚本批量执行指定路径下sql脚本-蒲公英云

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

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

1. 场景描述

Linux环境下通过shell脚本批量执行指定目录下所有sql语句，用来建表建库，初始化项目sql等。
Linux 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