add etc

OpenAtomFoundation · Jul 23, 2024 · e407427 · e407427
1 parent 2848690
commit e407427
Show file tree

Hide file tree

Showing 6 changed files with 124 additions and 103 deletions.
diff --git a/build.sh b/build.sh
diff --git a/clear.sh b/clear.sh
diff --git a/pikiwidb.conf → etc/conf/pikiwidb.conf b/pikiwidb.conf → etc/conf/pikiwidb.conf
diff --git a/etc/script/load.sh b/etc/script/load.sh
@@ -0,0 +1,117 @@
+#!/bin/bash
+
+#color code
+C_RED="\033[31m"
+C_GREEN="\033[32m"
+
+C_END="\033[0m"
+
+BUILD_TIME=$(git log -1 --format=%ai)
+BUILD_TIME=${BUILD_TIME: 0: 10}
+
+COMMIT_ID=$(git rev-parse HEAD)
+SHORT_COMMIT_ID=${COMMIT_ID: 0: 8}
+
+BUILD_TYPE=Release
+VERBOSE=0
+CMAKE_FLAGS=""
+MAKE_FLAGS=""
+PREFIX="cmake-build"
+
+PWD=`pwd`
+PROJECT_HOME="${PWD}/../"
+CONF_FILE="${PROJECT_HOME}/etc/conf/pikiwidb.conf"
+ARGS=`getopt -a -o h -l help,debug,verbose,prefix: -- "$@"`
+
+function build() {
+  if [ ! -f "/proc/cpuinfo" ];then
+    CPU_CORE=$(sysctl -n hw.ncpu)
+  else
+    CPU_CORE=$(cat /proc/cpuinfo| grep "processor"| wc -l)
+  fi
+  if [ ${CPU_CORE} -eq 0 ]; then
+    CPU_CORE=1
+  fi
+
+  echo "cpu core ${CPU_CORE}"
+
+  echo "BUILD_TYPE:" $BUILD_TYPE
+  echo "CMAKE_FLAGS:" $CMAKE_FLAGS
+  echo "MAKE_FLAGS:" $MAKE_FLAGS
+
+  if [ "${BUILD_TYPE}" == "Release" ]; then
+    PREFIX="${PREFIX}-release"
+  else
+    PREFIX="${PREFIX}-debug"
+  fi
+
+  cmake -DCMAKE_BUILD_TYPE=${BUILD_TYPE} ${CMAKE_FLAGS} -S . -B ${PREFIX}
+  cmake --build ${PREFIX} -- ${MAKE_FLAGS} -j ${CPU_CORE}
+
+  if [ $? -eq 0 ]; then
+    echo -e "pikiwidb compile complete, output file ${C_GREEN} ${BUILD_DIR}/pikiwidb ${C_END}"
+  else
+    echo -e "${C_RED} pikiwidb compile fail ${C_END}"
+    exit 1
+  fi
+}
+
+function clear() {
+  rm -rf ${PROJECT_HOME}/deps-debug
+  rm -rf ${PROJECT_HOME}/deps-release
+  rm -rf ${PROJECT_HOME}/cmake-build-debug
+  rm -rf ${PROJECT_HOME}/cmake-build-release
+  rm -rf ${PROJECT_HOME}/build
+  rm -rf ${PROJECT_HOME}/build-release
+  rm -rf ${PROJECT_HOME}/build-debug
+  rm -rf ${PROJECT_HOME}/bin
+}
+
+function show_help() {
+  echo "
+  $0 --debug     compile with debug
+  $0 --clear     clear useless directory
+  $0 -h|--help   show help
+  $0 --prefix    compile output path
+  $0 --verbose   compile with verbose
+  "
+  exit 0
+}
+
+eval set -- "${ARGS}"
+while true
+do
+  case "$1" in
+  -c|--clear)
+    clear
+    exit 0
+    ;;
+
+  --debug)
+    BUILD_TYPE=debug
+    build
+    ;;
+
+  -h|--help)
+    show_help
+    ;;
+
+  --prefix)
+    PREFIX=$2
+    shift
+    build
+    ;;
+
+  --verbose)
+    CMAKE_FLAGS="${CMAKE_FLAGS} -DCMAKE_VERBOSE_MAKEFILE:BOOL=ON"
+    MAKE_FLAGS="${MAKE_FLAGS} VERBOSE=1"
+    build
+    ;;
+
+  --)
+    shift
+    break
+    ;;
+  esac
+shift
+done
diff --git a/pikiwidbtests.sh → etc/script/pikiwidbtests.sh b/pikiwidbtests.sh → etc/script/pikiwidbtests.sh
diff --git a/save_load.sh → etc/script/save_load.sh b/save_load.sh → etc/script/save_load.sh
@@ -3,8 +3,13 @@
 killall -9 pikiwidb
 mkdir leader follower1
 
-cd leader && ulimit -n 99999  && rm -fr *  && ../bin/pikiwidb ../pikiwidb.conf --port 7777 &
-cd follower1 && ulimit -n 99999 && rm -fr * && ../bin/pikiwidb ../pikiwidb.conf --port 8888 &
+
+PWD=`pwd`
+PROJECT_HOME="${PWD}/../"
+BIN="${PROJECT_HOME}/bin/pikiwidb"
+CONF="${PROJECT_HOME}/etc/conf/pikiwidb.conf"
+cd leader    && ulimit -n 99999  && rm -fr *  && ${BIN} ${CONF} --port 7777 &
+cd follower1 && ulimit -n 99999 && rm -fr *   && ${BIN} ${CONF} --port 8888 &
 sleep 5
 
 redis-cli -p 7777 raft.cluster init