db: ensure vm interupt is cleared

Signed-off-by: Sander Pick <sanderpick@gmail.com>
textileio · Feb 26, 2022 · 9fc2d3d · 9fc2d3d
1 parent b74267e
commit 9fc2d3d
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 3 deletions.
diff --git a/db/collection.go b/db/collection.go
@@ -44,7 +44,7 @@ var (
 
 	baseKey = dsPrefix.ChildString("collection")
 
-	vmTimeout = time.Second * 30
+	vmTimeout = time.Second
 )
 
 const (
@@ -341,13 +341,15 @@ func (c *Collection) validWrite(identity thread.PubKey, e core.Event) error {
 			return fmt.Errorf("parsing instance in validate write: %v", err)
 		}
 	}
-	time.AfterFunc(vmTimeout, func() {
+	c.vm.ClearInterrupt()
+	timer := time.AfterFunc(vmTimeout, func() {
 		c.vm.Interrupt("validator timed out")
 	})
 	res, err := c.writeValidator(nil, writer, event, inv)
 	if err != nil {
 		return fmt.Errorf("running write validator func: %v", err)
 	}
+	timer.Stop()
 	out := res.Export()
 	switch out.(type) {
 	case bool:
@@ -377,13 +379,15 @@ func (c *Collection) filterRead(identity thread.PubKey, instance []byte) ([]byte
 	if err != nil {
 		return nil, fmt.Errorf("parsing instance in filter read: %v", err)
 	}
-	time.AfterFunc(vmTimeout, func() {
+	c.vm.ClearInterrupt()
+	timer := time.AfterFunc(vmTimeout, func() {
 		c.vm.Interrupt("filter timed out")
 	})
 	res, err := c.readFilter(nil, reader, inv)
 	if err != nil {
 		return nil, fmt.Errorf("running read filter func: %v", err)
 	}
+	timer.Stop()
 	out := res.Export()
 	switch out.(type) {
 	case nil:

diff --git a/db/collection_test.go b/db/collection_test.go
@@ -158,6 +158,21 @@ func TestNewCollection(t *testing.T) {
 		if err == nil {
 			t.Fatal("should get validator timed out error")
 		}
+
+		// test clear interupt
+		c, err = db.UpdateCollection(CollectionConfig{
+			Name:   "Dog",
+			Schema: util.SchemaFromInstance(&Dog{}, false),
+			WriteValidator: `
+				return true
+			`,
+		})
+		checkErr(t, err)
+		id, err = c.Create(util.JSONFromInstance(dog))
+		checkErr(t, err)
+		time.Sleep(time.Second) // sleep vm timeout
+		id, err = c.Create(util.JSONFromInstance(dog))
+		checkErr(t, err)
 	})
 	t.Run("WithReadFilter", func(t *testing.T) {
 		t.Parallel()
@@ -196,6 +211,21 @@ func TestNewCollection(t *testing.T) {
 		if err == nil {
 			t.Fatal("should get filter timed out error")
 		}
+
+		// test clear interupt
+		c, err = db.UpdateCollection(CollectionConfig{
+			Name:   "Dog",
+			Schema: util.SchemaFromInstance(&Dog{}, false),
+			ReadFilter: `
+				return instance
+			`,
+		})
+		checkErr(t, err)
+		_, err = c.FindByID(id)
+		checkErr(t, err)
+		time.Sleep(time.Second) // sleep vm timeout
+		_, err = c.FindByID(id)
+		checkErr(t, err)
 	})
 	t.Run("SingleExpandedSchemaStruct", func(t *testing.T) {
 		t.Parallel()