mysql

type CommitLabelMappingDAOImpl struct {
	// contains filtered or unexported fields
}

type DebugContextDAOImpl struct {
	// contains filtered or unexported fields
}

type DebugLogDAOImpl struct {
	// contains filtered or unexported fields
}

type ICommitLabelMappingDAO interface {
	BatchCreate(ctx context.Context, mappings []*model.PromptCommitLabelMapping, opts ...db.Option) error
	BatchUpdate(ctx context.Context, mappings []*model.PromptCommitLabelMapping, opts ...db.Option) error
	BatchDelete(ctx context.Context, ids []int64, opts ...db.Option) error
	ListByPromptIDAndLabelKeys(ctx context.Context, promptID int64, labelKeys []string, opts ...db.Option) ([]*model.PromptCommitLabelMapping, error)
	ListByPromptIDAndVersions(ctx context.Context, promptID int64, versions []string, opts ...db.Option) ([]*model.PromptCommitLabelMapping, error)
	MGetPromptVersionByLabelQuery(ctx context.Context, param MGetPromptVersionByLabelQueryParam, opts ...db.Option) ([]*model.PromptCommitLabelMapping, error)
}

type IDebugContextDAO interface {
	Save(ctx context.Context, debugContext *model.PromptDebugContext, opts ...db.Option) error
	Get(ctx context.Context, promptID int64, userID string, opts ...db.Option) (*model.PromptDebugContext, error)
}

type IDebugLogDAO interface {
	List(ctx context.Context, param ListParam, opts ...db.Option) ([]*model.PromptDebugLog, error)
	Save(ctx context.Context, debugLog *model.PromptDebugLog, opts ...db.Option) error
}

type ILabelDAO interface {
	Create(ctx context.Context, labelPO *model.PromptLabel, opts ...db.Option) error
	List(ctx context.Context, param ListLabelDAOParam, opts ...db.Option) ([]*model.PromptLabel, error)
	BatchGet(ctx context.Context, spaceID int64, labelKeys []string, opts ...db.Option) ([]*model.PromptLabel, error)
}

type IPromptBasicDAO interface {
	Create(ctx context.Context, basicPO *model.PromptBasic, opts ...db.Option) (err error)

	Delete(ctx context.Context, promptID int64, opts ...db.Option) (err error)

	Get(ctx context.Context, promptID int64, opts ...db.Option) (basicPO *model.PromptBasic, err error)
	MGet(ctx context.Context, promptIDs []int64, opts ...db.Option) (idPromptPOMap map[int64]*model.PromptBasic, err error)
	MGetByPromptKey(ctx context.Context, spaceID int64, promptKeys []string, opts ...db.Option) (promptPOs []*model.PromptBasic, err error)
	List(ctx context.Context, param ListPromptBasicParam, opts ...db.Option) (basicPOs []*model.PromptBasic, total int64, err error)

	Update(ctx context.Context, promptID int64, updateFields map[string]interface{}, opts ...db.Option) (err error)
}

type IPromptCommitDAO interface {
	Create(ctx context.Context, promptCommitPO *model.PromptCommit, time time.Time, opts ...db.Option) (err error)
	Get(ctx context.Context, promptID int64, commitVersion string, opts ...db.Option) (promptCommitPO *model.PromptCommit, err error)
	MGet(ctx context.Context, pairs []PromptIDCommitVersionPair, opts ...db.Option) (pairCommitPOMap map[PromptIDCommitVersionPair]*model.PromptCommit, err error)
	List(ctx context.Context, param ListCommitParam, opts ...db.Option) (commitPOs []*model.PromptCommit, err error)
}

type IPromptUserDraftDAO interface {
	Create(ctx context.Context, promptDraftPO *model.PromptUserDraft, opts ...db.Option) (err error)
	Delete(ctx context.Context, draftID int64, opts ...db.Option) (err error)
	Get(ctx context.Context, promptID int64, userID string, opts ...db.Option) (promptDraftPO *model.PromptUserDraft, err error)
	GetByID(ctx context.Context, draftID int64, opts ...db.Option) (promptDraftPO *model.PromptUserDraft, err error)
	MGet(ctx context.Context, pairs []PromptIDUserIDPair, opts ...db.Option) (pairDraftPOMap map[PromptIDUserIDPair]*model.PromptUserDraft, err error)
	Update(ctx context.Context, promptDraftPO *model.PromptUserDraft, opts ...db.Option) (err error)
}

type LabelDAOImpl struct {
	// contains filtered or unexported fields
}

type ListCommitParam struct {
	PromptID int64

	Cursor *int64
	Limit  int
	Asc    bool
}

type ListLabelDAOParam struct {
	SpaceID      int64
	LabelKeyLike string
	Cursor       *int64
	Limit        int
}

type ListParam struct {
	PromptID    *int64
	UserID      *string
	StartBefore *int64
	StartAfter  *int64
	DebugIDs    []int64
	DebugStep   *int32
	Limit       *int
}

type ListPromptBasicParam struct {
	SpaceID int64

	KeyWord       string
	CreatedBys    []string
	CommittedOnly bool

	Offset  int
	Limit   int
	OrderBy int
	Asc     bool
}

type MGetPromptVersionByLabelParam struct {
	SpaceID    int64
	PromptKeys []string
	LabelKeys  []string
}

type MGetPromptVersionByLabelQueryParam struct {
	Queries []PromptLabelQuery
}

type PromptBasicDAOImpl struct {
	// contains filtered or unexported fields
}

type PromptCommitDAOImpl struct {
	// contains filtered or unexported fields
}

type PromptIDCommitVersionPair struct {
	PromptID      int64
	CommitVersion string
}

type PromptIDUserIDPair struct {
	PromptID int64
	UserID   string
}

type PromptLabelQuery struct {
	PromptID int64
	LabelKey string
}

type PromptUserDraftDAOImpl struct {
	// contains filtered or unexported fields
}