go: statspro: Add session lifecycle callbacks to integrate with sessi…

…on-aware GC safepoint controller.
dolthub · Feb 12, 2025 · 349d147 · 349d147
1 parent 846b67e
commit 349d147
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/go/libraries/doltcore/sqle/statspro/auto_refresh.go b/go/libraries/doltcore/sqle/statspro/auto_refresh.go
@@ -76,6 +76,10 @@ func (p *Provider) InitAutoRefreshWithParams(ctxFactory func(ctx context.Context
 				}
 
 				dSess := dsess.DSessFromSess(sqlCtx.Session)
+				defer sql.SessionEnd(dSess)
+				sql.SessionCommandBegin(dSess)
+				defer sql.SessionCommandEnd(dSess)
+
 				ddb, ok := dSess.GetDoltDB(sqlCtx, dbName)
 				if !ok {
 					sqlCtx.GetLogger().Debugf("statistics refresh error: database not found %s", dbName)

diff --git a/go/libraries/doltcore/sqle/statspro/configure.go b/go/libraries/doltcore/sqle/statspro/configure.go
@@ -42,6 +42,9 @@ func (p *Provider) Configure(ctx context.Context, ctxFactory func(ctx context.Co
 	if err != nil {
 		return err
 	}
+	defer sql.SessionEnd(loadCtx.Session)
+	sql.SessionCommandBegin(loadCtx.Session)
+	defer sql.SessionCommandEnd(loadCtx.Session)
 
 	branches := p.getStatsBranches(loadCtx)