-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmain.go
120 lines (97 loc) · 2.05 KB
/
main.go
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package main
import (
"encoding/csv"
"errors"
"io/ioutil"
"log"
"os"
"strings"
)
const (
dDir = "/data"
rDir = "/result"
)
var (
results []string
resultsExtracted []dataExtract
)
type dataExtract struct {
date string
nickname string
data string
}
func checkError(e error) {
if e != nil {
panic(e)
}
}
func main() {
// Get Work Directory
wd, err := os.Getwd()
checkError(err)
// Read Directory First
log.Println("Directory:", wd+dDir+"/donate")
files, err := ioutil.ReadDir(wd + dDir + "/donate")
checkError(err)
flen := len(files)
log.Println("Files Count:", flen)
if flen <= 0 {
checkError(errors.New("no such files"))
}
// Read all files
for _, f := range files {
f, err := ioutil.ReadFile(wd + dDir + "/donate/" + f.Name())
checkError(err)
sf := strings.Split(string(f), "\n")
results = append(results, sf...)
}
log.Println("Result Count:", len(results))
// Extract to struct
err = extract(results)
checkError(err)
log.Println("Data Extracted")
// Convert to csv
err = convertToCsv(resultsExtracted)
checkError(err)
log.Println("Csv Successfully Created")
}
func extract(d []string) error {
if len(d) == 0 {
return errors.New("data can not be null")
}
// loop data to be extracted
for _, ld := range d {
if len(ld) == 0 {
continue
}
extLd := strings.Split(ld, " ")
ext := dataExtract{
date: strings.Join(extLd[0:2], " "),
nickname: extLd[3],
data: strings.Join(extLd[5:], " "),
}
resultsExtracted = append(resultsExtracted, ext)
}
return nil
}
func convertToCsv(d []dataExtract) error {
wd, err := os.Getwd()
checkError(err)
// Create csv result file
file, err := os.Create(wd + rDir + "/result.csv")
checkError(err)
defer file.Close()
writer := csv.NewWriter(file)
defer writer.Flush()
// Write header to csv
h := []string{"Date", "Nickname", "Value"}
err = writer.Write(h)
checkError(err)
// Store data to csv
for _, value := range d {
dt := []string{value.date, value.nickname, value.data}
err := writer.Write(dt)
checkError(err)
}
return nil
}