#!/bin/bash

#############################################################################

#

# GNU LESSER GENERAL PUBLIC LICENSE

# Version 3, 29 June 2007

#

# Copyright (C) [2007] [TRON Foundation], Inc. <https://fsf.org/>

# Everyone is permitted to copy and distribute verbatim copies

# of this license document, but changing it is not allowed.

#

#

# This version of the GNU Lesser General Public License incorporates

# the terms and conditions of version 3 of the GNU General Public

# License, supplemented by the additional permissions listed below.

#

# You can find java-tron at https://github.com/tronprotocol/java-tron/

#

##############################################################################

# TRON Full Node Management Simple Script

#

# NOTE: This is a simple and concise script to start and stop the java-tron full node,

# designed for developers to quickly get started and learn.

# It may not be suitable for production environments.

#

# Usage:

# sh start.sh # Start the java-tron FullNode

# sh start.sh -s # Stop the java-tron FullNode

# sh start.sh [options] # Start with additional java-tron options,such as: -c config.conf -d /path_to_data, etc.

#

##############################################################################

# adjust JVM start

# Set the maximum heap size to 9G, adjust as needed

VM_XMX="9G"

# adjust JVM end

FULL_NODE_JAR="FullNode.jar"

FULL_START_OPT=()

PID=""

MAX_STOP_TIME=60

JAVACMD=""

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

RED='\033[0;31m'

GREEN='\033[0;32m'

YELLOW='\033[1;33m'

NC='\033[0m'

log() {

local level="$1"; shift

local timestamp color=""

timestamp=$(date '+%Y-%m-%d %H:%M:%S')

case "$level" in

INFO) color="$GREEN" ;;

WARN) color="$YELLOW" ;;

ERROR) color="$RED" ;;

esac

printf "%b[%s] [%s]:%b %s\n" "$color" "$timestamp" "$level" "$NC" "$*" | tee -a "${SCRIPT_DIR}/start.log"

}

info() { log INFO "$@"; }

warn() { log WARN "$@"; }

error() { log ERROR "$@"; }

die() { error "$@"; exit 1; }

ulimit -n 65535 || warn "Failed to set ulimit -n 65535"

findJava() {

if [ -n "${JAVA_HOME:-}" ]; then

if [ -x "$JAVA_HOME/jre/sh/java" ]; then

JAVACMD="$JAVA_HOME/jre/sh/java"

else

JAVACMD="$JAVA_HOME/bin/java"

fi

[ -x "$JAVACMD" ] || die "JAVA_HOME is invalid: $JAVA_HOME"

else

JAVACMD="java"

which java >/dev/null 2>&1 || die "JAVA_HOME not set and no 'java' in PATH"

fi

"$JAVACMD" -version > /dev/null 2>&1 || die "Java command not working"

}

checkPid() {

# shellcheck disable=SC2009

PID=$(ps -ef |grep $FULL_NODE_JAR |grep -v grep |awk '{print $2}')

}

stopService() {

checkPid

if ! kill -0 "$PID" 2>/dev/null; then

info "java-tron is not running."

return 0

fi

info "Stopping java-tron service (PID: $PID)"

local count=1

while [ -n "$PID" ] && [ $count -le $MAX_STOP_TIME ]; do

kill -TERM "$PID" 2>/dev/null && info "Sent SIGTERM to java-tron (PID: $PID), attempt $count"

sleep 1

checkPid

count=$((count + 1))

done

if [ -n "$PID" ]; then

warn "Forcing kill java-tron (PID: $PID) after $MAX_STOP_TIME seconds"

kill -KILL "$PID" 2>/dev/null

sleep 1

checkPid

fi

if [ -n "$PID" ]; then

die "Failed to stop the service (PID: $PID)"

else

info "java-tron stopped"

wait_with_info 2 "Cleaning up..."

fi

}

startService() {

if [ -n "${FULL_START_OPT[*]}" ]; then

info "Starting java-tron service with options: ${FULL_START_OPT[*]}"

fi

if [ ! -f "$FULL_NODE_JAR" ]; then

die "$FULL_NODE_JAR not found in path $SCRIPT_DIR."

fi

nohup "$JAVACMD" \

-Xmx"$VM_XMX" \

-XX:+UseZGC \

-Xlog:gc,gc+heap:file=gc.log:time,tags,level:filecount=10,filesize=100M \

-XX:ReservedCodeCacheSize=256m \

-XX:+UseCodeCacheFlushing \

-XX:MetaspaceSize=256m \

-XX:MaxMetaspaceSize=512m \

-XX:MaxDirectMemorySize=1g \

-XX:+HeapDumpOnOutOfMemoryError \

-jar "$FULL_NODE_JAR" "${FULL_START_OPT[@]}" \

>> start.log 2>&1 &

info "Waiting for the service to start..."

wait_with_info 5 "Starting..."

checkPid

if [ -n "$PID" ]; then

info "Started java-tron with PID $PID on $HOSTNAME."

else

die "Failed to start java-tron, see start.log or logs/tron.log for details."

fi

}

wait_with_info() {

local seconds=$1

local message=$2

for i in $(seq "$seconds" -1 1); do

info "$message wait ($i) s"

sleep 1

done

}

start() {

checkPid

if [ -n "$PID" ]; then

info "java-tron is already running (PID: $PID), to stop the service: sh start.sh -s"

return

fi

findJava

startService

}

while [ -n "$1" ]; do

case "$1" in

-s)

stopService

exit 0

;;

*)

FULL_START_OPT+=("$@")

break

;;

esac

done

start

exit 0