mirror of
https://github.com/ente-io/ente.git
synced 2025-05-04 12:27:54 +00:00
34 lines
1.2 KiB
Go
34 lines
1.2 KiB
Go
package embedding
|
|
|
|
import (
|
|
"github.com/ente-io/museum/pkg/controller"
|
|
"github.com/ente-io/museum/pkg/repo"
|
|
"github.com/ente-io/museum/pkg/repo/embedding"
|
|
"strconv"
|
|
)
|
|
|
|
type Controller struct {
|
|
Repo *embedding.Repository
|
|
ObjectCleanupController *controller.ObjectCleanupController
|
|
QueueRepo *repo.QueueRepository
|
|
TaskLockingRepo *repo.TaskLockRepository
|
|
FileRepo *repo.FileRepository
|
|
HostName string
|
|
cleanupCronRunning bool
|
|
}
|
|
|
|
func New(repo *embedding.Repository, objectCleanupController *controller.ObjectCleanupController, queueRepo *repo.QueueRepository, taskLockingRepo *repo.TaskLockRepository, fileRepo *repo.FileRepository, hostName string) *Controller {
|
|
return &Controller{
|
|
Repo: repo,
|
|
ObjectCleanupController: objectCleanupController,
|
|
QueueRepo: queueRepo,
|
|
TaskLockingRepo: taskLockingRepo,
|
|
FileRepo: fileRepo,
|
|
HostName: hostName,
|
|
}
|
|
}
|
|
|
|
func (c *Controller) getEmbeddingObjectPrefix(userID int64, fileID int64) string {
|
|
return strconv.FormatInt(userID, 10) + "/ml-data/" + strconv.FormatInt(fileID, 10) + "/"
|
|
}
|