diff --git a/core/app/repo/upgrade_log.go b/core/app/repo/upgrade_log.go index 9e0b129dc..6e3df813f 100644 --- a/core/app/repo/upgrade_log.go +++ b/core/app/repo/upgrade_log.go @@ -16,6 +16,7 @@ type IUpgradeLogRepo interface { Delete(opts ...global.DBOption) error WithByNodeID(nodeID uint) global.DBOption + WithByUpgradeVersion(oldVersion, newVersion string) global.DBOption } func NewIUpgradeLogRepo() IUpgradeLogRepo { @@ -79,3 +80,9 @@ func (c *UpgradeLogRepo) WithByNodeID(nodeID uint) global.DBOption { return g.Where("node_id = ?", nodeID) } } + +func (c *UpgradeLogRepo) WithByUpgradeVersion(oldVersion, newVersion string) global.DBOption { + return func(g *gorm.DB) *gorm.DB { + return g.Where("old_version = ? AND new_version = ?", oldVersion, newVersion) + } +}