github.com/klaytn/klaytn@v1.10.2/build/packaging/linux/bin/kpnd (about)

     1  #!/usr/bin/env bash
     2  
     3  BIN=$(cd "$(dirname $0)"; pwd)
     4  CMD_HOME=$(dirname $BIN)
     5  CONF=$CMD_HOME/conf
     6  STOP_TIMEOUT=10 # seconds to wait for a clean exit
     7  
     8  source $CONF/kpnd.conf
     9  
    10  if [ -z $DATA_DIR ]; then
    11      echo
    12      echo "  [ERROR] : DATA_DIR in conf/kpnd.conf is not defined."
    13      exit 1
    14  fi
    15  
    16  if [ ! -d $DATA_DIR ]; then
    17      echo
    18      echo "  [ERROR] : Genesis block is not initiated: [conf/kpnd.conf - DATA_DIR=$DATA_DIR]"
    19      exit 1
    20  fi
    21  
    22  pidfile=$DATA_DIR/kpnd.pid
    23  auto_restart_daemon_pidfile=$DATA_DIR/restart_daemon_kpnd.pid
    24  
    25  __pid_run() {
    26      unset pid
    27      if [ ! -f $pidfile ]; then
    28          return
    29      fi
    30      PID_NUM=$(cat $pidfile)
    31      if [[ ! -z "$PID_NUM" ]]; then
    32          export pid=$(ps -p $PID_NUM -o pid=)
    33      fi
    34  }
    35  
    36  __auto_restart_daemon_pid_run() {
    37      unset auto_restart_daemon_pid
    38      if [ ! -f $auto_restart_daemon_pidfile ]; then
    39          return
    40      fi
    41      AUTO_RESTART_DAEMON_PID_NUM=$(cat $auto_restart_daemon_pidfile)
    42      if [[ ! -z "$AUTO_RESTART_DAEMON_PID_NUM" ]]; then
    43          export auto_restart_daemon_pid=$(ps -p $AUTO_RESTART_DAEMON_PID_NUM -o pid=)
    44      fi
    45  }
    46  
    47  __kill_timeout() {
    48      local PIDNUM=$1
    49      kill $PIDNUM
    50      for i in `seq 0 100 $((1000 * $STOP_TIMEOUT))`; do
    51          if ! kill -0 $PIDNUM 2> /dev/null; then
    52              echo "OK"
    53              return
    54          fi
    55          sleep 0.1
    56      done
    57      kill -9 $PIDNUM && echo "Killed"
    58  }
    59  
    60  #------------------------Related Auto restart daemon functions-----------------------------
    61  __auto_restart_daemon() {
    62      local backOffTime=$AUTO_RESTART_INTERVAL
    63      local coeff=2
    64      while :
    65      do
    66          sleep 1
    67          __pid_run
    68          if [ -z "$pid" ]; then
    69              echo "INFO[`date`] node[${PID_NUM}] is down"
    70              if [ -f $pidfile ]; then
    71                  echo "INFO[`date`] remove redundant pid file"
    72                  rm -f ${pidfile}
    73              fi
    74              echo "INFO[`date`] Sleep for backOffTime.... ${backOffTime} seconds."
    75              sleep $backOffTime
    76              echo -n "INFO[`date`] "
    77              start_node
    78              backOffTime=$(echo $backOffTime $coeff | awk '{printf "%.1f\n",$1*$2}')
    79  
    80              echo "INFO[`date`] backOffTime = ${backOffTime}, Restarted node pid = ${PID_NUM}"
    81              PID_NUM=$(cat $pidfile)
    82              echo ""
    83          fi
    84      done
    85  }
    86  
    87  start_auto_restart_daemon() {
    88      __auto_restart_daemon_pid_run
    89      if [ -z $auto_restart_daemon_pid ]; then
    90          __auto_restart_daemon >> ${LOG_DIR}/restart_daemon.out 2>&1 &
    91          disown
    92          AUTO_RESTART_DAEMON_PID_NUM=$!
    93          AUTO_RESTART_DAEMON_RETVAL=$?
    94  
    95          set +f
    96          if [ $AUTO_RESTART_DAEMON_RETVAL = 0 ]; then
    97              echo $AUTO_RESTART_DAEMON_PID_NUM > ${auto_restart_daemon_pidfile}
    98              echo "Success to start auto restart daemon."
    99          else
   100              echo "Fail to start auto restart daemon."
   101          fi
   102      fi
   103  }
   104  
   105  stop_auto_restart_daemon() {
   106      __auto_restart_daemon_pid_run
   107      [ -z "$auto_restart_daemon_pid" ] && echo "auto restart daemon is not running" && return
   108      echo -n "Shutting down auto restart daemon:  "
   109      __kill_timeout $(cat ${auto_restart_daemon_pidfile}) && rm -f ${auto_restart_daemon_pidfile}
   110  }
   111  
   112  status_auto_restart_daemon() {
   113      __auto_restart_daemon_pid_run
   114      if [ -n "$auto_restart_daemon_pid" ]; then
   115          echo "auto restart daemon is running."
   116      else
   117          echo "auto restart daemon is down."
   118      fi
   119  }
   120  
   121  #------------------------Related to Klaytn node functions-----------------------------
   122  
   123  __check_option() {
   124      if [ ! -d $LOG_DIR ]; then
   125          mkdir -p $LOG_DIR
   126      fi
   127  
   128      set -f
   129      OPTIONS=""
   130  
   131      if [[ ! -z $METRICS ]] && [[ $METRICS -eq 1 ]]; then
   132          OPTIONS="$OPTIONS --metrics"
   133      fi
   134  
   135      if [[ ! -z $PROMETHEUS ]] && [[ $PROMETHEUS -eq 1 ]]; then
   136          OPTIONS="$OPTIONS --prometheus"
   137      fi
   138  
   139      if [[ ! -z $DB_NO_PARALLEL_WRITE ]] && [[ $DB_NO_PARALLEL_WRITE -eq 1 ]]; then
   140          OPTIONS="$OPTIONS --db.no-parallel-write"
   141      fi
   142  
   143      if [[ ! -z $MULTICHANNEL ]] && [[ $MULTICHANNEL -eq 1 ]]; then
   144          OPTIONS="$OPTIONS --multichannel"
   145      fi
   146  
   147      if [[ ! -z $RPC_ENABLE ]] && [[ $RPC_ENABLE -eq 1 ]]; then
   148          OPTIONS="$OPTIONS --rpc"
   149          RPC_API=`echo $RPC_API | tr -d "[:space:]"`
   150          if [ ! -z $RPC_API ]; then
   151              OPTIONS="$OPTIONS --rpcapi $RPC_API"
   152          fi
   153          if [ ! -z $RPC_PORT ]; then
   154              OPTIONS="$OPTIONS --rpcport $RPC_PORT"
   155          fi
   156          if [ ! -z $RPC_ADDR ]; then
   157              OPTIONS="$OPTIONS --rpcaddr $RPC_ADDR"
   158          fi
   159          if [ ! -z $RPC_CORSDOMAIN ]; then
   160              OPTIONS="$OPTIONS --rpccorsdomain $RPC_CORSDOMAIN"
   161          fi
   162          if [ ! -z $RPC_VHOSTS ]; then
   163              OPTIONS="$OPTIONS --rpcvhosts $RPC_VHOSTS"
   164          fi
   165      fi
   166  
   167      if [[ ! -z $WS_ENABLE ]] && [[ $WS_ENABLE -eq 1 ]]; then
   168          OPTIONS="$OPTIONS --ws"
   169          WS_API=`echo $WS_API | tr -d "[:space:]"`
   170          if [ ! -z $WS_API ]; then
   171              OPTIONS="$OPTIONS --wsapi $WS_API"
   172          fi
   173          if [ ! -z $WS_PORT ]; then
   174              OPTIONS="$OPTIONS --wsport $WS_PORT"
   175          fi
   176          if [ ! -z $WS_ADDR ]; then
   177              OPTIONS="$OPTIONS --wsaddr $WS_ADDR"
   178          fi
   179          if [ ! -z $WS_ORIGINS ]; then
   180              OPTIONS="$OPTIONS --wsorigins $WS_ORIGINS"
   181          fi
   182      fi
   183  
   184      # Cypress network => NETWORK_ID is null && NETWORK = "cypress"
   185      # Baobab network => NETWORK_ID is null && NETWORK = "baobab"
   186      # Else => private network
   187      if [[ -z $NETWORK_ID ]]; then
   188          if [[ $NETWORK == "baobab" ]]; then
   189              OPTIONS="$OPTIONS --baobab"
   190          elif [[ $NETWORK == "cypress" ]]; then
   191              OPTIONS="$OPTIONS --cypress"
   192          else
   193              echo
   194              echo "[ERROR] network id is not specified and network is not available."
   195              echo "Available network: baobab, cypress"
   196              exit 1
   197          fi
   198      else
   199          OPTIONS="$OPTIONS --networkid $NETWORK_ID"
   200          echo "[INFO] creating a private network: $NETWORK_ID"
   201          if [[ ! -z $NETWORK ]]; then
   202              echo
   203              echo "[WARN] ignoring the specified network: $NETWORK"
   204          fi
   205      fi
   206  
   207      if [ ! -z $DATA_DIR ]; then
   208          OPTIONS="$OPTIONS --datadir $DATA_DIR"
   209      fi
   210  
   211      if [ ! -z $PORT ]; then
   212          OPTIONS="$OPTIONS --port $PORT"
   213      fi
   214  
   215      if [ ! -z $SUBPORT ]; then
   216          OPTIONS="$OPTIONS --subport $SUBPORT"
   217      fi
   218  
   219      if [ ! -z $SERVER_TYPE ]; then
   220          OPTIONS="$OPTIONS --srvtype $SERVER_TYPE"
   221      fi
   222  
   223      if [ ! -z $VERBOSITY ]; then
   224          OPTIONS="$OPTIONS --verbosity $VERBOSITY"
   225      fi
   226  
   227      if [ ! -z $TXPOOL_EXEC_SLOTS_ALL ]; then
   228          OPTIONS="$OPTIONS --txpool.exec-slots.all $TXPOOL_EXEC_SLOTS_ALL"
   229      fi
   230  
   231      if [ ! -z $TXPOOL_NONEXEC_SLOTS_ALL ]; then
   232          OPTIONS="$OPTIONS --txpool.nonexec-slots.all $TXPOOL_NONEXEC_SLOTS_ALL"
   233      fi
   234  
   235      if [ ! -z $TXPOOL_EXEC_SLOTS_ACCOUNT ]; then
   236          OPTIONS="$OPTIONS --txpool.exec-slots.account $TXPOOL_EXEC_SLOTS_ACCOUNT"
   237      fi
   238  
   239      if [ ! -z $TXPOOL_NONEXEC_SLOTS_ACCOUNT ]; then
   240          OPTIONS="$OPTIONS --txpool.nonexec-slots.account $TXPOOL_NONEXEC_SLOTS_ACCOUNT"
   241      fi
   242  
   243      if [ ! -z $SYNCMODE ]; then
   244          OPTIONS="$OPTIONS --syncmode $SYNCMODE"
   245      fi
   246  
   247      if [ ! -z $MAXCONNECTIONS ]; then
   248          OPTIONS="$OPTIONS --maxconnections $MAXCONNECTIONS"
   249      fi
   250  
   251      if [ ! -z $LDBCACHESIZE ]; then
   252          OPTIONS="$OPTIONS --db.leveldb.cache-size $LDBCACHESIZE"
   253      fi
   254  
   255      if [[ ! -z $NO_DISCOVER ]] && [[ $NO_DISCOVER -eq 1 ]]; then
   256          OPTIONS="$OPTIONS --nodiscover"
   257      fi
   258  
   259      if [[ ! -z $BOOTNODES ]] && [[ $BOOTNODES != "" ]]; then
   260          OPTIONS="$OPTIONS --bootnodes $BOOTNODES"
   261      fi
   262  
   263      if [[ ! -z $ADDITIONAL ]] && [[ $ADDITIONAL != "" ]]; then
   264          OPTIONS="$OPTIONS $ADDITIONAL"
   265      fi
   266  }
   267  
   268  start_node() {
   269      __pid_run
   270      [ -n "$pid" ] && echo "kpnd already running...[$pid]" && return
   271  
   272      echo -n "Starting kpnd: "
   273  
   274      __check_option
   275  
   276      BASEDIR="$( cd "$(dirname "$0")" >/dev/null 2>&1 ; pwd -P )"
   277      CURRENTFILE=`basename "$0"`
   278      OPTIONS="$OPTIONS --autorestart.daemon.path $BASEDIR/$CURRENTFILE"
   279  
   280      $BIN/kpn $OPTIONS >> ${LOG_DIR}/kpnd.out 2>&1 &
   281      RETVAL=$?
   282      PIDNUM=$!
   283      set +f
   284      if [ $RETVAL = 0 ]; then
   285          echo $PIDNUM > ${pidfile}
   286          echo "Success to start node."
   287      else
   288          echo "Fail to start node."
   289      fi
   290      return $RETVAL
   291  }
   292  
   293  stop_node() {
   294      __pid_run
   295      [ -z "$pid" ] && echo "kpnd is not running" && return
   296      echo -n "Shutting down kpnd: "
   297      __kill_timeout $(cat ${pidfile}) && rm -f ${pidfile}
   298  }
   299  
   300  status_node() {
   301      __pid_run
   302      if [ -n "$pid" ]; then
   303          echo "kpnd is running"
   304      else
   305          echo "kpnd is down"
   306      fi
   307  }
   308  
   309  #--------------------- Public functions --------------------------
   310  start() {
   311      if [ ! -d $LOG_DIR ]; then
   312          mkdir -p $LOG_DIR
   313      fi
   314  
   315      if [ ! -z $AUTO_RESTART_NODE ] && [[ $AUTO_RESTART_NODE -eq 1 ]]; then
   316          start_auto_restart_daemon
   317      else
   318          start_node
   319      fi
   320  }
   321  
   322  start_docker() {
   323      echo -n "Starting kpnd: "
   324      __check_option
   325  
   326      echo "$BIN/kpn $OPTIONS"
   327      $BIN/kpn $OPTIONS
   328  }
   329  
   330  stop() {
   331      if [ ! -z $AUTO_RESTART_NODE ] && [[ $AUTO_RESTART_NODE -eq 1 ]]; then
   332          stop_auto_restart_daemon
   333      fi
   334      stop_node
   335  }
   336  
   337  status() {
   338      status_node
   339      if [ ! -z $AUTO_RESTART_NODE ] && [[ $AUTO_RESTART_NODE -eq 1 ]]; then
   340          status_auto_restart_daemon
   341      fi
   342  }
   343  
   344  restart() {
   345      stop
   346      sleep 3
   347      start
   348  }
   349  
   350  case "$1" in
   351      start)
   352          start
   353          ;;
   354      start-docker)
   355          start_docker
   356          ;;
   357      stop)
   358          stop
   359          ;;
   360      status)
   361          status
   362          ;;
   363      restart)
   364          restart
   365          ;;
   366      *)
   367          echo "Usages: kpnd {start|start-docker|stop|restart|status}"
   368          exit 1
   369          ;;
   370  esac
   371  exit 0