-
-
Notifications
You must be signed in to change notification settings - Fork 36
/
index.go
94 lines (77 loc) · 2.16 KB
/
index.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
package file
import (
"fmt"
"strings"
v2 "github.com/shortlink-org/shortlink/boundaries/shortdb/shortdb/domain/index/v1"
v1 "github.com/shortlink-org/shortlink/boundaries/shortdb/shortdb/domain/query/v1"
"github.com/shortlink-org/shortlink/boundaries/shortdb/shortdb/engine/file/index"
parser "github.com/shortlink-org/shortlink/boundaries/shortdb/shortdb/parser/v1"
)
func (f *File) CreateIndex(query *v1.Query) error {
t := f.database.GetTables()[query.GetTableName()]
if t.GetIndex() == nil {
t.Index = make(map[string]*v2.Index)
}
// check
for i := range query.GetIndexs() {
if t.GetIndex()[query.GetIndexs()[i].GetName()] != nil {
return &CreateExistIndexError{Name: query.GetIndexs()[i].GetName()}
}
}
// create
for i := range query.GetIndexs() {
// create index
t.Index[query.GetIndexs()[i].GetName()] = &v2.Index{
Name: query.GetIndexs()[i].GetName(),
Type: query.GetIndexs()[i].GetType(),
Fields: query.GetIndexs()[i].GetFields(),
}
// get all values
// TODO: use pattern iterator
cmd, err := parser.New(fmt.Sprintf("SELECT %s from %s", strings.Join(query.GetIndexs()[i].GetFields(), ","), query.GetTableName()))
if err != nil {
return err
}
rows, err := f.Select(cmd.GetQuery())
if err != nil { //nolint:staticcheck // ignore
// NOTE: ignore empty table
}
// build index
tree, err := index.New(t.GetIndex()[query.GetIndexs()[i].GetName()], rows)
if err != nil {
return err
}
// save to file
payload, err := tree.Marshal()
if err != nil {
return err
}
// save date
err = f.saveData(query, payload, i)
if err != nil {
return err
}
}
return nil
}
func (f *File) saveData(query *v1.Query, payload []byte, i int) error {
openFile, err := f.createFile(fmt.Sprintf("%s_%s_%s.index.json", f.database.GetName(), query.GetTableName(), query.GetIndexs()[i].GetName()))
if err != nil {
return err
}
defer func() {
_ = openFile.Close() // #nosec
}()
// Write something
err = f.writeFile(openFile.Name(), payload)
if err != nil {
return err
}
return nil
}
func (f *File) DropIndex(name string) error {
f.mu.Lock()
defer f.mu.Unlock()
// TODO implement me
panic("implement me")
}