First Commit -- adding files

1 files changed, 37 insertions, 0 deletions

diff --git a/files.go b/files.go
new file mode 100644
index 0000000..def0761
--- /dev/null
+++ b/files.go
@@ -0,0 +1,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)
+}