Function loadDatabaseObjects

backend/plugin/parser/tidb/completion.go:170–195

loadDatabaseObjects fully loads one database's tables and views into the catalog, under that database's namespace.

(ctx context.Context, cCtx base.CompletionContext, cat *catalog.Catalog, dbName string)

// loadDatabaseObjects fully loads one database's tables and views into the
// catalog, under that database's namespace.
func loadDatabaseObjects(ctx context.Context, cCtx base.CompletionContext, cat *catalog.Catalog, dbName string) {
	db := backtickIdentifier(dbName)
	_, dbMeta, err := cCtx.Metadata(ctx, cCtx.InstanceID, dbName)
	if err != nil || dbMeta == nil {
		// Still register the name so it can be a database candidate.
		_, _ = cat.Exec("CREATE DATABASE "+db+";", &catalog.ExecOptions{ContinueOnError: true})
		return
	}
	if _, err := cat.Exec("CREATE DATABASE "+db+"; USE "+db+";", &catalog.ExecOptions{ContinueOnError: true}); err != nil {
		return
	}
	schema := dbMeta.GetSchemaMetadata("")
	if schema == nil {
		return
	}
	for _, tableName := range schema.ListTableNames() {
		if table := schema.GetTable(tableName); table != nil {
			defineTable(cat, tableName, table.GetProto().GetColumns())
		}
	}
	for _, viewName := range schema.ListViewNames() {
		if view := schema.GetView(viewName); view != nil {
			defineView(cat, viewName, view.GetDefinition())
		}
	}
}

// listAllDatabaseNames returns the instance's database names, or nil if the
// completion context cannot enumerate them.