create Collection.Update stub

Preetam · Aug 24, 2016 · 1dcd1ac · 1dcd1ac
1 parent aa4b8c2
commit 1dcd1ac
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/lm2.go b/lm2.go
@@ -320,6 +320,15 @@ func (c *Collection) Delete(key string) error {
 	return nil
 }
 
+func (c *Collection) Update(wb *WriteBatch) error {
+	// Clean up.
+	wb.cleanup()
+
+	// TODO
+
+	return nil
+}
+
 func NewCollection(file string) (*Collection, error) {
 	f, err := os.OpenFile(file, os.O_CREATE|os.O_RDWR, 0666)
 	if err != nil {

diff --git a/write_batch.go b/write_batch.go
@@ -25,3 +25,9 @@ func (wb *WriteBatch) Set(key, value string) {
 func (wb *WriteBatch) Delete(key string) {
 	wb.deletes[key] = struct{}{}
 }
+
+func (wb *WriteBatch) cleanup() {
+	for key := range wb.deletes {
+		delete(wb.sets, key)
+	}
+}