prerna.zookeeper.ZKTesting Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.zookeeper;

import java.io.IOException;
import java.time.LocalDateTime;
import java.time.temporal.ChronoUnit;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import org.apache.curator.framework.recipes.locks.InterProcessSemaphoreV2;
import org.apache.curator.framework.recipes.locks.Lease;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.Watcher.Event.EventType;

import com.google.gson.Gson;

import prerna.engine.impl.storage.S3StorageEngine;
import prerna.util.Constants;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;


public class ZKTesting {

//	public static void main(String[] args) throws Exception {
//		ZKEngine engine = new ZKEngine();
//		try {
//			Properties prop = new Properties();
//			prop.put(ZKEngine.ZOOKEEPER_ADDRESS_KEY, "localhost:2181");
//			engine.open(prop);
//
//			ZKCuratorUtility utility = engine.getCuratorUtility();
//			startTimeLockThread(utility);
//			Thread.sleep(3_000);
//			printTimeLockDetails(utility);
//		} finally {
//			engine.close();
//		}
//	}

	private static final Logger classLogger = LogManager.getLogger(ZKTesting.class);

	public static void startTimeLockThread(ZKCuratorUtility utility) {
		Thread t = new Thread(new Runnable() {

			@Override
			public void run() {
				InterProcessSemaphoreV2 timebasedLock = null;
				Lease lease = null;
				try {
					timebasedLock = utility.getTimeBasedLock("/testLock", 1);
					lease = timebasedLock.acquire(10, TimeUnit.MINUTES);
					
					LocalDateTime startTime = LocalDateTime.now();
					
					Map dataMap = new HashMap<>();
					dataMap.put("startTime", startTime);
					dataMap.put("endTime", startTime.plusMinutes(10));
					dataMap.put("lockTime", 10);
					dataMap.put("lockUnit", TimeUnit.MINUTES);
					dataMap.put("user", "[email protected]");
					Gson gson = new Gson();
					byte[] data = gson.toJson(dataMap).getBytes();
					utility.createEphemeralNode("/userWithLock", data);
				} catch(Exception e) {
					classLogger.error(Constants.STACKTRACE, e);
				} finally {
					if(timebasedLock != null && lease != null) {
						timebasedLock.returnLease(lease);
					}
				}
			}
		});
		t.start();
	}

	public static void printTimeLockDetails(ZKCuratorUtility utility) throws Exception {
		byte[] data = utility.getDataFromNode("/userWithLock");
		String dataStr = new String(data);
		Map dataMap = new Gson().fromJson(dataStr, Map.class);
		System.out.println(dataMap);
		
		
		LocalDateTime currentTime = LocalDateTime.now();
		LocalDateTime endTime = new Gson().fromJson(dataMap.get("endTime").toString(), LocalDateTime.class);

		System.out.println("Time left for lock = " + currentTime.until(endTime, ChronoUnit.MINUTES));
		System.out.println("Time left for lock = " + currentTime.until(endTime, ChronoUnit.MINUTES));
		System.out.println("Time left for lock = " + currentTime.until(endTime, ChronoUnit.MINUTES));
		System.out.println("Time left for lock = " + currentTime.until(endTime, ChronoUnit.MINUTES));
		System.out.println("Time left for lock = " + currentTime.until(endTime, ChronoUnit.MINUTES));
		System.out.println("Time left for lock = " + currentTime.until(endTime, ChronoUnit.MINUTES));
		System.out.println("Time left for lock = " + currentTime.until(endTime, ChronoUnit.MINUTES));
	}
	
	public static void testWatcher(ZKCuratorUtility utility) throws Exception {
		utility.setWatcherForPath("/catalogPictures", new Watcher() {
			
			@Override
			public void process(WatchedEvent event) {
				if(event.getType() == EventType.NodeChildrenChanged) {
					String path = event.getPath();
					String itemName = path.substring(path.lastIndexOf("/"));
					
					S3StorageEngine engine = new S3StorageEngine();
					try {
						engine.copyToLocal("/ncrt/catalogPictures/" + itemName, "/opt/semosshome/catalogAttachments" + itemName);
					} catch (IOException | InterruptedException e) {
						classLogger.error(Constants.STACKTRACE, e);
					}
				}
				
			}
		});
	}
}