forked from cwtch.im/cwtch
94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
package metrics
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"git.openprivacy.ca/openprivacy/libricochet-go/application"
|
|
"github.com/struCoder/pidusage"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
reportFile = "serverMonitorReport.txt"
|
|
)
|
|
|
|
// Monitors is a package of metrics for a Cwtch Server including message count, CPU, Mem, and conns
|
|
type Monitors struct {
|
|
MessageCounter Counter
|
|
Messages MonitorHistory
|
|
CPU MonitorHistory
|
|
Memory MonitorHistory
|
|
ClientConns MonitorHistory
|
|
starttime time.Time
|
|
breakChannel chan bool
|
|
}
|
|
|
|
// Start initializes a Monitors's monitors
|
|
func (mp *Monitors) Start(ra *application.RicochetApplication) {
|
|
mp.starttime = time.Now()
|
|
mp.breakChannel = make(chan bool)
|
|
mp.MessageCounter = NewCounter()
|
|
mp.Messages = NewMonitorHistory(Count, func() (c float64) { c = float64(mp.MessageCounter.Count()); mp.MessageCounter.Reset(); return })
|
|
mp.CPU = NewMonitorHistory(Percent, func() float64 { sysInfo, _ := pidusage.GetStat(os.Getpid()); return float64(sysInfo.CPU) })
|
|
mp.Memory = NewMonitorHistory(MegaBytes, func() float64 { sysInfo, _ := pidusage.GetStat(os.Getpid()); return float64(sysInfo.Memory) })
|
|
mp.ClientConns = NewMonitorHistory(Count, func() float64 { return float64(ra.ConnectionCount()) })
|
|
|
|
// Todo: replace with proper reporting
|
|
go mp.log()
|
|
}
|
|
|
|
func (mp *Monitors) log() {
|
|
for {
|
|
select {
|
|
case <-time.After(time.Minute):
|
|
messageMinutes := mp.Messages.Minutes()
|
|
cpuMinutes := mp.CPU.Minutes()
|
|
memoryMinutes := mp.Memory.Minutes()
|
|
listenConnsMinutes := mp.ClientConns.Minutes()
|
|
log.Printf("METRICS: Messages: %.0f ClientConns: %.0f CPU: %.2f Memory: %.0fMBs\n", messageMinutes[0], listenConnsMinutes[0], cpuMinutes[0], memoryMinutes[0]/1024/1024)
|
|
mp.report()
|
|
case <-mp.breakChannel:
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (mp *Monitors) report() {
|
|
f, err := os.Create(reportFile)
|
|
if err != nil {
|
|
log.Println("ERROR: Could not open monitor reporting file: ", err)
|
|
return
|
|
}
|
|
defer f.Close()
|
|
|
|
w := bufio.NewWriter(f)
|
|
|
|
fmt.Fprintf(w, "Started: %v\n\n", mp.starttime)
|
|
|
|
fmt.Fprintln(w, "Messages:")
|
|
mp.Messages.Report(w)
|
|
|
|
fmt.Fprintln(w, "\nClient Connections:")
|
|
mp.ClientConns.Report(w)
|
|
|
|
fmt.Fprintln(w, "\nCPU:")
|
|
mp.CPU.Report(w)
|
|
|
|
fmt.Fprintln(w, "\nMemory:")
|
|
mp.Memory.Report(w)
|
|
|
|
w.Flush()
|
|
|
|
}
|
|
|
|
// Stop stops all the monitors in a Monitors
|
|
func (mp *Monitors) Stop() {
|
|
mp.breakChannel <- true
|
|
mp.Messages.Stop()
|
|
mp.CPU.Stop()
|
|
mp.Memory.Stop()
|
|
mp.ClientConns.Stop()
|
|
}
|