Method getDatabases

backend/plugin/db/redshift/sync.go:724–768

getDatabases gets all databases of an instance.

(ctx context.Context)

// getDatabases gets all databases of an instance.
func (d *Driver) getDatabases(ctx context.Context) ([]*storepb.DatabaseSchemaMetadata, error) {
	consumerDatabases := make(map[string]bool)
	dsRows, err := d.db.QueryContext(ctx, `
		SELECT consumer_database FROM SVV_DATASHARES WHERE share_type = 'INBOUND';
	`)
	if err != nil {
		return nil, err
	}
	defer dsRows.Close()

	for dsRows.Next() {
		var v string
		if err := dsRows.Scan(&v); err != nil {
			return nil, err
		}
		consumerDatabases[v] = true
	}
	if err := dsRows.Err(); err != nil {
		return nil, err
	}

	var databases []*storepb.DatabaseSchemaMetadata
	rows, err := d.db.QueryContext(ctx, `
		SELECT datname,
		pg_encoding_to_char(encoding)
		FROM pg_database;
	`)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	for rows.Next() {
		database := storepb.DatabaseSchemaMetadata{}
		if err := rows.Scan(&database.Name, &database.CharacterSet); err != nil {
			return nil, err
		}
		database.Datashare = consumerDatabases[database.Name]
		databases = append(databases, &database)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return databases, nil
}