#!/bin/bash

function ti_cmd_tpch_create
{
	local index="${1}"
	local mod_name="${2}"
	local dir="${3}"
	local conf_rel_path="${4}"
	local host="${5}"

	shift 5

	if [ "${mod_name}" != 'tidb' ]; then
		return
	fi

	if [ -z "${2+x}" ]; then
		echo "[cmd tpch/create] usage: <cmd> scale table(all|lineitem|...) [db_suffix=\"\"] [float_type=decimal|double]" >&2
		return
	fi

	local scale="${1}"
	local table="${2}"

	# Append db_suffix to `db` if not empty
	if [ ! -z "${3+x}" ] && [ ! -z "${3}" ]; then
		local db=`echo "tpch_${scale}_${3}" | tr '.' '_'`
	else
		local db=`echo "tpch_${scale}" | tr '.' '_'`
	fi

	# Use Decimal or Double
	if [ ! -z "${4+x}" ] && [ ! -z "${4}" ]; then
		local field_type="${4}"
	else
		local field_type="decimal"
	fi

	if [ "${field_type}" == "decimal" ]; then
		local schema_dir="${integrated}/resource/tpch/mysql/schema"
	elif [ "${field_type}" == "double" ]; then
		local ori_schema_dir="${integrated}/resource/tpch/mysql/schema"
		local schema_dir="${ori_schema_dir}_double"
		# Replace fields from `DECIMAL(..,..)` to `DOUBLE`
		trans_schema_fields_decimal_to_double "${ori_schema_dir}" "${schema_dir}"
	else
		echo "unknown field_type: ${field_type}" >&2
		echo "[cmd tpch/create] usage: <cmd> scale table(all|lineitem|...) [db_suffix=\"\"] [decimal_or_double=decimal|double]" >&2
		return
	fi

	local port=`ssh_get_value_from_proc_info "${host}" "${dir}" 'tidb_port'`
	if [ -z "${port}" ]; then
		echo "[${host}] ${dir}: getting tidb port failed" >&2
		return 1
	fi

	if [ "${table}" != 'all' ]; then
		local tables=("${table}")
	else
		local tables=(nation customer orders part region supplier partsupp lineitem)
	fi

	for table in ${tables[@]}; do
		local start_time=`date +%s`
		echo "=> [$host] creating ${db}.${table}"
		create_tpch_table_to_mysql "${host}" "${port}" "${schema_dir}" "${db}" "${table}"
		local finish_time=`date +%s`
		local duration=$((finish_time-start_time))
		echo "   created in ${duration}s"
	done
}

ti_cmd_tpch_create "${@}"