154 lines
3.1 KiB
Go
154 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
type dumpStruct struct {
|
|
Update string `json:"LastUpdate"`
|
|
GOOD []GOOD `json:"GOOD"`
|
|
BAD []BAD `json:"BAD"`
|
|
MEH []MEH `json:"MEH"`
|
|
STATS []STATS `json:"STATS"`
|
|
}
|
|
type GOOD struct {
|
|
Token string `json:"Token"`
|
|
LastSeen string `json:"LastSeen"`
|
|
Delta string `json:"Age"`
|
|
}
|
|
type BAD struct {
|
|
Token string `json:"Token"`
|
|
LastSeen string `json:"LastSeen"`
|
|
Delta string `json:"Age"`
|
|
}
|
|
type MEH struct {
|
|
Token string `json:"Token"`
|
|
LastSeen string `json:"LastSeen"`
|
|
Delta string `json:"Age"`
|
|
}
|
|
type STATS struct {
|
|
Decision string `json:"Decision"`
|
|
Amount int64 `json:"Amount"`
|
|
}
|
|
|
|
// WriteToFile will print any string of text to a file safely by
|
|
// checking for errors and syncing at the end.
|
|
func writeToFile(filename string, data string) error {
|
|
file, err := os.Create(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
_, err = io.WriteString(file, data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return file.Sync()
|
|
}
|
|
|
|
func handlepanic() {
|
|
|
|
if a := recover(); a != nil {
|
|
fmt.Println("OPS!: Recovering from:", a)
|
|
}
|
|
}
|
|
|
|
func saveBayesToFile() {
|
|
|
|
var tmpJSON, tmpBayes string
|
|
var DumpJson = new(dumpStruct)
|
|
|
|
DumpJson.Update = time.Now().String()
|
|
|
|
log.Println("Trying to write json file")
|
|
|
|
Classifier.BAD.Range(func(key interface{}, value interface{}) bool {
|
|
var t BAD
|
|
v := value.(int64)
|
|
t.Token = key.(string)
|
|
t.LastSeen = time.Unix(0, v).String()
|
|
t.Delta = time.Since(time.Unix(0, v)).String()
|
|
DumpJson.BAD = append(DumpJson.BAD, t)
|
|
return true
|
|
})
|
|
|
|
Classifier.GOOD.Range(func(key interface{}, value interface{}) bool {
|
|
var t GOOD
|
|
v := value.(int64)
|
|
t.Token = key.(string)
|
|
t.LastSeen = time.Unix(0, v).String()
|
|
t.Delta = time.Since(time.Unix(0, v)).String()
|
|
DumpJson.GOOD = append(DumpJson.GOOD, t)
|
|
return true
|
|
})
|
|
|
|
Classifier.MEH.Range(func(key interface{}, value interface{}) bool {
|
|
var t MEH
|
|
v := value.(int64)
|
|
t.Token = key.(string)
|
|
t.LastSeen = time.Unix(0, v).String()
|
|
t.Delta = time.Since(time.Unix(0, v)).String()
|
|
DumpJson.MEH = append(DumpJson.MEH, t)
|
|
return true
|
|
})
|
|
|
|
Classifier.STATS.Range(func(key interface{}, value interface{}) bool {
|
|
var t STATS
|
|
t.Decision = key.(string)
|
|
t.Amount = value.(int64)
|
|
DumpJson.STATS = append(DumpJson.STATS, t)
|
|
|
|
return true
|
|
})
|
|
|
|
if tmpJ, e := json.MarshalIndent(DumpJson, "", " "); e == nil {
|
|
tmpJSON = fmt.Sprintf("%s", tmpJ)
|
|
} else {
|
|
tmpJSON = e.Error()
|
|
}
|
|
|
|
if tmpB, er := json.MarshalIndent(Classifier.bayez.Matrix, "", " "); er == nil {
|
|
tmpBayes = fmt.Sprintf("%s", tmpB)
|
|
} else {
|
|
tmpBayes = er.Error()
|
|
}
|
|
|
|
tmpJSON = fmt.Sprintf("%s\n%s", tmpJSON, tmpBayes)
|
|
|
|
dumpfile := os.Getenv("DUMPFILE")
|
|
if dumpfile == "" {
|
|
dumpfile = "bayes.json"
|
|
}
|
|
|
|
if DebugLog {
|
|
log.Println(tmpJSON)
|
|
} else {
|
|
writeToFile(dumpfile, tmpJSON)
|
|
log.Println("File saved: ", dumpfile)
|
|
}
|
|
|
|
}
|
|
|
|
func jsonEngine() {
|
|
|
|
for {
|
|
log.Println("Zardoz Seniority: ", ProxyFlow.seniority)
|
|
saveBayesToFile()
|
|
time.Sleep(1 * time.Minute)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func init() {
|
|
log.Printf("File Engine Starting")
|
|
|
|
go jsonEngine()
|
|
log.Printf("FIle Engine Started")
|
|
}
|