65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package pfs_db
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type DirModel struct {
|
|
ID uuid.UUID `gorm:"type:uuid;primaryKey"`
|
|
ParentID *uuid.UUID `gorm:"type:uuid;index"`
|
|
Parent *DirModel `gorm:"foreignKey:ParentID;constraint:OnDelete:CASCADE"`
|
|
|
|
Name string
|
|
OwnerID uuid.UUID
|
|
IsPublic bool
|
|
Created time.Time
|
|
LastModified time.Time
|
|
}
|
|
|
|
func (d *DirModel) BeforeCreate(tx *gorm.DB) error {
|
|
now := time.Now()
|
|
d.Created = now
|
|
d.LastModified = now
|
|
return nil
|
|
}
|
|
|
|
func (d *DirModel) BeforeUpdate(tx *gorm.DB) error {
|
|
d.LastModified = time.Now()
|
|
return nil
|
|
}
|
|
|
|
type FileModel struct {
|
|
ID uuid.UUID `gorm:"type:uuid;primaryKey"`
|
|
DirID *uuid.UUID `gorm:"type:uuid;index"`
|
|
Dir *DirModel `gorm:"foreignKey:DirID;constraint:OnDelete:CASCADE"`
|
|
|
|
Name string
|
|
OwnerID uuid.UUID
|
|
IsPublic bool
|
|
Created time.Time
|
|
LastModified time.Time
|
|
|
|
FileData FileDataModel `gorm:"foreignKey:FileID;references:ID;constraint:OnDelete:CASCADE"`
|
|
}
|
|
|
|
func (f *FileModel) BeforeCreate(tx *gorm.DB) error {
|
|
now := time.Now()
|
|
f.Created = now
|
|
f.LastModified = now
|
|
return nil
|
|
}
|
|
|
|
func (f *FileModel) BeforeUpdate(tx *gorm.DB) error {
|
|
f.LastModified = time.Now()
|
|
return nil
|
|
}
|
|
|
|
type FileDataModel struct {
|
|
ID uuid.UUID `gorm:"type:uuid;primaryKey"`
|
|
FileID uuid.UUID `gorm:"type:uuid;uniqueIndex;not null"`
|
|
Data []byte
|
|
}
|