AdGuardDNS/scripts/backend/ratelimiter.go

package main

import (
	"context"
	"log/slog"

	"github.com/AdguardTeam/AdGuardDNS/internal/backendpb"
	"github.com/AdguardTeam/golibs/httphdr"
	"github.com/AdguardTeam/golibs/netutil"
	"google.golang.org/grpc/metadata"
)

// mockRateLimitServiceServer is the mock [backendpb.RateLimiteServiceServer].
type mockRateLimitServiceServer struct {
	backendpb.UnimplementedRateLimitServiceServer
	log *slog.Logger
}

// newMockRateLimitServiceServer creates a new instance of
// *mockRateLimitServiceServer.
func newMockRateLimitServiceServer(log *slog.Logger) (srv *mockRateLimitServiceServer) {
	return &mockRateLimitServiceServer{
		log: log,
	}
}

// type check
var _ backendpb.RateLimitServiceServer = (*mockRateLimitServiceServer)(nil)

// Get implements the [backendpb.RateLimitServiceServer] interface for
// *mockRateLimitServiceServer.
func (s *mockRateLimitServiceServer) GetRateLimitSettings(
	ctx context.Context,
	req *backendpb.RateLimitSettingsRequest,
) (resp *backendpb.RateLimitSettingsResponse, err error) {
	md, _ := metadata.FromIncomingContext(ctx)

	s.log.InfoContext(
		ctx,
		"getting",
		"auth", md.Get(httphdr.Authorization),
		"req", req,
	)

	return &backendpb.RateLimitSettingsResponse{
		AllowedSubnets: []*backendpb.CidrRange{{
			Address: netutil.IPv4Localhost().AsSlice(),
			Prefix:  8,
		}},
	}, nil
}