From e84fe449590ba6cb80d98397e9dcf349732a975f Mon Sep 17 00:00:00 2001
From: Tobie Morgan Hitchcock <tobie@abcum.com>
Date: Thu, 21 Nov 2019 09:21:29 +0000
Subject: [PATCH] Tidy up live queries on startup

---
 db/db.go     |  1 +
 db/socket.go | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+)

diff --git a/db/db.go b/db/db.go
index b4d25265..f8722887 100644
--- a/db/db.go
+++ b/db/db.go
@@ -59,6 +59,7 @@ func init() {
 // Setup sets up the connection with the data layer
 func Setup(opts *cnf.Options) (err error) {
 	KV = cnf.Settings.DB.Base
+	err = tidy()
 	return
 }
 
diff --git a/db/socket.go b/db/socket.go
index aa88a627..aa04fe0b 100644
--- a/db/socket.go
+++ b/db/socket.go
@@ -25,6 +25,7 @@ import (
 	"github.com/abcum/surreal/cnf"
 	"github.com/abcum/surreal/kvs"
 	"github.com/abcum/surreal/sql"
+	"github.com/abcum/surreal/txn"
 	"github.com/abcum/surreal/util/data"
 	"github.com/abcum/surreal/util/keys"
 	"github.com/abcum/surreal/util/uuid"
@@ -55,6 +56,57 @@ func flush(id string) {
 	}()
 }
 
+// IMPORTANT remove this when distributed
+// We need to remove this when moving
+// to a distributed cluster as
+// websockets might be managed by an
+// alternative server, and should not
+// be removed on node startup.
+
+func tidy() error {
+
+	ctx := context.Background()
+
+	txn, _ := txn.New(ctx, true)
+
+	defer txn.Commit()
+
+	nss, err := txn.AllNS(ctx)
+	if err != nil {
+		return err
+	}
+
+	for _, ns := range nss {
+
+		dbs, err := txn.AllDB(ctx, ns.Name.VA)
+		if err != nil {
+			return err
+		}
+
+		for _, db := range dbs {
+
+			tbs, err := txn.AllTB(ctx, ns.Name.VA, db.Name.VA)
+			if err != nil {
+				return err
+			}
+
+			for _, tb := range tbs {
+
+				key := &keys.LV{KV: KV, NS: ns.Name.VA, DB: db.Name.VA, TB: tb.Name.VA, LV: keys.Ignore}
+				if _, err = txn.ClrP(ctx, key.Encode(), 0); err != nil {
+					return err
+				}
+
+			}
+
+		}
+
+	}
+
+	return nil
+
+}
+
 func (s *socket) ctx() (ctx context.Context) {
 
 	ctx = context.Background()