From da3cd5bfe53f2e1794bca1bb7f6be7b2e12e9396 Mon Sep 17 00:00:00 2001
From: Yeshodhan Kulkarni <yashodhan@securiace.com>
Date: Thu, 19 Nov 2015 01:40:22 +0530
Subject: [PATCH] Automate Swap Partition Setup
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Script created when —setup-swap is provided as an command line argument
to setup_frappe.sh, based on available RAM size swap will be created if
exists already, it will be ignored and setup will be continued as
normally.
---
 README.md                       |   6 ++
 install_scripts/setup_frappe.sh | 113 +++++++++++++++++++++++++++++++-
 2 files changed, 118 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index bbb9a09c..17ee8c8a 100644
--- a/README.md
+++ b/README.md
@@ -26,6 +26,12 @@ This script should install the pre-requisites, install bench and setup an ERPNex
 
 If you want to develop ERPNext or any Frappe App, you can omit the "--setup-production" part from the command.
 
+For advanced users, `setup_frappe.sh` script offers following options:
+```
+sudo bash setup_frappe.sh --setup-swap --setup-production
+```
+This will check if your system does require swap partition, if required it will create new one, otherwise it will skip the swap setup part and continue with rest of the setup instructions.
+
 
 Manual Install
 --------------
diff --git a/install_scripts/setup_frappe.sh b/install_scripts/setup_frappe.sh
index 9a90f815..16c7d7c2 100755
--- a/install_scripts/setup_frappe.sh
+++ b/install_scripts/setup_frappe.sh
@@ -16,7 +16,7 @@ get_passwd() {
 }
 
 set_opts () {
-	OPTS=`getopt -o v --long verbose,mysql-root-password:,frappe-user:,bench-branch:,setup-production,skip-setup-bench,help -n 'parse-options' -- "$@"`
+	OPTS=`getopt -o v --long verbose,mysql-root-password:,frappe-user:,bench-branch:,setup-production,skip-setup-bench,setup-swap,help -n 'parse-options' -- "$@"`
 
 	if [ $? != 0 ] ; then echo "Failed parsing options." >&2 ; exit 1 ; fi
 
@@ -28,6 +28,7 @@ set_opts () {
 	BENCH_BRANCH="master"
 	SETUP_PROD=false
 	SETUP_BENCH=true
+	SETUP_SWAP=false
 
 	if [ -f ~/frappe_passwords.sh ]; then
 		source ~/frappe_passwords.sh
@@ -50,6 +51,7 @@ set_opts () {
 	--setup-production ) SETUP_PROD=true; shift;;
 	--bench-branch ) BENCH_BRANCH="$2"; shift;;
 	--skip-setup-bench ) SETUP_BENCH=false; shift;;
+	--setup-swap ) SETUP_SWAP=true; shift;;
 	-- ) shift; break ;;
 	* ) break ;;
 	esac
@@ -101,6 +103,112 @@ run_cmd() {
 	fi
 }
 
+## setup swap
+setup_swap() {
+	check_memory_and_swap() {
+		mem_count=$(free -m|grep Mem|awk '{print $2}')
+		swap_count=$(free -m|grep Swap|awk '{print $2}')
+		if  [ "$mem_count" -le 2048 ]
+			then
+			if [ "$swap_count" -ne 0 ]
+				then
+				remove_old_swap
+			fi
+				create_swap $((mem_count*3))
+		else
+			if [ "$swap_count" -ne 0 ]
+				then
+				remove_old_swap
+			fi
+			create_swap $((mem_count+2048))
+		fi
+	}
+
+	create_swap() {
+		root_disk_size=$(df -m|grep -w "/"|awk '{print $4}'|head -1)
+		if [ "$1" -gt "$((root_disk_size-1024))" ]
+			then
+			echo "The root disk partition has no space for $1M swap file. Returing to frappe installation."
+			return 0
+		fi
+		if [ ! -e $swapfile ]
+			then
+			run_cmd sudo fallocate -l ${1}M $swapfile
+			run_cmd sudo chmod 600 $swapfile
+			run_cmd sudo mkswap $swapfile
+			run_cmd sudo swapon $swapfile
+			echo "The swap partition setup successful."
+		else
+			echo "The /swapfile already exists. Returing to frappe installation."
+			return 0
+		fi
+	}
+
+	remove_old_swap() {
+		old_swap_file=$(grep swap $fstab|grep -v "#"|awk '{print $1}')
+		run_cmd sudo swapoff $old_swap_file
+		run_cmd sudo cp -f $fstab ${fstab}_bak
+		run_cmd sudo sed -i '/swap/d' $fstab
+	}
+
+	adjust_swappiness() {
+		echo "Adjusting swappiness & vfs_cache_pressure"
+		run_cmd sudo sed '/vm\.swappiness/d' -ibak /etc/sysctl.conf
+		echo "vm.swappiness = 10" >> /etc/sysctl.conf
+		run_cmd sudo sysctl vm.swappiness=10
+
+		run_cmd sudo sed '/vm\.vfs_cache_pressure/d' -ibak /etc/sysctl.conf
+		echo "vm.vfs_cache_pressure = 50" >> /etc/sysctl.conf
+		run_cmd sudo sysctl vm.vfs_cache_pressure=50
+	}
+
+	config_centos_fstab() {
+		if ! grep $swapfile $fstab >/dev/null 2>&1
+			then
+			echo "Begin to modify $fstab."
+			echo "$swapfile	 swap	 swap defaults 0 0" >>$fstab
+			adjust_swappiness
+		else
+			echo "/etc/fstab is already configured. Returing to frappe installation."
+			return 0
+		fi
+	}
+
+	config_debian_fstab() {
+		if ! grep $swapfile $fstab >/dev/null 2>&1
+			then
+			echo "Begin to modify $fstab."
+			echo "$swapfile	 none	 swap sw 0 0" >>$fstab
+			adjust_swappiness
+		else
+			echo "/etc/fstab is already configured. Returing to frappe installation."
+			return 0
+		fi
+	}
+
+#################### swap start ####################
+swapfile=/swapfile
+fstab=/etc/fstab
+
+echo "Check the memory and swap."
+check_memory_and_swap
+
+echo "Making swap partition persistant by adding it to $fstab."
+case "$OS" in
+	centos)
+	config_centos_fstab
+	;;
+	Ubuntu|debian)
+	config_debian_fstab
+	;;
+	*)
+	echo "Unable to identify OS, ignoring swap setup instructions."
+	return 0
+	;;
+esac
+echo "Returing to frappe installation."
+}
+
 ## add repos
 
 add_centos6_mariadb_repo() {
@@ -428,6 +536,9 @@ add_user() {
 main() {
 	set_opts $@
 	get_distro
+	if $SETUP_SWAP; then
+		setup_swap
+	fi
 	add_maria_db_repo
 	echo Installing packages for $OS\. This might take time...
 	install_packages