free memory

fhmq · Sep 12, 2017 · 6b50060 · 6b50060
1 parent 9627799
commit 6b50060
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/broker/broker.go b/broker/broker.go
@@ -11,6 +11,7 @@ import (
 	"time"
 
 	"github.com/eclipse/paho.mqtt.golang/packets"
+	"github.com/shirou/gopsutil/mem"
 
 	"golang.org/x/net/websocket"
 
@@ -84,12 +85,14 @@ func (b *Broker) Start() {
 }
 
 func StateMonitor() {
-	// v, _ := mem.VirtualMemory()
-	timeSticker := time.NewTicker(time.Second * 30)
+	v, _ := mem.VirtualMemory()
+	timeSticker := time.NewTicker(time.Second * 5)
 	for {
 		select {
 		case <-timeSticker.C:
-			debug.FreeOSMemory()
+			if v.UsedPercent > 0.8 {
+				debug.FreeOSMemory()
+			}
 			// fmt.Printf("Total: %v, Free:%v, UsedPercent:%f%%\n", v.Total, v.Free, v.UsedPercent)
 		}
 	}