From 12b04ed0af74727b1c0b91f3aee9fdd9f652b29d Mon Sep 17 00:00:00 2001
From: midoks <midoks@163.com>
Date: Fri, 29 Jan 2021 19:13:11 +0800
Subject: [PATCH] yp

---
 plugins/swap/index.py        | 35 ++++++++++++--
 plugins/swap/init.d/swap.tpl | 92 +++++++++++++++---------------------
 2 files changed, 68 insertions(+), 59 deletions(-)

diff --git a/plugins/swap/index.py b/plugins/swap/index.py
index 64e9e7e81..f42b9ec6e 100755
--- a/plugins/swap/index.py
+++ b/plugins/swap/index.py
@@ -65,22 +65,47 @@ def status():
     return 'start'
 
 
+def getInitDTpl():
+    path = getPluginDir() + "/init.d/" + getPluginName() + ".tpl"
+    return path
+
+
 def initDreplace():
-    return '/swapfile'
+
+    file_tpl = getInitDTpl()
+    service_path = os.path.dirname(os.getcwd())
+
+    initD_path = getServerDir() + '/init.d'
+    if not os.path.exists(initD_path):
+        os.mkdir(initD_path)
+    file_bin = initD_path + '/' + getPluginName()
+
+    # initd replace
+    content = mw.readFile(file_tpl)
+    content = content.replace('{$SERVER_PATH}', '/swapfile')
+    mw.writeFile(file_bin, content)
+    mw.execShell('chmod +x ' + file_bin)
+
+    # config replace
+    conf_content = mw.readFile(getConf())
+    conf_content = conf_content.replace('{$SERVER_PATH}', service_path)
+    mw.writeFile(getServerDir() + '/redis.conf', conf_content)
+
+    return file_bin
 
 
 def start():
     file = initDreplace()
-    data = mw.execShell('swapon ' + file)
-    if data[1] == '':
+    data = mw.execShell(file + ' start')
+    if data[0] == '':
         return 'ok'
     return 'fail'
 
 
 def stop():
     file = initDreplace()
-    data = mw.execShell('swapoff ' + file)
-    if data[1] == '':
+    data = mw.execShell(file + ' stop')
+    if data[0] == '':
         return 'ok'
     return 'fail'
 
diff --git a/plugins/swap/init.d/swap.tpl b/plugins/swap/init.d/swap.tpl
index b61210ac4..b27a957f8 100644
--- a/plugins/swap/init.d/swap.tpl
+++ b/plugins/swap/init.d/swap.tpl
@@ -1,73 +1,57 @@
-#!/bin/sh
+#!/bin/bash
 # chkconfig: 2345 55 25
-# description: Redis Service
+# description: MW Cloud Service
 
 ### BEGIN INIT INFO
-# Provides:          Redis
+# Provides:          bt
 # Required-Start:    $all
 # Required-Stop:     $all
 # Default-Start:     2 3 4 5
 # Default-Stop:      0 1 6
-# Short-Description: starts Redis
-# Description:       starts the MDW-Web
+# Short-Description: starts mw
+# Description:       starts the mw
 ### END INIT INFO
 
-# Simple Redis init.d script conceived to work on Linux systems
-# as it does use of the /proc filesystem.
 
-CONF="{$SERVER_PATH}/redis/redis.conf"
-REDISPORT=$(cat $CONF |grep port|grep -v '#'|awk '{print $2}')
-REDISPASS=$(cat $CONF |grep requirepass|grep -v '#'|awk '{print $2}')
-if [ "$REDISPASS" != "" ];then
-	REDISPASS=" -a $REDISPASS"
-fi
-EXEC={$SERVER_PATH}/redis/bin/redis-server
-CLIEXEC="{$SERVER_PATH}/redis/bin/redis-cli -p $REDISPORT$REDISPASS"
-PIDFILE={$SERVER_PATH}/redis/redis_6379.pid
+PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:~/bin
 
-mkdir -p {$SERVER_PATH}/redis/data
+app_file={$SERVER_PATH}
 
 app_start(){
-	if [ -f $PIDFILE ];then
-		kill -9 `cat $PIDFILE`
-	fi
-	
-	echo "Starting Redis server..."
-	nohup $EXEC $CONF >> {$SERVER_PATH}/redis/logs.pl 2>&1 &
+	isStart=`free -m|grep Swap|awk '{print $2}'`
+	if [ "$isStart" == '0' ];then
+            echo -e "Starting swap... \c"
+            swapon $app_file
+            echo -e "\033[32mdone\033[0m"
+    else
+            echo "Starting swap already running"
+    fi
 }
-app_stop(){
-	if [ ! -f $PIDFILE ]
-	then
-			echo "$PIDFILE does not exist, process is not running"
-	else
-			PID=$(cat $PIDFILE)
-			echo "Stopping ..."
-			$CLIEXEC shutdown
-			while [ -x /proc/${PID} ]
-			do
-				echo "Waiting for Redis to shutdown ..."
-				sleep 1
-			done
-			echo "Redis stopped"
-			rm -rf $PIDFILE
-	fi
+
+app_stop()
+{
+
+    echo -e "Stopping swap... \c";
+    swapoff $app_file
+    echo -e "\033[32mdone\033[0m"
 }
 
+app_status()
+{
+    isStart=`free -m|grep Swap|awk '{print $2}'`
+    if [ "$isStart" == '0' ];then
+            echo -e "\033[32mswap already running\033[0m"
+    else
+            echo -e "\033[31mswap not running\033[0m"
+    fi
+}
 
 case "$1" in
-    start)
-		app_start
-        ;;
-    stop)
+    'start') app_start;;
+    'stop') app_stop;;
+    'reload')
+    'restart') 
         app_stop
-        ;;
-	restart|reload)
-		app_stop
-		sleep 0.3
-		app_start
-		;;
-    *)
-        echo "Please use start or stop as first argument"
-        ;;
-esac
-
+        app_start;;
+    'status') app_status;;
+esac
\ No newline at end of file