-
-
Save camlafit/4690c2fa948f2e760d566875daa39a14 to your computer and use it in GitHub Desktop.
Add external tar.gz et and drop collections
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
LOADING=false | |
usage() | |
{ | |
cat << EOF | |
usage: $0 [options] <DBNAME> | |
OPTIONS: | |
-h Show this help. | |
-l Load instead of export | |
-u Mongo username | |
-p Mongo password | |
-H Mongo host string (ex. localhost:27017) | |
-s Mongo database in tar.gz | |
EOF | |
} | |
while getopts "hlu:p:H:s:" opt; do | |
MAXOPTIND=$OPTIND | |
case $opt in | |
h) | |
usage | |
exit | |
;; | |
l) | |
LOADING=true | |
;; | |
u) | |
USERNAME="$OPTARG" | |
;; | |
p) | |
PASSWORD="$OPTARG" | |
;; | |
H) | |
HOST="$OPTARG" | |
;; | |
s) | |
DB_FILE="$OPTARG" | |
if [ ! -f $DB_FILE ]; then | |
echo "Invalid db file" | |
exit 1 | |
fi | |
;; | |
\?) | |
echo "Invalid option $opt" | |
exit 1 | |
;; | |
esac | |
done | |
shift $(($MAXOPTIND-1)) | |
if [ -z "$1" ]; then | |
echo "Usage: $0 [options] <DBNAME>" | |
exit 1 | |
fi | |
DB="$1" | |
if [ -z $DB_FILE ]; then | |
DB_FILE=$DB.tar.gz | |
fi | |
if [ -z "$HOST" ]; then | |
HOST="localhost:27017" | |
CONN="localhost:27017/$DB" | |
else | |
CONN="$HOST/$DB" | |
fi | |
ARGS="" | |
if [ -n "$USERNAME" ]; then | |
ARGS="-u $USERNAME" | |
fi | |
if [ -n "$PASSWORD" ]; then | |
ARGS="$ARGS -p $PASSWORD" | |
fi | |
echo "*************************** Mongo Export ************************" | |
echo "**** Host: $HOST" | |
echo "**** Database: $DB" | |
echo "**** Username: $USERNAME" | |
echo "**** Password: $PASSWORD" | |
echo "**** Loading: $LOADING" | |
echo "*****************************************************************" | |
if $LOADING ; then | |
echo "Loading into $CONN" | |
tar -xzf $DB_FILE | |
pushd $DB >/dev/null | |
for path in *.json; do | |
collection=${path%.json} | |
echo "Loading into $DB/$collection from $path" | |
mongoimport --host $HOST $ARGS -d $DB -c $collection $path --jsonArray --drop | |
done | |
popd >/dev/null | |
rm -rf $DB | |
else | |
DATABASE_COLLECTIONS=$(mongo $CONN $ARGS --quiet --eval 'db.getCollectionNames()' | sed 's/,/ /g') | |
mkdir /tmp/$DB | |
pushd /tmp/$DB 2>/dev/null | |
for collection in $DATABASE_COLLECTIONS; do | |
mongoexport --host $HOST $ARGS -db $DB -c $collection --jsonArray -o $collection.json >/dev/null | |
done | |
pushd /tmp 2>/dev/null | |
tar -czf "$DB.tar.gz" $DB 2>/dev/null | |
popd 2>/dev/null | |
popd 2>/dev/null | |
mv /tmp/$DB.tar.gz ./ 2>/dev/null | |
rm -rf /tmp/$DB 2>/dev/null | |
fi |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment