Changed file destinations

2016-03-29 02:01:54 +03:00
parent cd69c063b9
commit 00090a0fdc
4 changed files with 0 additions and 0 deletions
--- a/saving.go
+++ b/saving.go
@@ -0,0 +1,204 @@
+// saving.go
+package main
+
+import (
+	"compress/gzip"
+	"encoding/csv"
+	"encoding/json"
+	"io"
+	"log"
+	"os"
+	"sort"
+	"strconv"
+)
+
+func SaveToCsvWriter(data []StatForTime, keyMap map[uint8]string, writerOut io.Writer, fullExport bool) {
+
+	numKeysInt := make([]int, 0)
+	for key := range keyMap {
+		numKeysInt = append(numKeysInt, int(key))
+	}
+	sort.Ints(numKeysInt)
+	numKeys := make([]uint8, 0)
+	for _, key := range numKeysInt {
+		numKeys = append(numKeys, uint8(key))
+	}
+
+	titleLine := make([]string, 0)
+	titleLine = append(titleLine, "Time")
+	if fullExport {
+		for _, key := range numKeys {
+			titleLine = append(titleLine, keyMap[key])
+		}
+	}
+	titleLine = append(titleLine, "Sum")
+
+	table := make([][]string, 0)
+	table = append(table, titleLine)
+	for _, rec := range data {
+		line := make([]string, 0)
+		line = append(line, strconv.Itoa(int(rec.time)))
+		var sum int
+		for _, key := range numKeys {
+			sum += rec.keys[key]
+			if fullExport {
+				line = append(line, strconv.Itoa(rec.keys[key]))
+			}
+		}
+		line = append(line, strconv.Itoa(sum))
+		table = append(table, line)
+	}
+
+	writer := csv.NewWriter(writerOut)
+	writer.WriteAll(table)
+}
+
+func SaveToCsvFile(data []StatForTime, keyMap map[uint8]string, path string, fullExport bool) {
+	csvfile, err := os.Create(path)
+	if err != nil {
+		log.Fatal(err)
+		return
+	}
+	defer csvfile.Close()
+
+	SaveToCsvWriter(data, keyMap, csvfile, fullExport)
+}
+
+func SaveToJSONWriter(data []StatForTime, keyMap map[uint8]string, writerOut io.Writer, fullExport bool) {
+	type JSONStatForTime struct {
+		Time int64
+		Keys map[string]int
+	}
+
+	table := make([]JSONStatForTime, len(data))
+	for i, stat := range data {
+		table[i].Keys = make(map[string]int)
+
+		table[i].Time = stat.time
+		var sum int
+		for numKey, key := range keyMap {
+			if fullExport {
+				table[i].Keys[key] = stat.keys[numKey]
+			}
+			sum += stat.keys[numKey]
+		}
+		table[i].Keys["sum"] = sum
+	}
+
+	outString, err := json.Marshal(table)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	writerOut.Write(outString)
+}
+
+func SaveToJSONFile(data []StatForTime, keyMap map[uint8]string, path string, fullExport bool) {
+	jsonFile, err := os.Create(path)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer jsonFile.Close()
+
+	SaveToJSONWriter(data, keyMap, jsonFile, fullExport)
+}
+
+func SaveToJSLWriter(data []StatForTime, keyMap map[uint8]string, writerOut io.Writer, fullExport bool) {
+	type JSLStatForTime struct {
+		Time int64
+		Keys map[string]int
+	}
+
+	table := make([]JSLStatForTime, len(data))
+	for i, stat := range data {
+		table[i].Keys = make(map[string]int)
+
+		table[i].Time = stat.time
+		var sum int
+		for numKey, key := range keyMap {
+			if fullExport {
+				table[i].Keys[key] = stat.keys[numKey]
+			}
+			sum += stat.keys[numKey]
+		}
+		table[i].Keys["sum"] = sum
+	}
+
+	for ind, line := range table {
+		lineBytes, err := json.Marshal(line)
+		if err != nil {
+			log.Fatal(err)
+		}
+		writerOut.Write(lineBytes)
+		if ind != len(table)-1 {
+			writerOut.Write([]byte("\n"))
+		}
+	}
+}
+
+func SaveToJSLFile(data []StatForTime, keyMap map[uint8]string, path string, fullExport bool) {
+	jslFile, err := os.Create(path)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer jslFile.Close()
+
+	SaveToJSLWriter(data, keyMap, jslFile, fullExport)
+}
+
+func SaveToCsvGzWriter(data []StatForTime, keyMap map[uint8]string, writerOut io.Writer, fullExport bool) {
+	gzipWriter := gzip.NewWriter(writerOut)
+	defer gzipWriter.Close()
+
+	SaveToCsvWriter(data, keyMap, gzipWriter, fullExport)
+}
+
+func SaveToCsvGzFile(data []StatForTime, keyMap map[uint8]string, path string, fullExport bool) {
+	jsonFile, err := os.Create(path)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer jsonFile.Close()
+
+	gzipWriter := gzip.NewWriter(jsonFile)
+	defer gzipWriter.Close()
+
+	SaveToCsvWriter(data, keyMap, gzipWriter, fullExport)
+}
+
+func SaveToJSONGzWriter(data []StatForTime, keyMap map[uint8]string, writerOut io.Writer, fullExport bool) {
+	gzipWriter := gzip.NewWriter(writerOut)
+	defer gzipWriter.Close()
+
+	SaveToJSONWriter(data, keyMap, gzipWriter, fullExport)
+}
+
+func SaveToJSONGzFile(data []StatForTime, keyMap map[uint8]string, path string, fullExport bool) {
+	jsonFile, err := os.Create(path)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer jsonFile.Close()
+
+	gzipWriter := gzip.NewWriter(jsonFile)
+	defer gzipWriter.Close()
+
+	SaveToJSONWriter(data, keyMap, gzipWriter, fullExport)
+}
+
+func SaveToJSLGzWriter(data []StatForTime, keyMap map[uint8]string, writerOut io.Writer, fullExport bool) {
+	gzipWriter := gzip.NewWriter(writerOut)
+	defer gzipWriter.Close()
+
+	SaveToJSLWriter(data, keyMap, gzipWriter, fullExport)
+}
+
+func SaveToJSLGzFile(data []StatForTime, keyMap map[uint8]string, path string, fullExport bool) {
+	jslFile, err := os.Create(path)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer jslFile.Close()
+
+	SaveToJSLGzWriter(data, keyMap, jslFile, fullExport)
+}