protection_impls.go

// Copyright (c) The Thanos Authors.
// Licensed under the Apache License 2.0.

package queryfrontend

import (
	"context"

	"github.com/pkg/errors"
)

// This file contains all protection rule implementations.
// Each protection implements the Protection interface defined in protection.go.

// AlwaysMatchProtection is a protection that always matches every query.
type AlwaysMatchProtection struct{}

func (n *AlwaysMatchProtection) Name() string { return "always-match" }

func (n *AlwaysMatchProtection) Run(_ context.Context, _ thanosQueryReq) (bool, error) {
	return true, nil
}

// protectionRegistry maps protection names (as used in config) to their factory functions.
var protectionRegistry = map[string]ProtectionFactory{
	"always-match": func(_ map[string]string) (Protection, error) {
		return &AlwaysMatchProtection{}, nil
	},
}

// LookupProtection returns the factory for the given protection name.
func LookupProtection(name string) (ProtectionFactory, error) {
	factory, ok := protectionRegistry[name]
	if !ok {
		return nil, errors.Errorf("unknown protection %q", name)
	}
	return factory, nil
}