s3 complete

pkg/common/db/relation/object_hash_model.go

@@ -38,3 +38,15 @@ func (o *ObjectHashGorm) Create(ctx context.Context, h []*relation.ObjectHashMod
 	}()
 	return utils.Wrap1(o.DB.Create(h).Error)
 }
+
+func (o *ObjectHashGorm) DeleteNoCitation(ctx context.Context, engine string, num int) (list []*relation.ObjectHashModel, err error) {
+	defer func() {
+		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "engine", engine, "num", num, "objectHash", list)
+	}()
+	err = o.DB.Table(relation.ObjectHashModelTableName, "as h").Select("h.*").
+		Joins("LEFT JOIN "+relation.ObjectInfoModelTableName+" as i ON h.hash = i.hash").
+		Where("h.engine = ? AND i.hash IS NULL", engine).
+		Limit(num).
+		Find(&list).Error
+	return list, utils.Wrap1(err)
+}

pkg/common/db/relation/object_info_model.go

@@ -6,6 +6,7 @@ import (
 	"OpenIM/pkg/utils"
 	"context"
 	"gorm.io/gorm"
+	"time"
 )
 
 func NewObjectInfo(db *gorm.DB) relation.ObjectInfoModelInterface {
@@ -43,3 +44,10 @@ func (o *ObjectInfoGorm) Take(ctx context.Context, name string) (info *relation.
 	info = &relation.ObjectInfoModel{}
 	return info, utils.Wrap1(o.DB.Where("name = ?", name).Take(info).Error)
 }
+
+func (o *ObjectInfoGorm) DeleteExpiration(ctx context.Context, expiration time.Time) (err error) {
+	defer func() {
+		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "expiration", expiration)
+	}()
+	return utils.Wrap1(o.DB.Where("expiration_time IS NOT NULL AND expiration_time <= ?", expiration).Delete(&relation.ObjectInfoModel{}).Error)
+}

pkg/common/db/relation/object_put_model.go

@@ -6,6 +6,7 @@ import (
 	"OpenIM/pkg/utils"
 	"context"
 	"gorm.io/gorm"
+	"time"
 )
 
 func NewObjectPut(db *gorm.DB) relation.ObjectPutModelInterface {
@@ -45,3 +46,18 @@ func (o *ObjectPutGorm) SetCompleted(ctx context.Context, putID string) (err err
 	}()
 	return utils.Wrap1(o.DB.Model(&relation.ObjectPutModel{}).Where("put_id = ?", putID).Update("complete", true).Error)
 }
+
+func (o *ObjectPutGorm) FindExpirationPut(ctx context.Context, expirationTime time.Time, num int) (list []*relation.ObjectPutModel, err error) {
+	defer func() {
+		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "expirationTime", expirationTime, "num", num, "list", list)
+	}()
+	err = o.DB.Where("effective_time <= ?", expirationTime).Limit(num).Find(&list).Error
+	return list, utils.Wrap1(err)
+}
+
+func (o *ObjectPutGorm) DelPut(ctx context.Context, ids []string) (err error) {
+	defer func() {
+		tracelog.SetCtxDebug(ctx, utils.GetFuncName(1), err, "ids", ids)
+	}()
+	return utils.Wrap1(o.DB.Where("put_id IN ?", ids).Delete(&relation.ObjectPutModel{}).Error)
+}