Function buildCompletionCatalog

backend/plugin/parser/redshift/completion.go:93–153

(ctx context.Context, cCtx base.CompletionContext)

}

func buildCompletionCatalog(ctx context.Context, cCtx base.CompletionContext) *redshiftcatalog.Catalog {
	if cCtx.Metadata == nil || cCtx.DefaultDatabase == "" {
		return nil
	}

	_, metadata, err := cCtx.Metadata(ctx, cCtx.InstanceID, cCtx.DefaultDatabase)
	if err != nil || metadata == nil {
		return nil
	}

	cat := redshiftcatalog.New()
	for _, schemaName := range metadata.ListSchemaNames() {
		schemaMeta := metadata.GetSchemaMetadata(schemaName)
		if schemaMeta == nil {
			continue
		}
		execCompletionDDL(cat, fmt.Sprintf("CREATE SCHEMA IF NOT EXISTS %s;", quoteIdent(schemaName)))

		for _, tableName := range schemaMeta.ListTableNames() {
			tableMeta := schemaMeta.GetTable(tableName)
			if tableMeta == nil {
				continue
			}
			execCompletionDDL(cat, createTableDDL(schemaName, tableName, tableMeta.GetProto().GetColumns()))
		}
		for _, tableName := range schemaMeta.ListForeignTableNames() {
			tableMeta := schemaMeta.GetExternalTable(tableName)
			if tableMeta == nil {
				continue
			}
			execCompletionDDL(cat, createTableDDL(schemaName, tableName, tableMeta.GetProto().GetColumns()))
		}
		for _, viewName := range schemaMeta.ListViewNames() {
			viewMeta := schemaMeta.GetView(viewName)
			if viewMeta == nil {
				continue
			}
			execCompletionDDL(cat, createViewDDL("VIEW", schemaName, viewName, viewMeta.GetColumns()))
		}
		for _, viewName := range schemaMeta.ListMaterializedViewNames() {
			viewMeta := schemaMeta.GetMaterializedView(viewName)
			if viewMeta == nil {
				continue
			}
			cat.SetSearchPath([]string{schemaName, "public"})
			if viewMeta.GetDefinition() == "" || !execCompletionDDL(cat, createMaterializedViewDDL(schemaName, viewName, viewMeta.GetDefinition())) {
				execCompletionDDL(cat, createViewDDL("MATERIALIZED VIEW", schemaName, viewName, nil))
			}
		}
		for _, sequenceName := range schemaMeta.ListSequenceNames() {
			execCompletionDDL(cat, fmt.Sprintf("CREATE SEQUENCE %s.%s;", quoteIdent(schemaName), quoteIdent(sequenceName)))
		}
	}

	searchPath := []string{"public"}
	if cCtx.DefaultSchema != "" {
		searchPath = []string{cCtx.DefaultSchema, "public"}
	}