bad refactor

dolthub · May 15, 2024 · 51caf2a · 51caf2a
1 parent ffe7e0a
commit 51caf2a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/go/libraries/doltcore/sqle/writer/prolly_table_writer.go b/go/libraries/doltcore/sqle/writer/prolly_table_writer.go
@@ -311,6 +311,7 @@ func (w *prollyTableWriter) Reset(ctx *sql.Context, sess *prollyWriteSession, tb
 			if err != nil {
 				return err
 			}
+			dbState.SessionCache().CacheWriterState(schKey, schState)
 		}
 	} else {
 		schState, err = getWriterSchemas(ctx, w.tbl, w.tableName.Name)

diff --git a/go/libraries/doltcore/sqle/writer/prolly_write_session.go b/go/libraries/doltcore/sqle/writer/prolly_write_session.go
@@ -74,6 +74,7 @@ func (s *prollyWriteSession) GetTableWriter(ctx *sql.Context, table doltdb.Table
 			if err != nil {
 				return nil, err
 			}
+			dbState.SessionCache().CacheWriterState(schKey, schState)
 		}
 	} else {
 		schState, err = getWriterSchemas(ctx, t, table.Name)