| #!/bin/bash |
| # |
| # Licensed to the Apache Software Foundation (ASF) under one or more |
| # contributor license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright ownership. |
| # The ASF licenses this file to You under the Apache License, Version 2.0 |
| # (the "License"); you may not use this file except in compliance with |
| # the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| # Shell script to install a snapshot into place as the Lucene collection |
| # for a Solr server |
| |
| orig_dir=$(pwd) |
| cd ${0%/*}/.. |
| solr_root=$(pwd) |
| cd ${orig_dir} |
| |
| unset master_host master_status_dir data_dir user verbose debug |
| . ${solr_root}/bin/scripts-util |
| |
| # set up variables |
| prog=${0##*/} |
| log=${solr_root}/logs/${prog}.log |
| |
| LOCKDIR="${solr_root}/logs/snapinstaller-lock" |
| PIDFILE="${LOCKDIR}/PID" |
| |
| |
| # define usage string |
| USAGE="\ |
| usage: $prog [-M master] [-S sdir] [-d dir] [-u username] [-v] [-V] |
| -M master specify hostname of master server from where to pull index |
| snapshot |
| -S specify directory holding snapshot status on master server |
| -d specify directory holding index data on local machine |
| -u specify user to sudo to before running script |
| -v increase verbosity |
| -V output debugging info |
| " |
| |
| # parse args |
| while getopts M:S:d:u:vV OPTION |
| do |
| case $OPTION in |
| M) |
| master_host="$OPTARG" |
| ;; |
| S) |
| master_status_dir="$OPTARG" |
| ;; |
| d) |
| data_dir="$OPTARG" |
| ;; |
| u) |
| user="$OPTARG" |
| ;; |
| v) |
| verbose="v" |
| ;; |
| V) |
| debug="V" |
| ;; |
| *) |
| echo "$USAGE" |
| exit 1 |
| esac |
| done |
| |
| [[ -n $debug ]] && set -x |
| |
| if [[ -z ${master_host} ]] |
| then |
| echo "name of master server missing in $confFile or command line." |
| echo "$USAGE" |
| exit 1 |
| fi |
| |
| if [[ -z ${master_status_dir} ]] |
| then |
| echo "directory holding snapshot status on master server missing in $confFile or command line." |
| echo "$USAGE" |
| exit 1 |
| fi |
| |
| fixUser "$@" |
| |
| dataDir |
| |
| # assume relative path to start at ${solr_root} |
| if [[ "`echo ${master_status_dir}|cut -c1`" != "/" ]] |
| then |
| master_status_dir=${solr_root}/${master_status_dir} |
| fi |
| |
| setStartTime |
| |
| if test -r $PIDFILE |
| then |
| OTHERPID="$(cat "${PIDFILE}")" |
| if ! kill -0 $OTHERPID &>/dev/null; then |
| logMessage removing stale lock ${OTHERPID} |
| rm -rf "${LOCKDIR}" |
| else |
| logExit "lock failed, PID ${OTHERPID} is active" 1 |
| fi |
| fi |
| |
| mkdir "${LOCKDIR}" &>/dev/null |
| echo "$$" >"${PIDFILE}" |
| |
| logMessage started by $oldwhoami |
| logMessage command: $0 $@ |
| |
| # get directory name of latest snapshot |
| name=`perl -e 'chdir q|'${data_dir}'|; print ((sort grep {/^snapshot[.][1-9][0-9]{13}$/} <*>)[-1])'` |
| |
| # clean up after INT/TERM |
| trap 'echo "caught INT/TERM, exiting now but partial installation may have already occured";/bin/rm -rf ${data_dir}/index.tmp$$;logExit aborted 13' INT TERM |
| |
| # is there a snapshot |
| if [[ "${name}" == "" ]] |
| then |
| logMessage no shapshot available |
| logExit ended 0 |
| fi |
| |
| name=${data_dir}/${name} |
| |
| # has snapshot already been installed |
| if [[ ${name} == `cat ${solr_root}/logs/snapshot.current 2>/dev/null` ]] |
| then |
| logMessage latest snapshot ${name} already installed |
| logExit ended 0 |
| fi |
| |
| # make sure master has directory for hold slaves stats/state |
| if |
| ! ssh -o StrictHostKeyChecking=no ${master_host} mkdir -p ${master_status_dir} |
| then |
| logMessage failed to ssh to master ${master_host}, snapshot status not updated on master |
| fi |
| |
| # install using hard links into temporary directory |
| # remove original index and then atomically copy new one into place |
| logMessage installing snapshot ${name} |
| if [[ "${OS}" == "SunOS" || "${OS}" == "Darwin" || "${OS}" == "FreeBSD" ]] |
| then |
| orig_dir=$(pwd) |
| mkdir ${data_dir}/index.tmp$$ && \ |
| cd ${name} && \ |
| find . -print|cpio -pdlmu ${data_dir}/index.tmp$$ 1>/dev/null 2>&1 && \ |
| /bin/rm -rf ${data_dir}/index && \ |
| mv -f ${data_dir}/index.tmp$$ ${data_dir}/index |
| cd ${orig_dir} |
| else |
| cp -lr ${name}/ ${data_dir}/index.tmp$$ && \ |
| /bin/rm -rf ${data_dir}/index && \ |
| mv -f ${data_dir}/index.tmp$$ ${data_dir}/index |
| fi |
| |
| # update distribution stats |
| echo ${name} > ${solr_root}/logs/snapshot.current |
| |
| # push stats/state to master |
| if |
| ! scp -q -o StrictHostKeyChecking=no ${solr_root}/logs/snapshot.current ${master_host}:${master_status_dir}/snapshot.current.`uname -n` |
| then |
| logMessage failed to ssh to master ${master_host}, snapshot status not updated on master |
| fi |
| |
| # notify Solr to open a new Searcher |
| logMessage notifing Solr to open a new Searcher |
| ${solr_root}/bin/commit |
| if [[ $? != 0 ]] |
| then |
| logMessage failed to connect to Solr server |
| logMessage snapshot installed but Solr server has not open a new Searcher |
| logExit failed 1 |
| fi |
| |
| rm -rf "${LOCKDIR}" |
| logExit ended 0 |