package storage import ( "context" "errors" "time" "forge.cadoles.com/arcad/edge/pkg/storage/filter" "github.com/oklog/ulid/v2" ) var ErrDocumentNotFound = errors.New("document not found") type DocumentID string const ( DocumentAttrID = "_id" DocumentAttrCreatedAt = "_createdAt" DocumentAttrUpdatedAt = "_updatedAt" ) func NewDocumentID() DocumentID { return DocumentID(ulid.Make().String()) } type Document map[string]interface{} func (d Document) ID() (DocumentID, bool) { rawID, exists := d[DocumentAttrID] if !exists { return "", false } strID, ok := rawID.(string) if ok { return DocumentID(strID), true } docID, ok := rawID.(DocumentID) if ok { return docID, true } return "", false } func (d Document) CreatedAt() (time.Time, bool) { return d.timeAttr(DocumentAttrCreatedAt) } func (d Document) UpdatedAt() (time.Time, bool) { return d.timeAttr(DocumentAttrUpdatedAt) } func (d Document) timeAttr(attr string) (time.Time, bool) { rawTime, exists := d[attr] if !exists { return time.Time{}, false } t, ok := rawTime.(time.Time) if !ok { return time.Time{}, false } return t, true } type DocumentStore interface { Get(ctx context.Context, collection string, id DocumentID) (Document, error) Query(ctx context.Context, collection string, filter *filter.Filter, funcs ...QueryOptionFunc) ([]Document, error) Upsert(ctx context.Context, collection string, document Document) (Document, error) Delete(ctx context.Context, collection string, id DocumentID) error }