/
Scanner.go
80 lines (74 loc) · 1.73 KB
/
Scanner.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
package simple_util
import (
"bufio"
"regexp"
"strings"
)
func Scanner2Array(scanner *bufio.Scanner) []string {
var array []string
for scanner.Scan() {
array = append(array, scanner.Text())
}
CheckErr(scanner.Err())
return array
}
func Scanner2Slice(scanner *bufio.Scanner, sep string) [][]string {
var slice [][]string
for scanner.Scan() {
line := scanner.Text()
array := strings.Split(line, sep)
slice = append(slice, array)
}
CheckErr(scanner.Err())
return slice
}
func ScanTitle(scanner *bufio.Scanner, sep string, skip *regexp.Regexp) (title []string) {
for scanner.Scan() {
line := scanner.Text()
if skip != nil && skip.MatchString(line) {
continue
}
title = strings.Split(line, sep)
}
CheckErr(scanner.Err())
return
}
func Scanner2MapArray(scanner *bufio.Scanner, sep string, skip *regexp.Regexp) ([]map[string]string, []string) {
var mapArray []map[string]string
var title = ScanTitle(scanner, sep, skip)
for scanner.Scan() {
line := scanner.Text()
if skip != nil && skip.MatchString(line) {
continue
}
array := strings.Split(line, sep)
var dataHash = make(map[string]string)
for j, k := range array {
dataHash[title[j]] = k
}
mapArray = append(mapArray, dataHash)
}
CheckErr(scanner.Err())
return mapArray, title
}
func Scanner2MapMap(scanner *bufio.Scanner, key, sep string) map[string]map[string]string {
var db = make(map[string]map[string]string)
var title []string
var i = 0
for scanner.Scan() {
line := scanner.Text()
array := strings.Split(line, sep)
if i == 0 {
title = array
} else {
var dataHash = make(map[string]string)
for j, k := range array {
dataHash[title[j]] = k
}
db[dataHash[key]] = dataHash
}
i++
}
CheckErr(scanner.Err())
return db
}