Function WrapSqlDatabase

server/tables/pg_database.go:57–65

WrapSqlDatabase wraps any Dolt database variant as a Pg-aware database. ReadOnlyDatabase embeds sqle.Database by value, so a plain sqle.Database type assertion does not match it; this function handles both cases.

(db sql.Database)

// ReadOnlyDatabase embeds sqle.Database by value, so a plain sqle.Database type
// assertion does not match it; this function handles both cases.
func WrapSqlDatabase(db sql.Database) sql.Database {
	if rodb, ok := db.(sqle.ReadOnlyDatabase); ok {
		return &PgReadOnlyDatabase{rodb}
	}
	if sdb, ok := db.(sqle.Database); ok {
		return WrapSqleDatabase(sdb)
	}
	return db
}

// applySchemaWrap wraps a single schema returned by the underlying sqle.Database methods.
// System schemas (those with registered virtual-table handlers) get a Database wrapper