Update existing values if found

Author: thespad

root/etc/cont-init.d/20-config

@@ -21,14 +21,30 @@ if [[ ! -e /config/data/system.properties ]]; then
     cp /defaults/system.properties /config/data
 fi
 
-if [[ -n $MEM_LIMIT ]]; then
-    echo "*** Setting Java memory limit to $MEM_LIMIT ***"
-    echo "unifi.xmx=$MEM_LIMIT" >> /config/data/system.properties 
-fi
+if grep -q "xmx" /config/data/system.properties && grep -q "xms" /config/data/system.properties; then
+
+  if [[ -n $MEM_LIMIT ]]; then
+      echo "*** Setting Java memory limit to $MEM_LIMIT ***"
+      sed -i "s/unifi.xmx=.*/unifi.xmx=$MEM_LIMIT/" /config/data/system.properties 
+  fi
+
+  if [[ -n $MEM_STARTUP ]]; then
+      echo "*** Setting Java memory minimum to $MEM_STARTUP ***"
+      sed -i "s/unifi.xms=.*/unifi.xms=$MEM_STARTUP/" /config/data/system.properties
+  fi
+
+else
+
+  if [[ -n $MEM_LIMIT ]]; then
+      echo "*** Setting Java memory limit to $MEM_LIMIT ***"
+      echo "unifi.xmx=$MEM_LIMIT" >> /config/data/system.properties 
+  fi
+
+  if [[ -n $MEM_STARTUP ]]; then
+      echo "*** Setting Java memory minimum to $MEM_STARTUP ***"
+      echo "unifi.xms=$MEM_STARTUP" >> /config/data/system.properties
+  fi
 
-if [[ -n $MEM_STARTUP ]]; then
-    echo "*** Setting Java memory minimum to $MEM_STARTUP ***"
-    echo "unifi.xms=$MEM_STARTUP" >> /config/data/system.properties
 fi
 
 # permissions