package main
import (
"fmt"
"log"
"os"
"sort"
"strconv"
"time"
)
type entry struct /* sorting datausage */ {
key string
value datausage
}
type byPriority []entry
func (d byPriority) Len() int { return len(d) }
func (d byPriority) Less(i, j int) bool {
// sort by rx or tx ?
if *srtx == "RX" {
return d[i].value.rx < d[j].value.rx
}
return d[i].value.tx < d[j].value.tx
}
func (d byPriority) Swap(i, j int) { d[i], d[j] = d[j], d[i] }
func saveTOdatabases(dnm string) /*save captured data to file and database */ {
dbx := initDB(dnm) // connect to database
defer dbx.Close()
createTable(dbx) // create tb if not exist
storeItem(dbx, nmap) // save to db
if *svtf != "false" {
saveTOfile(*svtf, readItem(dbx)) // save to file or not ?
}
nmap = make(map[string]datausage) // flushing ...
}
func saveTOfile(filename string, data map[string]datausage) {
f, err := os.OpenFile(filename, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
if err != nil {
log.Fatal(err)
}
var maxLenrx int
slice := make(byPriority, 0, len(data))
for key, value := range data {
slice = append(slice, entry{key, value})
if len(strconv.FormatUint(uint64(value.rx), 10)) > maxLenrx {
maxLenrx = len(strconv.FormatUint(uint64(value.rx), 10))
}
}
// printing header to file
if _, err := fmt.Fprintf(f, "Go-IPFM 0.34 - Capturing Data On %v - Time: %v\n%-28v %-*v %v(%s)\n", *infc, time.Now().Format("2006-01-02 15:04:05"), "Source-IPV4-Add", maxLenrx+12, "RX("+*trm+")", "TX", *trm); err != nil {
log.Fatal(err)
}
// sorting data usage
if *srfm == "descending" {
sort.Sort(sort.Reverse(slice))
} else {
sort.Sort(slice)
}
for _, v := range slice { /*save sorted data to file ...*/
if _, err = fmt.Fprintf(f, "%-28v %-*.3f %.3f\n", v.value.ip, maxLenrx+12, float64(v.value.rx)/float64(cHz), float64(v.value.tx)/float64(cHz)); err != nil {
log.Fatal(err)
}
// fmt.Println(cHz)
}
if err = f.Close(); err != nil {
log.Fatal(err)
}
}