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

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