From 16dd4663adbbf5f6be9429e8fe2003f5acae06a2 Mon Sep 17 00:00:00 2001 From: Eric Lippmann Date: Thu, 30 Sep 2021 15:27:15 +0200 Subject: [PATCH] Move method DB.getSemaphoreForTable() --- pkg/icingadb/db.go | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/pkg/icingadb/db.go b/pkg/icingadb/db.go index 0be11d90..9a6dbb16 100644 --- a/pkg/icingadb/db.go +++ b/pkg/icingadb/db.go @@ -496,6 +496,19 @@ func (db *DB) Delete(ctx context.Context, entityType contracts.Entity, ids []int return db.DeleteStreamed(ctx, entityType, idsCh) } +func (db *DB) getSemaphoreForTable(table string) *semaphore.Weighted { + db.tableSemaphoresMu.Lock() + defer db.tableSemaphoresMu.Unlock() + + if sem, ok := db.tableSemaphores[table]; ok { + return sem + } else { + sem = semaphore.NewWeighted(int64(db.Options.MaxConnectionsPerTable)) + db.tableSemaphores[table] = sem + return sem + } +} + // IsRetryable checks whether the given error is retryable. func IsRetryable(err error) bool { if errors.Is(err, driver.ErrBadConn) { @@ -520,16 +533,3 @@ func IsRetryable(err error) bool { return false } - -func (db *DB) getSemaphoreForTable(table string) *semaphore.Weighted { - db.tableSemaphoresMu.Lock() - defer db.tableSemaphoresMu.Unlock() - - if sem, ok := db.tableSemaphores[table]; ok { - return sem - } else { - sem = semaphore.NewWeighted(int64(db.Options.MaxConnectionsPerTable)) - db.tableSemaphores[table] = sem - return sem - } -}