WEBPIECESxPACKAGE.base.libs.SimpleStorageImpl Maven / Gradle / Ivy

Go to download
package WEBPIECESxPACKAGE.base.libs;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletableFuture;

import javax.inject.Inject;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;

import org.webpieces.router.api.SimpleStorage;

public class SimpleStorageImpl implements SimpleStorage {
	
	private EntityManagerFactory factory;

	@Inject
	public SimpleStorageImpl(EntityManagerFactory factory) {
		this.factory = factory;
	}

	@Override
	public CompletableFuture save(String key, String subKey, String value) {
		EntityManager mgr = factory.createEntityManager();
		mgr.getTransaction().begin();

		mgr.persist(new SimpleStorageDbo(key, subKey, value));
		
		mgr.flush();
		mgr.getTransaction().commit();
		mgr.close();
		
		return CompletableFuture.completedFuture(null);
	}
	
	@Override
	public CompletableFuture save(String key, Map properties) {
		EntityManager mgr = factory.createEntityManager();
		mgr.getTransaction().begin();

		properties.forEach((mapKey, value) ->
			mgr.persist(new SimpleStorageDbo(key, mapKey, value))
		);
		
		mgr.flush();
		mgr.getTransaction().commit();
		mgr.close();
		
		return CompletableFuture.completedFuture(null);
	}

	@Override
	public CompletableFuture> read(String key) {
		EntityManager mgr = factory.createEntityManager();
		//mgr.getTransaction().begin();
		
		List rows = SimpleStorageDbo.findAll(mgr, key);
		Map properties = new HashMap<>();
		for(SimpleStorageDbo row: rows) {
			properties.put(row.getMapKey(), row.getValue());
		}
		
		mgr.close();
		
		return CompletableFuture.completedFuture(properties);
	}
	
}