feat: add initial tus db models

db/db.go

@@ -59,6 +59,8 @@ func (d *DatabaseImpl) Start() error {
 		&models.Upload{},
 		&models.User{},
 		&models.S5Challenge{},
+		&models.TusLock{},
+		&models.TusUpload{},
 	)
 }
 

db/models/tus_lock.go

@@ -0,0 +1,87 @@
+package models
+
+import (
+	"context"
+	"errors"
+	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
+	"time"
+)
+
+var (
+	ErrTusLockBusy = errors.New("lock is currently held by another process")
+)
+
+type TusLock struct {
+	gorm.Model
+	LockId           string `gorm:"uniqueIndex"`
+	HolderPID        int    `gorm:"index"`
+	AcquiredAt       time.Time
+	ExpiresAt        time.Time
+	ReleaseRequested bool `gorm:"default:false"`
+}
+
+func (t *TusLock) TryLock(db *gorm.DB, ctx context.Context) error {
+	return db.Transaction(func(tx *gorm.DB) error {
+		var existingLock TusLock
+
+		if err := tx.WithContext(ctx).Clauses(clause.Locking{Strength: "UPDATE"}).Where("lock_id = ?", t.LockId).First(&existingLock).Error; err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				// Insert new lock record
+				err := tx.WithContext(ctx).Create(t).Error
+				if err != nil {
+					return err
+				}
+
+				t = &existingLock
+				return nil
+			}
+
+			return err
+		}
+
+		// Check if existing lock is expired
+		if existingLock.ExpiresAt.Before(time.Now()) || existingLock.ReleaseRequested {
+
+			err := tx.Model(&existingLock).Updates(t).Error
+			if err != nil {
+				return err
+			}
+
+			t = &existingLock
+
+			return nil
+		}
+
+		// Lock is currently held by another process
+		return ErrTusLockBusy
+	})
+}
+func (t *TusLock) RequestRelease(db *gorm.DB) error {
+	return db.Transaction(func(tx *gorm.DB) error {
+		// Update the ReleaseRequested flag in the database for the specific lock
+		return tx.Model(t).Where("lock_id = ?", t.LockId).Update("release_requested", true).Error
+	})
+}
+func (t *TusLock) Released(db *gorm.DB) error {
+	return db.Transaction(func(tx *gorm.DB) error {
+		// Update the ReleaseRequested flag in the database for the specific lock
+		return tx.Model(t).Where("lock_id = ?", t.LockId).Update("release_requested", false).Error
+	})
+}
+
+func (t *TusLock) IsReleaseRequested(db *gorm.DB) (bool, error) {
+	var count int64
+	err := db.Model(&TusLock{}).Where(&TusLock{LockId: t.LockId, ReleaseRequested: true}).Count(&count).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return true, nil
+		}
+		return false, err
+	}
+
+	return count > 0, nil
+}
+func (t *TusLock) Delete(db *gorm.DB) error {
+	return db.Where("lock_id = ?", t.LockId).Delete(&TusLock{}).Error
+}

db/models/tus_upload.go

@@ -0,0 +1,13 @@
+package models
+
+import "gorm.io/gorm"
+
+type TusUpload struct {
+	gorm.Model
+	Hash       string `gorm:"uniqueIndex"`
+	UploadID   string `gorm:"uniqueIndex"`
+	UploaderID uint
+	UploaderIP string
+	Uploader   User `gorm:"foreignKey:UploaderID"`
+	Protocol   string
+}