From cdf7edee688e4b6a37b1f33ba6f00c5fce61f85e Mon Sep 17 00:00:00 2001 From: Matt Baer Date: Tue, 17 Feb 2015 22:42:48 -0500 Subject: [PATCH] Move scores data file writing to store pkg --- store/store.go | 57 ++++++++++++++++++++++++++++++++++++++++ tildescores/scores.go | 60 +++++++++++++++---------------------------- 2 files changed, 77 insertions(+), 40 deletions(-) create mode 100644 store/store.go diff --git a/store/store.go b/store/store.go new file mode 100644 index 0000000..1deacb2 --- /dev/null +++ b/store/store.go @@ -0,0 +1,57 @@ +package store + +import ( + "os" + "bufio" + "strings" +) + +type Row struct { + Data []string +} + +func ReadRows(path, delimiter string) *[]Row { + f, _ := os.Open(path) + + defer f.Close() + + rows := []Row{} + s := bufio.NewScanner(f) + s.Split(bufio.ScanLines) + + for s.Scan() { + data := strings.Split(s.Text(), delimiter) + r := &Row{Data: data} + rows = append(rows, *r) + } + + return &rows +} + +func WriteData(path, data, delimeter string) { + f, err := os.OpenFile(path, os.O_CREATE | os.O_RDWR, 0644) + if err != nil { + fmt.Println(err) + } + defer f.Close() + + _, err = f.WriteString(data) + if err != nil { + fmt.Println(err) + } +} + +func WriteRows(path string, rows *[]Row, delimeter string) { + f, err := os.OpenFile(path, os.O_CREATE | os.O_RDWR, 0644) + if err != nil { + fmt.Println(err) + } + defer f.Close() + + for _, r := range rows { + _, err = f.WriteString(strings.Join(r.Data, delimeter)) + if err != nil { + fmt.Println(err) + } + } +} diff --git a/tildescores/scores.go b/tildescores/scores.go index b1dc00e..7b38eb6 100644 --- a/tildescores/scores.go +++ b/tildescores/scores.go @@ -12,6 +12,8 @@ import ( "strings" "io/ioutil" "text/template" + + "github.com/thebaer/tildes/store" ) var ( @@ -41,8 +43,8 @@ func main() { headers := []string{ "User", "Tildes", "Last Collected", "Addiction", "# Asks", "Avg.", "Last Amt." } - scoresData := readData(scoresPath, "&^%") - updatesData := readData(scoreDeltasPath, deltaDelimiter) + scoresData := store.ReadRows(scoresPath, "&^%") + updatesData := store.ReadRows(scoreDeltasPath, deltaDelimiter) scoresData = checkScoreDelta(scoresData, updatesData) scoresTable := buildScoresTable(scoresData, headers) @@ -52,15 +54,11 @@ func main() { type Table struct { Headers []string - Rows []Row + Rows []store.Row } -type Row struct { - Data []string -} - -type By func(r1, r2 *Row) bool -func (by By) Sort(rows []Row) { +type By func(r1, r2 *store.Row) bool +func (by By) Sort(rows []store.Row) { rs := &rowSorter { rows: rows, by: by, @@ -68,8 +66,8 @@ func (by By) Sort(rows []Row) { sort.Sort(rs) } type rowSorter struct { - rows []Row - by func(r1, r2 *Row) bool + rows []store.Row + by func(r1, r2 *store.Row) bool } func (r *rowSorter) Len() int { return len(r.rows) @@ -82,12 +80,12 @@ func (r *rowSorter) Less(i, j int) bool { } func sortScore(table *Table) *Table { - score := func(r1, r2 *Row) bool { + score := func(r1, r2 *store.Row) bool { s1, _ := strconv.Atoi(r1.Data[1]) s2, _ := strconv.Atoi(r2.Data[1]) return s1 < s2 } - decScore := func(r1, r2 *Row) bool { + decScore := func(r1, r2 *store.Row) bool { return !score(r1, r2) } By(decScore).Sort(table.Rows) @@ -136,7 +134,7 @@ type LastScore struct { Addiction int } -func checkScoreDelta(scoreRows *[]Row, deltaRows *[]Row) *[]Row { +func checkScoreDelta(scoreRows, deltaRows *[]store.Row) *[]store.Row { users := make(map[string]LastScore) // Read score delta data @@ -261,38 +259,20 @@ func checkScoreDelta(scoreRows *[]Row, deltaRows *[]Row) *[]Row { return scoreRows } -func buildScoresTable(rows *[]Row, headers []string) *Table { - table := &Table{Headers: headers, Rows: nil} +func buildScoresTable(rows *[]store.Row, headers []string) *Table { + t := &Table{Headers: headers, Rows: nil} const layout = "Jan 2, 2006 3:04pm MST" for i, r := range *rows { data := r.Data - t := parseTimestamp(r.Data[2]) - r.Data[2] = t.UTC().Format(layout) - row := &Row{Data: data} - (*rows)[i] = *row + time := parseTimestamp(r.Data[2]) + r.Data[2] = time.UTC().Format(layout) + outRow := &store.Row{Data: data} + (*rows)[i] = *outRow } - table.Rows = *rows + t.Rows = *rows - return table -} - -func readData(path string, delimiter string) *[]Row { - f, _ := os.Open(path) - - defer f.Close() - - rows := []Row{} - s := bufio.NewScanner(f) - s.Split(bufio.ScanLines) - - for s.Scan() { - data := strings.Split(s.Text(), delimiter) - row := &Row{Data: data} - rows = append(rows, *row) - } - - return &rows + return t } func getFile(path string) string {