package batch import ( "fmt" "github.com/vocdoni/go-dvote/types" "github.com/vocdoni/go-dvote/db" "encoding/json" ) var rdb *db.LevelDbStorage var bdb *db.LevelDbStorage var BatchSignal chan bool var BatchSize int var err error func Setup(l *db.LevelDbStorage) { rdb = l.WithPrefix([]byte("relay_")) bdb = l.WithPrefix([]byte("batch_")) } //add (queue for counting) func Add(ballot types.Ballot) error { //this is probably adding [] //err := bdb.Put(fmt.Sprintf("%v", p.Nullifier)),[]byte(fmt.Sprintf("%v", p))) var b []byte var n []byte var err error b, err = json.Marshal(ballot) n, err = json.Marshal(ballot.Nullifier) err = bdb.Put(n,b) if err != nil { return err } //this actually needs to see if it was added if bdb.Count() >= BatchSize { BatchSignal <- true } return nil } //create (return batch) //k is []byte 'batch_' + nullifier //v is []byte package //returns slice of nullifiers, batch json func Fetch() ([]string, []string) { var n []string var b []string iter := bdb.Iter() for iter.Next() { k := iter.Key() v := iter.Value() err := iter.Error() if err != nil { panic(err) } n = append(n, string(k[6:])) b = append(b, string(v)) } iter.Release() // jn, err := json.Marshal(n) // if err != nil { // panic(err) // } // jb, err := json.Marshal(b) // if err != nil { // panic(err) // } return n, b } //move from bdb to rdb once pinned func Compact(n []string) { for _, k := range n { //fmt.Println(k) v, err := bdb.Get([]byte(k)) if err != nil { fmt.Println(err.Error()) } rdb.Put([]byte(k), v) bdb.Delete([]byte(k)) } }