fix: fix bug that the files associated with records still exist after…

… deleting mysql and ftp records
Li4n0 · Nov 16, 2022 · 5e80591 · 5e80591
1 parent 48f55c5
commit 5e80591
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/internal/file/file.go b/internal/file/file.go
@@ -10,7 +10,7 @@ import (
 
 type File struct {
 	ID       uint   `gorm:"primarykey" form:"id" json:"id"`
-	RecordID uint   `gorm:"constraint:OnUpdate:CASCADE,OnDelete:CASCADE;" json:"-"`
+	RecordID uint   `json:"-"`
 	Name     string `json:"name"`
 	Content  []byte `json:"-"`
 }

diff --git a/pkg/ftp/record.go b/pkg/ftp/record.go
@@ -24,7 +24,7 @@ type Record struct {
 	Path     string        `form:"path" json:"path"`
 	Method   Method        `form:"method" json:"method"`
 	Status   Status        `form:"status" json:"status"`
-	File     *file.FTPFile `form:"file" json:"file" notice:"-"`
+	File     *file.FTPFile `gorm:"foreignKey:RecordID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE;" form:"file" json:"file" notice:"-"`
 	Rule     Rule          `gorm:"foreignKey:RuleName;references:Name;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;" form:"-" json:"-" notice:"-"`
 }
 

diff --git a/pkg/mysql/record.go b/pkg/mysql/record.go
@@ -23,7 +23,7 @@ type Record struct {
 	Schema        string           `gorm:"index" form:"schema" json:"schema" notice:"schema"`
 	ClientOS      string           `gorm:"index" form:"client_os" json:"client_os" notice:"client_os"`
 	LoadLocalData bool             `gorm:"index" form:"load_local_data" json:"load_local_data" notice:"load_local_data"`
-	Files         []file.MySQLFile `form:"-" json:"files" notice:"-"`
+	Files         []file.MySQLFile `gorm:"foreignKey:RecordID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE;" form:"-" json:"files" notice:"-" `
 	Rule          Rule             `gorm:"foreignKey:RuleName;references:Name;constraint:OnUpdate:CASCADE,OnDelete:SET NULL;" form:"-" json:"-" notice:"-"`
 }