Newer
Older
package main
import (
"fmt"
"io/ioutil"
"os"
"testing"
"github.com/Sirupsen/logrus"
"github.com/dgraph-io/dgraph/loader"
"github.com/dgraph-io/dgraph/posting"
"github.com/dgraph-io/dgraph/store"
"github.com/dgraph-io/dgraph/store/rocksdb"
"github.com/dgraph-io/dgraph/uid"
"github.com/dgryski/go-farm"
)
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
logrus.SetLevel(logrus.DebugLevel)
rootDir, err := ioutil.TempDir("", "storetest_")
if err != nil {
t.Error(err)
return
}
defer os.RemoveAll(rootDir)
dir1, err := ioutil.TempDir(rootDir, "dir_")
if err != nil {
t.Error(err)
return
}
defer os.RemoveAll(dir1)
dir2, err := ioutil.TempDir(rootDir, "dir_")
if err != nil {
t.Error(err)
return
}
defer os.RemoveAll(dir2)
destDir, err := ioutil.TempDir("", "dest_")
if err != nil {
t.Error(err)
return
}
defer os.RemoveAll(destDir)
ps1 := new(store.Store)
ps1.Init(dir1)
ps2 := new(store.Store)
ps2.Init(dir2)
list := []string{"alice", "bob", "mallory", "ash", "man", "dgraph",
"ash", "alice"}
var numInstances uint64 = 2
posting.Init(nil)
uid.Init(ps1)
loader.Init(nil, ps1)
for _, str := range list {
if farm.Fingerprint64([]byte(str))%numInstances == 0 {
_, err := uid.GetOrAssign(str, 0, numInstances)
if err != nil {
fmt.Errorf("error while assigning uid")
}
}
}
uid.Init(ps2)
loader.Init(nil, ps2)
for _, str := range list {
if farm.Fingerprint64([]byte(str))%numInstances == 1 {
uid.Init(ps2)
_, err := uid.GetOrAssign(str, 1, numInstances)
if err != nil {
fmt.Errorf("error while assigning uid")
}
}
}
posting.MergeLists(100)
ps1.Close()
ps2.Close()
var opt *rocksdb.Options
var ropt *rocksdb.ReadOptions
opt = rocksdb.NewOptions()
ropt = rocksdb.NewReadOptions()
db, err := rocksdb.Open(destDir, opt)
it := db.NewIterator(ropt)
count := 0
for it.SeekToFirst(); it.Valid(); it.Next() {
count++
}
if count != 6 { // There are totally 6 unique strings
fmt.Errorf("Not all the items have been assigned uid")
}
}