mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2026-04-29 05:59:52 -05:00
Sean Reilly
5811929928
* progress commit of bulk inserts * in_flight: Add changes to metering finish the bulk insert * remove an attempt to overide enforce limits * merge in PR fixes * update docs to add in an additional section in the User guide to describe pushing single events and pushing multiple events * run lint fix --------- Co-authored-by: Sean Reilly <sean@hatchet.run>
113 lines
3.3 KiB
Go
113 lines
3.3 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/hatchet-dev/hatchet/pkg/repository/prisma/db"
|
|
"github.com/hatchet-dev/hatchet/pkg/repository/prisma/dbsqlc"
|
|
)
|
|
|
|
type BulkCreateEventOpts struct {
|
|
TenantId string `validate:"required,uuid"`
|
|
Events []*CreateEventOpts
|
|
}
|
|
|
|
type CreateEventOpts struct {
|
|
// (required) the tenant id
|
|
TenantId string `validate:"required,uuid"`
|
|
|
|
// (required) the event key
|
|
Key string `validate:"required"`
|
|
|
|
// (optional) the event data
|
|
Data []byte
|
|
|
|
// (optional) the event that this event is replaying
|
|
ReplayedEvent *string `validate:"omitempty,uuid"`
|
|
|
|
// (optional) the event metadata
|
|
AdditionalMetadata []byte
|
|
}
|
|
|
|
type ListEventOpts struct {
|
|
// (optional) a list of event keys to filter by
|
|
Keys []string
|
|
|
|
// (optional) a list of workflow IDs to filter by
|
|
Workflows []string
|
|
|
|
// (optional) a list of workflow run statuses to filter by
|
|
WorkflowRunStatus []db.WorkflowRunStatus
|
|
|
|
// (optional) number of events to skip
|
|
Offset *int
|
|
|
|
// (optional) number of events to return
|
|
Limit *int
|
|
|
|
// (optional) a search query
|
|
Search *string
|
|
|
|
// (optional) the event that this event is replaying
|
|
ReplayedEvent *string `validate:"omitempty,uuid"`
|
|
|
|
// (optional) the order by field
|
|
OrderBy *string `validate:"omitempty,oneof=createdAt"`
|
|
|
|
// (optional) the order direction
|
|
OrderDirection *string `validate:"omitempty,oneof=ASC DESC"`
|
|
|
|
// (optional) the event metadata
|
|
AdditionalMetadata []byte
|
|
|
|
// (optional) event ids to filter by
|
|
Ids []string
|
|
}
|
|
|
|
type ListEventResult struct {
|
|
Rows []*dbsqlc.ListEventsRow
|
|
Count int
|
|
}
|
|
|
|
type BulkCreateEventResult struct {
|
|
Events []*dbsqlc.Event
|
|
}
|
|
|
|
type EventAPIRepository interface {
|
|
// ListEvents returns all events for a given tenant.
|
|
ListEvents(ctx context.Context, tenantId string, opts *ListEventOpts) (*ListEventResult, error)
|
|
|
|
// ListEventKeys returns all unique event keys for a given tenant.
|
|
ListEventKeys(tenantId string) ([]string, error)
|
|
|
|
// GetEventById returns an event by id.
|
|
GetEventById(id string) (*db.EventModel, error)
|
|
|
|
// ListEventsById returns a list of events by id.
|
|
ListEventsById(tenantId string, ids []string) ([]db.EventModel, error)
|
|
}
|
|
|
|
type EventEngineRepository interface {
|
|
RegisterCreateCallback(callback Callback[*dbsqlc.Event])
|
|
|
|
// CreateEvent creates a new event for a given tenant.
|
|
CreateEvent(ctx context.Context, opts *CreateEventOpts) (*dbsqlc.Event, error)
|
|
|
|
// CreateEvent creates a new event for a given tenant.
|
|
BulkCreateEvent(ctx context.Context, opts *BulkCreateEventOpts) (*BulkCreateEventResult, error)
|
|
|
|
// GetEventForEngine returns an event for the engine by id.
|
|
GetEventForEngine(ctx context.Context, tenantId, id string) (*dbsqlc.Event, error)
|
|
|
|
ListEventsByIds(ctx context.Context, tenantId string, ids []string) ([]*dbsqlc.Event, error)
|
|
|
|
// DeleteExpiredEvents deletes events that were created before the given time. It returns the number of deleted events
|
|
// and the number of non-deleted events that match the conditions.
|
|
SoftDeleteExpiredEvents(ctx context.Context, tenantId string, before time.Time) (bool, error)
|
|
|
|
// ClearEventPayloadData removes the potentially large payload data of events that were created before the given time.
|
|
// It returns the number of events that were updated and the number of events that were not updated.
|
|
ClearEventPayloadData(ctx context.Context, tenantId string) (bool, error)
|
|
}
|