com.bazaarvoice.emodb.table.db.curator.TableMutexManager Maven / Gradle / Ivy

Go to download
package com.bazaarvoice.emodb.table.db.curator;

import com.bazaarvoice.emodb.table.db.Mutex;
import com.google.common.base.Charsets;
import com.google.common.hash.Hashing;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.utils.ZKPaths;

import java.time.Duration;

import static java.util.Objects.requireNonNull;

public class TableMutexManager {

    private static final int NUM_MUTEXES = 256;

    private final Mutex[] _mutexes;

    public TableMutexManager(CuratorFramework curator, String path) {
        _mutexes = new Mutex[NUM_MUTEXES];
        for (int i = 0; i < NUM_MUTEXES; i++) {
            _mutexes[i] = new CuratorMutex(curator, ZKPaths.makePath(path, Integer.toString(i)));
        }
    }

    public void runWithLockForTable(Runnable runnable, Duration acquireTimeout, String table) {
        requireNonNull(table, "table cannot be null");
        
        _mutexes[Math.abs(Hashing.murmur3_128().hashString(table, Charsets.UTF_8).asInt() % NUM_MUTEXES)]
                .runWithLock(runnable, acquireTimeout);

    }
}