Function WriteUserIndex

pkg/util/users/index.go:42–67

WriteUserIndex uploads the provided index to the storage.

(ctx context.Context, bkt objstore.Bucket, idx *UserIndex)

// WriteUserIndex uploads the provided index to the storage.
func WriteUserIndex(ctx context.Context, bkt objstore.Bucket, idx *UserIndex) error {
	// Marshal the index.
	content, err := json.Marshal(idx)
	if err != nil {
		return errors.Wrap(err, "marshal user index")
	}

	// Compress it.
	var gzipContent bytes.Buffer
	gzip := gzip.NewWriter(&gzipContent)
	gzip.Name = UserIndexFilename

	if _, err := gzip.Write(content); err != nil {
		return errors.Wrap(err, "gzip user index")
	}
	if err := gzip.Close(); err != nil {
		return errors.Wrap(err, "close gzip user index")
	}

	// Upload the index to the storage.
	if err := bkt.Upload(ctx, UserIndexCompressedFilename, bytes.NewReader(gzipContent.Bytes())); err != nil {
		return errors.Wrap(err, "upload user index")
	}

	return nil
}

func ReadUserIndex(ctx context.Context, bkt objstore.InstrumentedBucket, logger log.Logger) (*UserIndex, error) {
	// Get the user index.