/go/libraries/doltcore/{doltdb,sqle}: wip, figuring out foreign key junk

dolthub · Oct 18, 2024 · 93afc19 · 93afc19
1 parent 50e7254
commit 93afc19
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/go/libraries/doltcore/doltdb/system_table.go b/go/libraries/doltcore/doltdb/system_table.go
@@ -65,6 +65,11 @@ func IsFullTextTable(name string) bool {
 		strings.HasSuffix(name, "_fts_row_count"))
 }
 
+// IsDoltCITable returns whether the table name given is a dolt-ci table
+func IsDoltCITable(name string) bool {
+	return HasDoltPrefix(name) && set.NewStrSet(writeableSystemTables).Contains(name) && !IsFullTextTable(name)
+}
+
 // IsReadOnlySystemTable returns whether the table name given is a system table that should not be included in command line
 // output (e.g. dolt status) by default.
 func IsReadOnlySystemTable(name string) bool {

diff --git a/go/libraries/doltcore/sqle/database.go b/go/libraries/doltcore/sqle/database.go
@@ -877,6 +877,8 @@ func (db Database) newDoltTable(tableName string, sch schema.Schema, tbl *doltdb
 	var table sql.Table
 	if doltdb.IsReadOnlySystemTable(tableName) {
 		table = readonlyTable
+	} else if doltdb.IsDoltCITable(tableName) {
+		table = &AlterableDoltTable{WritableDoltTable{DoltTable: readonlyTable, db: db}}
 	} else if doltdb.HasDoltPrefix(tableName) && !doltdb.IsFullTextTable(tableName) {
 		table = &WritableDoltTable{DoltTable: readonlyTable, db: db}
 	} else {