Documentation ¶
Index ¶
- Variables
- type Job
- type JobState
- type Notebook
- type NotebookState
- type S
- func (s *S) AutoMigrate() error
- func (s *S) CreateJob(job *Job) error
- func (s *S) CreateNotebook(nb *Notebook) error
- func (s *S) GetJobByJobID(jobID string) (*Job, error)
- func (s *S) GetJobByJobIDAndTenantID(jobID, tenantID string) (*Job, error)
- func (s *S) ListJobsByTenantID(tenantID string) ([]*Job, error)
- func (s *S) ListJobsByTenantIDWithPagination(tenantID string, afterID uint, limit int) ([]*Job, bool, error)
- func (s *S) ListQueuedJobs() ([]*Job, error)
- func (s *S) ListQueuedJobsByTenantID(tenantID string) ([]*Job, error)
- func (s *S) UpdateJobState(jobID string, currentVersion int, newState JobState) error
- func (s *S) UpdateJobStateAndMessage(jobID string, currentVersion int, newState JobState, message []byte) error
- func (s *S) UpdateOutputModelID(jobID string, currentVersion int, outputModelID string) error
Constants ¶
This section is empty.
Variables ¶
var ( // ErrConcurrentUpdate is returned when there is a concurrent update. ErrConcurrentUpdate = fmt.Errorf("store: concurrent update") )
Functions ¶
This section is empty.
Types ¶
type Job ¶
type Job struct { gorm.Model JobID string `gorm:"uniqueIndex:idx_job_job_id"` // Message is the marshaled proto message of v1.Job. Message []byte // Suffix is a string that will be added to a fine-tuned model name. Suffix string State JobState `gorm:"index:idx_job_state_tenant_id"` TenantID string `gorm:"index:idx_job_state_tenant_id"` OrganizationID string ProjectID string KubernetesNamespace string // OutputModelID is the ID of a generated model. OutputModelID string Version int }
Job represents a job.
func (*Job) MutateMessage ¶ added in v0.14.0
MutateMessage mutates the message field of a job.
type JobState ¶
type JobState string
JobState represents the state of a job.
const ( // JobStateQueued represents the pending state. JobStateQueued JobState = "queued" // JobStateRunning represents the running state. JobStateRunning JobState = "running" // JobStatusFailed represents the failed state. JobStatusFailed JobState = "failed" // JobStateSucceeded represents the succeeded state. JobStateSucceeded JobState = "succeeded" // JobStateCancelled represents the cancelled state. JobStateCancelled JobState = "cancelled" )
type Notebook ¶ added in v0.90.0
type Notebook struct { gorm.Model NotebookID string `gorm:"uniqueIndex"` Image string // Message is the marshalled JSON of the v1.Notebook. Message []byte State NotebookState TenantID string OrganizationID string ProjectID string KubernetesNamespace string Version int }
Notebook is a model of notebook.
type NotebookState ¶ added in v0.90.0
type NotebookState string
NotebookState is the state of a notebook.
const ( // NotebookStateQueued is the state of a notebook that is waiting to be scheduled. NotebookStateQueued NotebookState = "queued" // NotebookStateRunning is the state of a notebook that is currently running. NotebookStateRunning NotebookState = "running" // NotebookStateStopped is the state of a notebook that has been stopped. NotebookStateStopped NotebookState = "stopped" // NotebookStateFailed is the state of a notebook that has failed. NotebookStateFailed NotebookState = "failed" )
type S ¶
type S struct {
// contains filtered or unexported fields
}
S represents the data store.
func (*S) AutoMigrate ¶
AutoMigrate sets up the auto-migration task of the database.
func (*S) CreateNotebook ¶ added in v0.90.0
CreateNotebook creates a new notebook.
func (*S) GetJobByJobID ¶
GetJobByJobID gets a job.
func (*S) GetJobByJobIDAndTenantID ¶ added in v0.76.0
GetJobByJobIDAndTenantID gets a job by its job ID and tenant ID.
func (*S) ListJobsByTenantID ¶
ListJobsByTenantID finds jobs.
func (*S) ListJobsByTenantIDWithPagination ¶ added in v0.82.0
func (s *S) ListJobsByTenantIDWithPagination(tenantID string, afterID uint, limit int) ([]*Job, bool, error)
ListJobsByTenantIDWithPagination finds jobs with pagination. Jobs are returned with a descending order of ID.
func (*S) ListQueuedJobs ¶ added in v0.11.0
ListQueuedJobs finds queued jobs.
func (*S) ListQueuedJobsByTenantID ¶ added in v0.11.0
ListQueuedJobsByTenantID finds queued jobs.
func (*S) UpdateJobState ¶
UpdateJobState updates a job.