models

type IdempotencyRecord struct {
	// Primary key: idempotency key (from header or request)
	IdempotencyKey string `dynamorm:"pk" json:"idempotency_key"`

	// Sort key: constant value for single item per key
	SK string `dynamorm:"sk" json:"sk" default:"IDEMPOTENCY"`

	// GSIs for querying
	FunctionName string    `dynamorm:"index:gsi-function,pk" json:"function_name"`
	TenantID     string    `dynamorm:"index:gsi-tenant,pk" json:"tenant_id,omitempty"`
	Status       string    `dynamorm:"index:gsi-status,pk" json:"status"`
	Timestamp    time.Time `dynamorm:"index:gsi-timestamp,pk" json:"timestamp"`

	// Request/Response data
	RequestHash string `json:"request_hash"`
	RequestBody string `dynamorm:"json" json:"request_body"` // Stored as JSON
	Response    string `dynamorm:"json" json:"response"`     // Can be up to 400KB
	StatusCode  int    `json:"status_code"`

	// State management
	LockToken   string    `json:"lock_token,omitempty"`
	LockedUntil time.Time `json:"locked_until,omitempty"`

	// TTL for automatic cleanup
	ExpiresAt time.Time `dynamorm:"ttl" json:"expires_at"`

	// Metadata
	CreatedAt   time.Time `dynamorm:"created_at" json:"created_at"`
	UpdatedAt   time.Time `dynamorm:"updated_at" json:"updated_at"`
	CompletedAt time.Time `json:"completed_at,omitempty"`

	// Error tracking
	ErrorMessage string `json:"error_message,omitempty"`
	RetryCount   int    `json:"retry_count,omitempty"`
}

type RateLimitRecord struct {
	// Primary key: identifier (could be IP, UserID, TenantID+UserID, etc.)
	Identifier string `dynamorm:"pk" json:"identifier"`

	// Sort key: window timestamp (for sliding window rate limiting)
	WindowTime string `dynamorm:"sk" json:"window_time"`

	// GSI for querying by different dimensions
	IPAddress string `dynamorm:"index:gsi-ip,pk" json:"ip_address,omitempty"`
	UserID    string `dynamorm:"index:gsi-user,pk" json:"user_id,omitempty"`
	TenantID  string `dynamorm:"index:gsi-tenant,pk" json:"tenant_id,omitempty"`

	// Rate limit data
	Count     int    `json:"count"`
	BucketKey string `dynamorm:"index:gsi-bucket,pk" json:"bucket_key"`

	// TTL for automatic cleanup (set to window end + buffer)
	ExpiresAt time.Time `dynamorm:"ttl" json:"expires_at"`

	// Metadata
	CreatedAt time.Time `dynamorm:"created_at" json:"created_at"`
	UpdatedAt time.Time `dynamorm:"updated_at" json:"updated_at"`
}

type SlidingWindowEntry struct {
	// Composite key: RateLimitKey#Timestamp
	PK string `dynamorm:"pk" json:"-"`
	SK string `dynamorm:"sk" json:"-"`

	// Attributes
	RateLimitKey string    `json:"rate_limit_key"`
	Timestamp    time.Time `json:"timestamp"`
	Weight       int       `json:"weight,omitempty"` // For weighted rate limiting
	RequestID    string    `json:"request_id"`

	// TTL for automatic cleanup (set to window duration + buffer)
	ExpiresAt int64 `dynamorm:"ttl" json:"-"`
}