package org.warp.commonutils.locks;

import java.io.IOException;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.StampedLock;
import java.util.function.Supplier;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.warp.commonutils.functional.IORunnable;
import org.warp.commonutils.functional.IOSupplier;

public class LockUtils {

	public static void lock(@Nullable Lock lock, @NotNull Runnable r) {
		if (lock != null) {
			lock.lock();
		}
		try {
			r.run();
		} finally {
			if (lock != null) {
				lock.unlock();
			}
		}
	}

	public static void readLock(@Nullable StampedLock lock, @NotNull Runnable r) {
		long lockValue;
		if (lock != null) {
			lockValue = lock.readLock();
		} else {
			lockValue = 0;
		}
		try {
			r.run();
		} finally {
			if (lock != null) {
				lock.unlockRead(lockValue);
			}
		}
	}

	public static void writeLock(@Nullable StampedLock lock, @NotNull Runnable r) {
		long lockValue;
		if (lock != null) {
			lockValue = lock.writeLock();
		} else {
			lockValue = 0;
		}
		try {
			r.run();
		} finally {
			if (lock != null) {
				lock.unlockWrite(lockValue);
			}
		}
	}

	public static void lockIO(@Nullable Lock lock, @NotNull IORunnable r) throws IOException {
		if (lock != null) {
			lock.lock();
		}
		try {
			r.run();
		} finally {
			if (lock != null) {
				lock.unlock();
			}
		}
	}

	public static void readLockIO(@Nullable StampedLock lock, @NotNull IORunnable r) throws IOException {
		long lockValue;
		if (lock != null) {
			lockValue = lock.readLock();
		} else {
			lockValue = 0;
		}
		try {
			r.run();
		} finally {
			if (lock != null) {
				lock.unlockRead(lockValue);
			}
		}
	}

	public static void writeLockIO(@Nullable StampedLock lock, @NotNull IORunnable r) throws IOException {
		long lockValue;
		if (lock != null) {
			lockValue = lock.writeLock();
		} else {
			lockValue = 0;
		}
		try {
			r.run();
		} finally {
			if (lock != null) {
				lock.unlockWrite(lockValue);
			}
		}
	}

	public static <T> T lock(@Nullable Lock lock, @NotNull Supplier<T> r) {
		if (lock != null) {
			lock.lock();
		}
		try {
			return r.get();
		} finally {
			if (lock != null) {
				lock.unlock();
			}
		}
	}

	public static <T> T readLock(@Nullable StampedLock lock, @NotNull Supplier<T> r) {
		long lockValue;
		if (lock != null) {
			lockValue = lock.readLock();
		} else {
			lockValue = 0;
		}
		try {
			return r.get();
		} finally {
			if (lock != null) {
				lock.unlockRead(lockValue);
			}
		}
	}

	public static <T> T writeLock(@Nullable StampedLock lock, @NotNull Supplier<T> r) {
		long lockValue;
		if (lock != null) {
			lockValue = lock.writeLock();
		} else {
			lockValue = 0;
		}
		try {
			return r.get();
		} finally {
			if (lock != null) {
				lock.unlockWrite(lockValue);
			}
		}
	}

	public static <T> T lockIO(@Nullable Lock lock, @NotNull IOSupplier<T> r) throws IOException {
		if (lock != null) {
			lock.lock();
		}
		try {
			return r.get();
		} finally {
			if (lock != null) {
				lock.unlock();
			}
		}
	}

	public static <T> T readLockIO(@Nullable StampedLock lock, @NotNull IOSupplier<T> r) throws IOException {
		long lockValue;
		if (lock != null) {
			lockValue = lock.readLock();
		} else {
			lockValue = 0;
		}
		try {
			return r.get();
		} finally {
			if (lock != null) {
				lock.unlockRead(lockValue);
			}
		}
	}

	public static <T> T writeLockIO(@Nullable StampedLock lock, @NotNull IOSupplier<T> r) throws IOException {
		long lockValue;
		if (lock != null) {
			lockValue = lock.writeLock();
		} else {
			lockValue = 0;
		}
		try {
			return r.get();
		} finally {
			if (lock != null) {
				lock.unlockWrite(lockValue);
			}
		}
	}
}