Method Query

pkg/storage/driver/sql.go:397–464

Query returns the set of releases that match the provided set of labels.

(labels map[string]string)

// Query returns the set of releases that match the provided set of labels.
func (s *SQL) Query(labels map[string]string) ([]release.Releaser, error) {
	sb := s.statementBuilder.
		Select(sqlReleaseTableKeyColumn, sqlReleaseTableNamespaceColumn, sqlReleaseTableBodyColumn).
		From(sqlReleaseTableName)

	keys := make([]string, 0, len(labels))
	for key := range labels {
		keys = append(keys, key)
	}
	sort.Strings(keys)
	for _, key := range keys {
		if _, ok := labelMap[key]; ok {
			sb = sb.Where(sq.Eq{key: labels[key]})
		} else {
			s.Logger().Debug("unknown label", "key", key)
			return nil, fmt.Errorf("unknown label %s", key)
		}
	}

	// If a namespace was specified, we only list releases from that namespace
	if s.namespace != "" {
		sb = sb.Where(sq.Eq{sqlReleaseTableNamespaceColumn: s.namespace})
	}

	// Build our query
	query, args, err := sb.ToSql()
	if err != nil {
		s.Logger().Debug("failed to build query", slog.Any("error", err))
		return nil, err
	}

	var records = []SQLReleaseWrapper{}
	if err := s.db.Select(&records, query, args...); err != nil {
		s.Logger().Debug("failed to query with labels", slog.Any("error", err))
		return nil, err
	}

	if len(records) == 0 {
		return nil, ErrReleaseNotFound
	}

	var releases []release.Releaser
	for _, record := range records {
		release, err := decodeRelease(record.Body)
		if err != nil {
			s.Logger().Debug("failed to decode release", slog.Any("record", record), slog.Any("error", err))
			continue
		}

		if release.Labels, err = s.getReleaseCustomLabels(record.Key, record.Namespace); err != nil {
			s.Logger().Debug(
				"failed to get release custom labels",
				slog.String("namespace", record.Namespace),
				slog.String("key", record.Key),
				slog.Any("error", err),
			)
			return nil, err
		}