1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
package main
import (
"fmt"
"log"
"os"
)
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)
}
if _, err := fmt.Fprintln(f, "SrcIP Address\t\tRX\t\tTX"); err != nil {
log.Fatal(err)
}
for _, v := range data {
if _, err = fmt.Fprintf(f, "%v\t\t%v\t\t%v\n", v.ip, v.rx, v.tx); err != nil {
f.Close()
log.Fatal(err)
}
}
if err = f.Close(); err != nil {
log.Fatal(err)
}
}
func saveTOdatabases(dnm string) {
dbx := initDB(dnm)
defer dbx.Close()
createTable(dbx)
storeItem(dbx, nmap)
if *svtf != "false" {
saveTOfile(*svtf, readItem(dbx))
}
nmap = make(map[string]datausage)
}
|