File size: 1,881 Bytes
cee2387 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | package entitlementdriverv2
import (
"github.com/openmeterio/openmeter/openmeter/customer"
"github.com/openmeterio/openmeter/openmeter/entitlement"
meteredentitlement "github.com/openmeterio/openmeter/openmeter/entitlement/metered"
"github.com/openmeterio/openmeter/openmeter/namespace/namespacedriver"
"github.com/openmeterio/openmeter/pkg/framework/transport/httptransport"
)
// EntitlementHandler exposes V2 customer entitlement endpoints
type EntitlementHandler interface {
CreateCustomerEntitlement() CreateCustomerEntitlementHandler
ListCustomerEntitlements() ListCustomerEntitlementsHandler
GetCustomerEntitlement() GetCustomerEntitlementHandler
DeleteCustomerEntitlement() DeleteCustomerEntitlementHandler
OverrideCustomerEntitlement() OverrideCustomerEntitlementHandler
ListCustomerEntitlementGrants() ListCustomerEntitlementGrantsHandler
CreateCustomerEntitlementGrant() CreateCustomerEntitlementGrantHandler
GetCustomerEntitlementHistory() GetCustomerEntitlementHistoryHandler
ResetCustomerEntitlementUsage() ResetCustomerEntitlementUsageHandler
ListEntitlements() ListEntitlementsHandler
GetEntitlement() GetEntitlementHandler
}
type entitlementHandler struct {
namespaceDecoder namespacedriver.NamespaceDecoder
options []httptransport.HandlerOption
connector entitlement.Service
balanceConnector meteredentitlement.Connector
customerService customer.Service
}
func NewEntitlementHandler(
connector entitlement.Service,
balanceConnector meteredentitlement.Connector,
customerService customer.Service,
namespaceDecoder namespacedriver.NamespaceDecoder,
options ...httptransport.HandlerOption,
) EntitlementHandler {
return &entitlementHandler{
namespaceDecoder: namespaceDecoder,
options: options,
connector: connector,
balanceConnector: balanceConnector,
customerService: customerService,
}
}
|