@@ -33,6 +33,7 @@ import (
33
33
"strings"
34
34
35
35
"github.com/abhinav-upadhyay/brickdb/index"
36
+ "github.com/abhinav-upadhyay/brickdb/pkg/brickdb"
36
37
)
37
38
38
39
func main () {
@@ -51,7 +52,7 @@ func main() {
51
52
}
52
53
}
53
54
54
- func openDB (name string ) * index .Brickdb {
55
+ func openDB (name string ) * brickdb .Brickdb {
55
56
finfo , err := os .Stat (name )
56
57
exists := false
57
58
if os .IsNotExist (err ) {
@@ -60,16 +61,16 @@ func openDB(name string) *index.Brickdb {
60
61
if exists {
61
62
exists = ! finfo .IsDir ()
62
63
}
63
- db := index . NewBrick ( )
64
+ db := brickdb . New ( name )
64
65
if exists {
65
- db .Open (name , os .O_RDWR )
66
+ db .Open (os .O_RDWR )
66
67
} else {
67
- db .Open ( name , os . O_RDWR | os . O_CREATE )
68
+ db .Create ( index . HashIndexType )
68
69
}
69
70
return db
70
71
}
71
72
72
- func executeCmd (db * index .Brickdb , cmdArgs string ) bool {
73
+ func executeCmd (db * brickdb .Brickdb , cmdArgs string ) bool {
73
74
args := strings .Split (cmdArgs , " " )
74
75
cmd := args [0 ]
75
76
switch cmd {
@@ -80,7 +81,7 @@ func executeCmd(db *index.Brickdb, cmdArgs string) bool {
80
81
}
81
82
key := args [1 ]
82
83
val := args [2 ]
83
- err := db .Store (key , val , index . INSERT )
84
+ err := db .Store (key , val , brickdb . Insert )
84
85
if err != nil {
85
86
fmt .Printf ("Failed to insert key %s with value %s due to error %v\n " , key , val , err )
86
87
return false
@@ -92,7 +93,7 @@ func executeCmd(db *index.Brickdb, cmdArgs string) bool {
92
93
}
93
94
key := args [1 ]
94
95
val := args [2 ]
95
- err := db .Store (key , val , index . STORE )
96
+ err := db .Store (key , val , brickdb . Update )
96
97
if err != nil {
97
98
fmt .Printf ("Failed to update key %s with value %s due to error %v\n " , key , val , err )
98
99
return false
0 commit comments