diff --git a/models/repo.go b/models/repo.go index 5e11df2b6..d104acce5 100755 --- a/models/repo.go +++ b/models/repo.go @@ -1788,7 +1788,8 @@ func DeleteRepository(doer *User, uid, repoID int64) error { // Delete dataset attachment record and remove related files deleteDatasetAttachmentByRepoId(sess, repoID) - + // Delete cloudbrain task record + deleteCloudbrainByRepoId(sess, repoID) if err = deleteBeans(sess, &Access{RepoID: repo.ID}, &Action{RepoID: repo.ID}, @@ -1988,6 +1989,26 @@ func deleteDatasetAttachmentByRepoId(sess *xorm.Session, repoId int64) error { return err } +func deleteCloudbrainByRepoId(sess *xorm.Session, repoId int64) error { + cloudbrains := make([]*Cloudbrain, 0) + if err := sess. + Where("repo_id = ?", repoId). + Find(&cloudbrains); err != nil { + return err + } + if len(cloudbrains) == 0 { + return nil + } + for _, cloudbrain := range cloudbrains { + err := DeleteJob(cloudbrain) + if err != nil { + return err + } + } + return nil + +} + // GetRepositoryByOwnerAndName returns the repository by given ownername and reponame. func GetRepositoryByOwnerAndName(ownerName, repoName string) (*Repository, error) { return getRepositoryByOwnerAndName(x, ownerName, repoName)