io.odysz.common.DocLocks Maven / Gradle / Ivy

Go to download
/**
 * 
 */
package io.odysz.common;

import java.nio.file.Path;
import java.util.HashMap;
import java.util.concurrent.locks.ReentrantReadWriteLock;

/**
 * @since 1.4.12
 * 
 * @author [email protected]
 *
 */
public class DocLocks {

	private static HashMap locks;

	static {
		locks = new HashMap();
	}
	
	public static void reading(String fullpath) {
		if (!locks.containsKey(fullpath))
			locks.put(fullpath, new ReentrantReadWriteLock());
		locks.get(fullpath).readLock().lock();
	}

	public static void reading(Path p) {
		reading(p.toAbsolutePath().toString());
	}

	public static void readed(String fullpath) {
		try {
			if (locks.containsKey(fullpath))
				locks.get(fullpath).readLock().unlock();
		} catch (Throwable t) {
			Utils.warn("Unlock error: %s", fullpath);
			t.printStackTrace();
		}
		try { locks.remove(fullpath); }
		catch (Throwable t) {}
	}

	public static void readed(Path p) {
		readed(p.toAbsolutePath().toString());
	}

	public static void writing(String fullpath) {
		Utils.warn("lock:   %s", fullpath);
		if (!locks.containsKey(fullpath))
			locks.put(fullpath, new ReentrantReadWriteLock());
		locks.get(fullpath).writeLock().lock();
	}

	public static void writing(Path p) {
		writing(p.toAbsolutePath().toString());
	}

	public static void writen(String fullpath) {
		Utils.warn("unlock: %s", fullpath);
		try {
			locks.get(fullpath).writeLock().unlock();
		} catch (Throwable t) {
			Utils.warn("Unlock error: %s", fullpath);
			t.printStackTrace();
		}
	}

	public static void writen(Path p) {
		writen(p.toAbsolutePath().toString());
	}
}