refactor: refactor

aquaproj · Jan 21, 2025 · 337188f · 337188f
1 parent 0dadc30
commit 337188f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/pkg/controller/vacuum/queue_store.go b/pkg/controller/vacuum/queue_store.go
@@ -67,8 +67,8 @@ func (sq *StoreQueue) worker(ctx context.Context) {
 	}
 }
 
-// enqueue adds a task to the queue.
-func (sq *StoreQueue) enqueue(logE *logrus.Entry, pkg *Package) {
+// Enqueue adds a task to the queue.
+func (sq *StoreQueue) Enqueue(logE *logrus.Entry, pkg *Package) {
 	select {
 	case <-sq.done:
 		return

diff --git a/pkg/controller/vacuum/vacuum.go b/pkg/controller/vacuum/vacuum.go
@@ -106,7 +106,7 @@ func (vc *Controller) handleAsyncStorePackage(logE *logrus.Entry, vacuumPkg *Pac
 	if vacuumPkg == nil {
 		return errors.New("vacuumPkg is nil")
 	}
-	vc.db.storeQueue.enqueue(logE, vacuumPkg)
+	vc.db.StoreAsync(logE, vacuumPkg)
 	return nil
 }
 

diff --git a/pkg/controller/vacuum/vacuum_db.go b/pkg/controller/vacuum/vacuum_db.go
@@ -91,6 +91,10 @@ func (d *DB) Store(ctx context.Context, logE *logrus.Entry, pkg *Package, lastUs
 	})
 }
 
+func (d *DB) StoreAsync(logE *logrus.Entry, pkg *Package) {
+	d.storeQueue.Enqueue(logE, pkg)
+}
+
 // List lists all stored package entries.
 func (d *DB) List(ctx context.Context, logE *logrus.Entry) ([]*PackageVacuumEntry, error) {
 	db, err := d.getDB()