-
Notifications
You must be signed in to change notification settings - Fork 329
/
main.go
95 lines (84 loc) · 2.12 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
package main
import (
"fmt"
"log"
"github.com/nutsdb/nutsdb"
"github.com/xujiajun/utils/strconv2"
"github.com/xujiajun/utils/time2"
)
var (
db *nutsdb.DB
bucket string
)
func init() {
db, _ = nutsdb.Open(
nutsdb.DefaultOptions,
nutsdb.WithDir("/tmp/nutsdbexample/example_batch"),
// nutsdb.WithRWMode(nutsdb.MMap),
// nutsdb.WithSyncEnable(false),
)
}
func createBucket() {
if err := db.Update(
func(tx *nutsdb.Tx) error {
return tx.NewBucket(nutsdb.DataStructureBTree, "bucket1")
}); err != nil {
//log.Fatal(err)
}
}
func main() {
time2.Start()
bucket = "bucket1"
// create bucket first
createBucket()
end := 1
for j := 1; j <= end; j++ {
if err := db.Update(
func(tx *nutsdb.Tx) error {
for i := (j - 1) * 10000; i < j*10000; i++ {
key := []byte("namename" + strconv2.IntToStr(i))
val := []byte("valvalvavalvalvalvavalvalvalvavalvalvalvaval" + strconv2.IntToStr(i))
if err := tx.Put(bucket, key, val, 0); err != nil {
return err
}
}
return nil
}); err != nil {
log.Fatal(err)
}
}
fmt.Println("batch put data cost: ", time2.End())
time2.Start()
wb, err := db.NewWriteBatch()
if err != nil {
fmt.Println("NewWriteBatch() fail, err=", err)
}
N := 10000
for i := 0; i < N; i++ {
key := []byte("namename" + strconv2.IntToStr(i))
val := []byte("valvalvavalvalvalvavalvalvalvavalvalvalvaval" + strconv2.IntToStr(i))
if err := wb.Put(bucket, key, val, 0); err != nil {
fmt.Println("batch write entry fail, err=", err)
log.Fatal(err)
}
}
if err := wb.Flush(); err != nil {
fmt.Println("batch write flush fail, err=", err)
}
fmt.Printf("Time taken via batch write: %v\n", time2.End())
time2.Start()
if err := wb.Reset(); err != nil {
fmt.Println("batch write Reset fail, err=", err)
}
for i := 0; i < N; i++ {
key := []byte("namename" + strconv2.IntToStr(i))
if err := wb.Delete(bucket, key); err != nil {
fmt.Println("batch delete entry fail, err=", err)
log.Fatal(err)
}
}
if err := wb.Flush(); err != nil {
fmt.Println("batch delete flush fail, err=", err)
}
fmt.Printf("Time taken via batch delete: %v\n", time2.End())
}