/
generate_db_postgres.sh
executable file
·54 lines (43 loc) · 1.33 KB
/
generate_db_postgres.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#!/bin/bash
RED='\033[0;31m'
CYAN='\033[0;36m'
MAGENTA='\033[0;35m'
NC='\033[0m'
base="bit_count"
database="bit_count"
batchSize=20
domain=500000
alignments="2048 4096 8192 15040 16384 32768 65536 131072"
while getopts 'd:a:s:' flag; do
case "${flag}" in
d) domain=$OPTARG ;;
b) batchSize=$OPTARG ;;
a) alignments=$OPTARG ;;
*) error "Unexpected option ${flag}" ;;
esac
done
read -a arr <<< "$alignments"
for alignment in "${arr[@]}"
do
table="${base}_${alignment}_$(printf %07d $domain)"
printf "> Generating table: ${MAGENTA}$table${NC}\n"
# psql -U postgres -d database_name -c
psql -q -c "DROP TABLE IF EXISTS \"$table\";"
psql -q -c "CREATE TABLE \"$table\" (\"id\" SERIAL, \"bit\" bit($alignment), PRIMARY KEY (\"id\"));"
psql -q -c "TRUNCATE TABLE \"$table\";"
hexLength=$(($alignment/8))
for (( j=0; j<($domain/$batchSize); j++))
do
echo -ne ">>> Adding rows ($(($j * $batchSize))/$domain).\r"
sql=''
for (( i=1; i<=$batchSize; i++ ))
do
randomhex=$(openssl rand -hex $hexLength)
sql="$sql(x'$randomhex'::bit($alignment))"
if [[ $i -ne $batchSize ]]; then sql="$sql," ; fi
done
# insert here
psql -q -c "INSERT INTO $table (bit) VALUES $sql;"
done
echo -ne "\n"
done