handle restoring only specific databases when DB_DUMP_BY_SCHEMA was used

Author: AxelGuignard

entrypoint

@@ -146,7 +146,17 @@ if [[ -n "$DB_RESTORE_TARGET" ]]; then
     mkdir -p $workdir
     $UNCOMPRESS < $TMPRESTORE | tar -C $workdir -xvf -
     RESTORE_OPTS=${RESTORE_OPTS:-}
-    cat $workdir/* | mysql $RESTORE_OPTS -h $DB_SERVER -P $DB_PORT $DBUSER $DBPASS $DBDATABASE
+    # If there are multiple schemas in the archive (e.g. DB_DUMP_BY_SCHEMA was used) and DB_NAMES is set,
+    # restore only the required databases
+    if [[ $(ls -1q $workdir/* | wc -l) -gt 1 ]] && [[ -n "$DB_NAMES" ]]; then
+      for onedb in $DB_NAMES; do
+        echo "Restoring $onedb from " $workdir/$onedb*
+        # /!\ If a schema has a name that begins with another one, it will executed multiple times the other one
+        cat $workdir/$onedb* | mysql $RESTORE_OPTS -h $DB_SERVER -P $DB_PORT $DBUSER $DBPASS
+      done
+    else
+      cat $workdir/* | mysql $RESTORE_OPTS -h $DB_SERVER -P $DB_PORT $DBUSER $DBPASS $DBDATABASE
+    fi
     rm -rf $workdir
     /bin/rm -f $TMPRESTORE
   else