Function getResourcesForPostgres

backend/component/export/resources.go:157–227

(
	ctx context.Context,
	storeInstance *store.Store,
	engine storepb.Engine,
	databaseName string,
	statement string,
	instance *store.InstanceMessage,
	defaultSchema string,
	getDatabaseMetadataFunc base.GetDatabaseMetadataFunc,
	listDatabaseNamesFunc base.ListDatabaseNamesFunc,
	getLinkedDatabaseMetadataFunc base.GetLinkedDatabaseMetadataFunc,
)

}

func getResourcesForPostgres(
	ctx context.Context,
	storeInstance *store.Store,
	engine storepb.Engine,
	databaseName string,
	statement string,
	instance *store.InstanceMessage,
	defaultSchema string,
	getDatabaseMetadataFunc base.GetDatabaseMetadataFunc,
	listDatabaseNamesFunc base.ListDatabaseNamesFunc,
	getLinkedDatabaseMetadataFunc base.GetLinkedDatabaseMetadataFunc,
) ([]base.SchemaResource, error) {
	statements, err := base.SplitMultiSQL(engine, statement)
	if err != nil {
		return nil, err
	}
	spans, err := base.GetQuerySpan(ctx, base.GetQuerySpanContext{
		InstanceID:                    instance.ResourceID,
		GetDatabaseMetadataFunc:       getDatabaseMetadataFunc,
		ListDatabaseNamesFunc:         listDatabaseNamesFunc,
		GetLinkedDatabaseMetadataFunc: getLinkedDatabaseMetadataFunc,
	}, engine, statements, databaseName, defaultSchema, !store.IsObjectCaseSensitive(instance))
	if err != nil {
		return nil, err
	}

	database, err := storeInstance.GetDatabase(ctx, &store.FindDatabaseMessage{
		Workspace:    instance.Workspace,
		InstanceID:   &instance.ResourceID,
		DatabaseName: &databaseName,
	})
	if err != nil {
		return nil, connect.NewError(connect.CodeInternal, errors.Wrap(err, "failed to fetch database"))
	}
	if database == nil {
		return nil, nil
	}

	dbMetadata, err := storeInstance.GetDBSchema(ctx, &store.FindDBSchemaMessage{
		Workspace:    instance.Workspace,
		InstanceID:   database.InstanceID,
		DatabaseName: database.DatabaseName,
	})
	if err != nil {
		return nil, connect.NewError(connect.CodeInternal, errors.Wrap(err, "failed to fetch database schema"))
	}

	var result []base.SchemaResource
	for _, span := range spans {
		for sourceColumn := range span.SourceColumns {
			sr := base.SchemaResource{
				Database:     sourceColumn.Database,
				Schema:       sourceColumn.Schema,
				Table:        sourceColumn.Table,
				LinkedServer: sourceColumn.Server,
			}

			if sourceColumn.Database != dbMetadata.GetProto().Name {