-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.go
89 lines (71 loc) · 2.18 KB
/
db.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
package db
import (
"context"
"github.com/MarioLegenda/cig/internal/db/conditionResolver"
"github.com/MarioLegenda/cig/internal/db/selectedColumnMetadata"
job2 "github.com/MarioLegenda/cig/internal/job"
"github.com/MarioLegenda/cig/internal/syntax"
"os"
"time"
)
type fileMetadata struct {
columns metadataColumns
originalPath string
}
type db struct {
openFs *os.File
metadata fileMetadata
}
type DB interface {
Run(s syntax.Structure) Data
Close() error
}
type Data struct {
SelectedColumns []string
AllColumns []string
Error error
Data []map[string]string
}
func (d *db) Run(s syntax.Structure) Data {
file := s.FileDB()
fileHandler, err := prepareRun(file, d)
if err != nil {
return newData(nil, nil, nil, err)
}
fsMetadata := d.metadata
conditionColumnMetadata := createConditionColumnMetadata(fsMetadata)
selectedColumns := createSelectedColumnMetadata(s, fsMetadata)
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
defer cancel()
res, err := job2.SearchFactory(selectedColumns, conditionColumnMetadata, s.Condition(), s.Constraints(), fileHandler)(0, ctx)
if err != nil {
return newData(selectedColumns.Names(), fsMetadata.columns.names(), nil, err)
}
return newData(selectedColumns.Names(), fsMetadata.columns.names(), res, nil)
}
func (d *db) Close() error {
return d.openFs.Close()
}
func New() DB {
return &db{}
}
func createConditionColumnMetadata(fsMetadata fileMetadata) conditionResolver.ColumnMetadata {
positions := make([]int, len(fsMetadata.columns))
columnNames := make([]string, len(fsMetadata.columns))
for i, m := range fsMetadata.columns {
positions[i] = m.position
columnNames[i] = m.name
}
return conditionResolver.NewColumnMetadata(positions, columnNames)
}
func createSelectedColumnMetadata(structure syntax.Structure, fsMetadata fileMetadata) selectedColumnMetadata.ColumnMetadata {
return selectedColumnMetadata.New(structure.Column().Columns(), fsMetadata.columns.names())
}
func newData(selected, all []string, data []map[string]string, err error) Data {
return Data{
SelectedColumns: selected,
AllColumns: all,
Error: err,
Data: data,
}
}