Add config to lesci experiment

1 files changed, 12 insertions, 1 deletions

diff --git a/research/lesci/context.go b/research/lesci/context.go
index df9b059..8496507 100644
--- a/research/lesci/context.go
+++ b/research/lesci/context.go
@@ -1,6 +1,7 @@
 package lesci
 
 import (
+	"context"
 	"database/sql"
 	"database/sql/driver"
 	"fmt"
@@ -39,7 +40,17 @@ func (e *Experiment) BuildContext(db *sql.DB) error {
 	} else if !ok {
 		fmt.Println("context table not empty")
 
-		return nil
+		if !e.options.ForceContext {
+			fmt.Println("skipping context collection")
+
+			return nil
+		}
+
+		fmt.Println("clearing context")
+
+		if _, err := db.ExecContext(context.Background(), `DELETE FROM context`); err != nil {
+			return err
+		}
 	}
 
 	return AppendRows(db, "context", func(append AppendFunc) error {