Method insertAllColumns

backend/plugin/parser/pg/completion.go:455–518

(c *Completer)

}

func (m CompletionMap) insertAllColumns(c *Completer) {
	if _, exists := c.metadataCache[c.defaultDatabase]; !exists {
		_, metadata, err := c.getMetadata(c.ctx, c.instanceID, c.defaultDatabase)
		if err != nil || metadata == nil {
			return
		}
		c.metadataCache[c.defaultDatabase] = metadata
	}

	metadata := c.metadataCache[c.defaultDatabase]
	for _, schema := range metadata.ListSchemaNames() {
		schemaMeta := metadata.GetSchemaMetadata(schema)
		if schemaMeta == nil {
			continue
		}
		for _, table := range schemaMeta.ListTableNames() {
			tableMeta := schemaMeta.GetTable(table)
			if tableMeta == nil {
				continue
			}
			for _, column := range tableMeta.GetProto().GetColumns() {
				definition := fmt.Sprintf("%s.%s | %s", schema, table, column.Type)
				if !column.Nullable {
					definition += ", NOT NULL"
				}
				m.Insert(base.Candidate{
					Type:       base.CandidateTypeColumn,
					Text:       c.quotedIdentifierIfNeeded(column.Name),
					Definition: definition,
					Comment:    column.Comment,
				})
			}
		}
		for _, ft := range schemaMeta.ListForeignTableNames() {
			extMeta := schemaMeta.GetExternalTable(ft)
			if extMeta == nil {
				continue
			}
			for _, column := range extMeta.GetProto().GetColumns() {
				definition := fmt.Sprintf("%s.%s | %s", schema, ft, column.Type)
				if !column.Nullable {
					definition += ", NOT NULL"
				}
				m.Insert(base.Candidate{
					Type:       base.CandidateTypeColumn,
					Text:       c.quotedIdentifierIfNeeded(column.Name),
					Definition: definition,
					Comment:    column.Comment,
				})
			}
		}
		for _, mv := range schemaMeta.ListMaterializedViewNames() {
			mvMeta := schemaMeta.GetMaterializedView(mv)
			if mvMeta == nil {
				continue
			}
			if columns, err := c.resolveMaterializedViewColumns(mvMeta.GetDefinition(), schema, mv); err == nil {
				for _, col := range columns {