From 74832a32dd4138f1508823f7492f66204fb94cdd Mon Sep 17 00:00:00 2001 From: Greg Burd Date: Thu, 12 Oct 2017 16:18:17 -0400 Subject: [PATCH] Formatting --- .../helenus/config/GetterMethodDetector.java | 2 +- .../core/AbstractAuditedEntityDraft.java | 47 ++- .../net/helenus/core/AbstractEntityDraft.java | 245 ++++++++-------- .../core/AbstractSessionOperations.java | 2 - .../net/helenus/core/AbstractUnitOfWork.java | 48 ++-- .../java/net/helenus/core/CommitThunk.java | 3 +- src/main/java/net/helenus/core/Helenus.java | 8 +- .../java/net/helenus/core/HelenusSession.java | 215 ++++++++------ .../net/helenus/core/PostCommitFunction.java | 37 ++- .../net/helenus/core/SessionInitializer.java | 33 +-- .../java/net/helenus/core/UnitOfWork.java | 52 ++-- .../java/net/helenus/core/UnitOfWorkImpl.java | 9 +- .../core/cache/EntityIdentifyingFacet.java | 8 +- .../java/net/helenus/core/cache/Facet.java | 3 +- .../AbstractFilterOptionalOperation.java | 4 +- .../AbstractFilterStreamOperation.java | 4 +- .../core/operation/AbstractOperation.java | 4 - .../operation/AbstractStatementOperation.java | 10 +- .../operation/AbstractStreamOperation.java | 10 - .../operation/BoundOptionalOperation.java | 4 +- .../core/operation/BoundStreamOperation.java | 11 +- .../core/operation/InsertOperation.java | 38 +-- .../net/helenus/core/operation/Operation.java | 99 ++++--- .../operation/PreparedStreamOperation.java | 1 - .../core/operation/SelectFirstOperation.java | 4 +- .../SelectFirstTransformingOperation.java | 4 +- .../core/operation/SelectOperation.java | 58 ++-- .../SelectTransformingOperation.java | 5 +- .../core/operation/UpdateOperation.java | 25 +- .../net/helenus/core/reflect/Drafted.java | 1 - .../core/reflect/DslInvocationHandler.java | 70 ++--- .../core/reflect/MapperInvocationHandler.java | 3 +- .../net/helenus/mapping/HelenusEntity.java | 4 +- .../helenus/mapping/HelenusMappingEntity.java | 59 ++-- .../javatype/AbstractCollectionJavaType.java | 5 +- .../mapping/javatype/AbstractJavaType.java | 4 +- .../helenus/mapping/javatype/MapJavaType.java | 1 - .../type/AbstractCollectionDataType.java | 11 +- .../mapping/type/AbstractDataType.java | 5 +- .../mapping/type/UDTKeyMapDataType.java | 2 - .../value/BeanColumnValueProvider.java | 2 +- .../core/collection/CollectionTest.java | 41 +-- .../core/draft/EntityDraftBuilderTest.java | 99 ++++--- .../integration/core/draft/Inventory.java | 125 ++++---- .../test/integration/core/draft/Supply.java | 218 +++++++------- .../core/hierarchy/HierarchyTest.java | 3 +- .../integration/core/hierarchy/Mammal.java | 5 +- .../core/simple/SimpleUserTest.java | 73 ++--- .../core/tuplecollection/TupleMapTest.java | 16 +- .../core/unitofwork/AndThenOrderTest.java | 182 +++++++----- .../core/unitofwork/Directory.java | 9 +- .../integration/core/unitofwork/File.java | 6 +- .../core/unitofwork/FileAttributes.java | 3 +- .../core/unitofwork/FilesystemNode.java | 16 +- .../core/unitofwork/UnitOfWorkTest.java | 271 +++++++++--------- .../helenus/test/unit/core/dsl/Account.java | 1 - 56 files changed, 1133 insertions(+), 1095 deletions(-) diff --git a/src/main/java/net/helenus/config/GetterMethodDetector.java b/src/main/java/net/helenus/config/GetterMethodDetector.java index 266a7e7..60a9ec0 100644 --- a/src/main/java/net/helenus/config/GetterMethodDetector.java +++ b/src/main/java/net/helenus/config/GetterMethodDetector.java @@ -35,7 +35,7 @@ public enum GetterMethodDetector implements Function { } if (Modifier.isStatic(method.getModifiers())) { - return false; + return false; } // Methods marked "Transient" are not mapped, skip them. diff --git a/src/main/java/net/helenus/core/AbstractAuditedEntityDraft.java b/src/main/java/net/helenus/core/AbstractAuditedEntityDraft.java index 5ec76d3..a9a09e2 100644 --- a/src/main/java/net/helenus/core/AbstractAuditedEntityDraft.java +++ b/src/main/java/net/helenus/core/AbstractAuditedEntityDraft.java @@ -1,41 +1,38 @@ package net.helenus.core; -import net.helenus.core.reflect.MapExportable; - import java.time.LocalDateTime; import java.time.ZoneId; import java.util.Date; - +import net.helenus.core.reflect.MapExportable; public abstract class AbstractAuditedEntityDraft extends AbstractEntityDraft { - public AbstractAuditedEntityDraft(MapExportable entity) { - super(entity); + public AbstractAuditedEntityDraft(MapExportable entity) { + super(entity); - Date in = new Date(); - LocalDateTime ldt = LocalDateTime.ofInstant(in.toInstant(), ZoneId.systemDefault()); - Date now = Date.from(ldt.atZone(ZoneId.systemDefault()).toInstant()); + Date in = new Date(); + LocalDateTime ldt = LocalDateTime.ofInstant(in.toInstant(), ZoneId.systemDefault()); + Date now = Date.from(ldt.atZone(ZoneId.systemDefault()).toInstant()); - String who = getCurrentAuditor(); + String who = getCurrentAuditor(); - if (entity == null) { - if (who != null) { - set("createdBy", who); - } - set("createdAt", now); - } - if (who != null) { - set("modifiedBy", who); - } - set("modifiedAt", now); + if (entity == null) { + if (who != null) { + set("createdBy", who); + } + set("createdAt", now); } - - protected String getCurrentAuditor() { - return null; + if (who != null) { + set("modifiedBy", who); } + set("modifiedAt", now); + } - public Date createdAt() { - return (Date) get("createdAt", Date.class); - } + protected String getCurrentAuditor() { + return null; + } + public Date createdAt() { + return (Date) get("createdAt", Date.class); + } } diff --git a/src/main/java/net/helenus/core/AbstractEntityDraft.java b/src/main/java/net/helenus/core/AbstractEntityDraft.java index 58a575d..fbccb3b 100644 --- a/src/main/java/net/helenus/core/AbstractEntityDraft.java +++ b/src/main/java/net/helenus/core/AbstractEntityDraft.java @@ -1,164 +1,159 @@ package net.helenus.core; -import java.util.*; - import com.google.common.primitives.Primitives; - +import java.util.*; import net.helenus.core.reflect.DefaultPrimitiveTypes; import net.helenus.core.reflect.Drafted; import net.helenus.core.reflect.MapExportable; import net.helenus.mapping.MappingUtil; - public abstract class AbstractEntityDraft implements Drafted { - private final Map backingMap = new HashMap(); - private final MapExportable entity; - private final Map entityMap; + private final Map backingMap = new HashMap(); + private final MapExportable entity; + private final Map entityMap; + public AbstractEntityDraft(MapExportable entity) { + this.entity = entity; + this.entityMap = entity != null ? entity.toMap() : new HashMap(); + } - public AbstractEntityDraft(MapExportable entity) { - this.entity = entity; - this.entityMap = entity != null ? entity.toMap() : new HashMap(); - } + public abstract Class getEntityClass(); - public abstract Class getEntityClass(); + public E build() { + return Helenus.map(getEntityClass(), toMap()); + } - public E build() { return Helenus.map(getEntityClass(), toMap()); } + @SuppressWarnings("unchecked") + protected T get(Getter getter, Class returnType) { + return (T) get(this.methodNameFor(getter), returnType); + } - @SuppressWarnings("unchecked") - protected T get(Getter getter, Class returnType) { - return (T) get(this.methodNameFor(getter), returnType); - } + @SuppressWarnings("unchecked") + protected T get(String key, Class returnType) { + T value = (T) backingMap.get(key); - @SuppressWarnings("unchecked") - protected T get(String key, Class returnType) { - T value = (T) backingMap.get(key); + if (value == null) { + value = (T) entityMap.get(key); + if (value == null) { - if (value == null) { - value = (T) entityMap.get(key); - if (value == null) { + if (Primitives.allPrimitiveTypes().contains(returnType)) { - if (Primitives.allPrimitiveTypes().contains(returnType)) { + DefaultPrimitiveTypes type = DefaultPrimitiveTypes.lookup(returnType); + if (type == null) { + throw new RuntimeException("unknown primitive type " + returnType); + } - DefaultPrimitiveTypes type = DefaultPrimitiveTypes.lookup(returnType); - if (type == null) { - throw new RuntimeException("unknown primitive type " + returnType); - } - - return (T) type.getDefaultValue(); - } - } + return (T) type.getDefaultValue(); } - - return value; + } } - protected Object set(Getter getter, Object value) { - return set(this.methodNameFor(getter), value); + return value; + } + + protected Object set(Getter getter, Object value) { + return set(this.methodNameFor(getter), value); + } + + protected Object set(String key, Object value) { + if (key == null || value == null) { + return null; } - protected Object set(String key, Object value) { - if (key == null || value == null) { - return null; - } + backingMap.put(key, value); + return value; + } + @SuppressWarnings("unchecked") + protected T mutate(Getter getter, T value) { + return (T) mutate(this.methodNameFor(getter), value); + } + + protected Object mutate(String key, Object value) { + Objects.requireNonNull(key); + + if (value == null) { + return null; + } + + if (entity != null) { + Map map = entity.toMap(); + + if (map.containsKey(key) && !value.equals(map.get(key))) { backingMap.put(key, value); return value; + } + + return map.get(key); + } else { + backingMap.put(key, value); + + return null; } + } - @SuppressWarnings("unchecked") - protected T mutate(Getter getter, T value) { - return (T) mutate(this.methodNameFor(getter), value); + private String methodNameFor(Getter getter) { + return MappingUtil.resolveMappingProperty(getter).getProperty().getPropertyName(); + } + + public Object unset(Getter getter) { + return unset(methodNameFor(getter)); + } + + public Object unset(String key) { + if (key != null) { + Object value = backingMap.get(key); + backingMap.put(key, null); + return value; } + return null; + } - protected Object mutate(String key, Object value) { - Objects.requireNonNull(key); + public boolean reset(Getter getter, T desiredValue) { + return this.reset(this.methodNameFor(getter), desiredValue); + } - if (value == null) { - return null; - } - - if (entity != null) { - Map map = entity.toMap(); - - if (map.containsKey(key) && !value.equals(map.get(key))) { - backingMap.put(key, value); - return value; - } - - return map.get(key); - } else { - backingMap.put(key, value); - - return null; - } + public boolean reset(String key, T desiredValue) { + if (key != null && desiredValue != null) { + @SuppressWarnings("unchecked") + T currentValue = (T) backingMap.get(key); + if (currentValue == null || !currentValue.equals(desiredValue)) { + set(key, desiredValue); + return true; + } } + return false; + } - private String methodNameFor(Getter getter) { - return MappingUtil.resolveMappingProperty(getter) - .getProperty() - .getPropertyName(); + @Override + public Map toMap() { + return toMap(entityMap); + } + + public Map toMap(Map entityMap) { + Map combined; + if (entityMap != null && entityMap.size() > 0) { + combined = new HashMap(entityMap.size()); + for (String key : entityMap.keySet()) { + combined.put(key, entityMap.get(key)); + } + } else { + combined = new HashMap(backingMap.size()); } - - public Object unset(Getter getter) { - return unset(methodNameFor(getter)); + for (String key : mutated()) { + combined.put(key, backingMap.get(key)); } + return combined; + } - public Object unset(String key) { - if (key != null) { - Object value = backingMap.get(key); - backingMap.put(key, null); - return value; - } - return null; - } - - public boolean reset(Getter getter, T desiredValue) { - return this.reset(this.methodNameFor(getter), desiredValue); - } - - public boolean reset(String key, T desiredValue) { - if (key != null && desiredValue != null) { - @SuppressWarnings("unchecked") - T currentValue = (T) backingMap.get(key); - if (currentValue == null || !currentValue.equals(desiredValue)) { - set(key, desiredValue); - return true; - } - } - return false; - } - - @Override - public Map toMap() { - return toMap(entityMap); - } - - public Map toMap(MapentityMap) { - Map combined; - if (entityMap != null && entityMap.size() > 0) { - combined = new HashMap(entityMap.size()); - for (String key : entityMap.keySet()) { - combined.put(key, entityMap.get(key)); - } - } else { - combined = new HashMap(backingMap.size()); - } - for (String key : mutated()) { - combined.put(key, backingMap.get(key)); - } - return combined; - } - - @Override - public Set mutated() { - return backingMap.keySet(); - } - - @Override - public String toString() { - return backingMap.toString(); - } + @Override + public Set mutated() { + return backingMap.keySet(); + } + @Override + public String toString() { + return backingMap.toString(); + } } diff --git a/src/main/java/net/helenus/core/AbstractSessionOperations.java b/src/main/java/net/helenus/core/AbstractSessionOperations.java index 1bba0ab..5721051 100644 --- a/src/main/java/net/helenus/core/AbstractSessionOperations.java +++ b/src/main/java/net/helenus/core/AbstractSessionOperations.java @@ -22,7 +22,6 @@ import com.datastax.driver.core.querybuilder.BuiltStatement; import com.google.common.util.concurrent.ListenableFuture; import java.io.PrintStream; import java.util.concurrent.Executor; - import net.helenus.mapping.value.ColumnValuePreparer; import net.helenus.mapping.value.ColumnValueProvider; import net.helenus.support.HelenusException; @@ -124,5 +123,4 @@ public abstract class AbstractSessionOperations { void printCql(String cql) { getPrintStream().println(cql); } - } diff --git a/src/main/java/net/helenus/core/AbstractUnitOfWork.java b/src/main/java/net/helenus/core/AbstractUnitOfWork.java index dfb1ff7..c621770 100644 --- a/src/main/java/net/helenus/core/AbstractUnitOfWork.java +++ b/src/main/java/net/helenus/core/AbstractUnitOfWork.java @@ -17,10 +17,8 @@ package net.helenus.core; import com.diffplug.common.base.Errors; import com.google.common.collect.TreeTraverser; - import java.util.*; - /** Encapsulates the concept of a "transaction" as a unit-of-work. */ public abstract class AbstractUnitOfWork implements UnitOfWork, AutoCloseable { private final List> nested = new ArrayList<>(); @@ -70,22 +68,25 @@ public abstract class AbstractUnitOfWork implements UnitOfW return null; } - public Map> getCache() { return cache; } + public Map> getCache() { + return cache; + } private Iterator> getChildNodes() { return nested.iterator(); } - /** - * Checks to see if the work performed between calling begin and now can be committed or not. - * - * @return a function from which to chain work that only happens when commit is successful - * @throws E when the work overlaps with other concurrent writers. - */ + /** + * Checks to see if the work performed between calling begin and now can be committed or not. + * + * @return a function from which to chain work that only happens when commit is successful + * @throws E when the work overlaps with other concurrent writers. + */ public PostCommitFunction commit() throws E { // All nested UnitOfWork should be committed (not aborted) before calls to commit, check. boolean canCommit = true; - TreeTraverser> traverser = TreeTraverser.using(node -> node::getChildNodes); + TreeTraverser> traverser = + TreeTraverser.using(node -> node::getChildNodes); for (AbstractUnitOfWork uow : traverser.postOrderTraversal(this)) { if (this != uow) { canCommit &= (!uow.aborted && uow.committed); @@ -112,7 +113,8 @@ public abstract class AbstractUnitOfWork implements UnitOfW if (parentCache.containsKey(key)) { // merge the sets Set ps = parentCache.get(key); - ps.addAll(cache.get(key)); //TODO(gburd): review this, likely not correct in all cases as-is. + ps.addAll( + cache.get(key)); //TODO(gburd): review this, likely not correct in all cases as-is. } else { // add the missing set parentCache.put(key, cache.get(key)); @@ -122,9 +124,12 @@ public abstract class AbstractUnitOfWork implements UnitOfW // Apply all post-commit functions for if (parent == null) { - traverser.postOrderTraversal(this).forEach(uow -> { - uow.applyPostCommitFunctions(); - }); + traverser + .postOrderTraversal(this) + .forEach( + uow -> { + uow.applyPostCommitFunctions(); + }); return new PostCommitFunction(this, null); } } @@ -137,11 +142,15 @@ public abstract class AbstractUnitOfWork implements UnitOfW /* Explicitly discard the work and mark it as as such in the log. */ public void abort() { - TreeTraverser> traverser = TreeTraverser.using(node -> node::getChildNodes); - traverser.postOrderTraversal(this).forEach(uow -> { - uow.committed = false; - uow.aborted = true; - }); + TreeTraverser> traverser = + TreeTraverser.using(node -> node::getChildNodes); + traverser + .postOrderTraversal(this) + .forEach( + uow -> { + uow.committed = false; + uow.aborted = true; + }); // log.record(txn::abort) // cache.invalidateSince(txn::start time) } @@ -165,5 +174,4 @@ public abstract class AbstractUnitOfWork implements UnitOfW public boolean hasCommitted() { return committed; } - } diff --git a/src/main/java/net/helenus/core/CommitThunk.java b/src/main/java/net/helenus/core/CommitThunk.java index c200061..ff50f4a 100644 --- a/src/main/java/net/helenus/core/CommitThunk.java +++ b/src/main/java/net/helenus/core/CommitThunk.java @@ -1,8 +1,7 @@ package net.helenus.core; -import java.util.function.Function; @FunctionalInterface public interface CommitThunk { - void apply(); + void apply(); } diff --git a/src/main/java/net/helenus/core/Helenus.java b/src/main/java/net/helenus/core/Helenus.java index 48c1aba..c0d9bc6 100644 --- a/src/main/java/net/helenus/core/Helenus.java +++ b/src/main/java/net/helenus/core/Helenus.java @@ -142,10 +142,10 @@ public final class Helenus { public static HelenusEntity entity(Class iface) { Metadata metadata = metadataForEntity.get(iface); if (metadata == null) { - HelenusSession session = session(); - if (session != null) { - metadata = session.getMetadata(); - } + HelenusSession session = session(); + if (session != null) { + metadata = session.getMetadata(); + } } return entity(iface, metadata); } diff --git a/src/main/java/net/helenus/core/HelenusSession.java b/src/main/java/net/helenus/core/HelenusSession.java index 785fba6..8fe9f90 100644 --- a/src/main/java/net/helenus/core/HelenusSession.java +++ b/src/main/java/net/helenus/core/HelenusSession.java @@ -15,10 +15,21 @@ */ package net.helenus.core; +import static net.helenus.core.Query.eq; + import brave.Tracer; -import com.codahale.metrics.Meter; import com.codahale.metrics.MetricRegistry; import com.datastax.driver.core.*; +import java.io.Closeable; +import java.io.PrintStream; +import java.lang.reflect.Constructor; +import java.lang.reflect.InvocationTargetException; +import java.util.Map; +import java.util.Objects; +import java.util.Optional; +import java.util.Set; +import java.util.concurrent.Executor; +import java.util.function.Function; import net.helenus.core.operation.*; import net.helenus.core.reflect.Drafted; import net.helenus.core.reflect.HelenusPropertyNode; @@ -33,19 +44,6 @@ import net.helenus.support.Fun.Tuple6; import net.helenus.support.HelenusException; import net.helenus.support.HelenusMappingException; -import java.io.Closeable; -import java.io.PrintStream; -import java.lang.reflect.Constructor; -import java.lang.reflect.InvocationTargetException; -import java.util.Map; -import java.util.Objects; -import java.util.Optional; -import java.util.Set; -import java.util.concurrent.Executor; -import java.util.function.Function; - -import static net.helenus.core.Query.eq; - public final class HelenusSession extends AbstractSessionOperations implements Closeable { private final int MAX_CACHE_SIZE = 10000; @@ -69,26 +67,25 @@ public final class HelenusSession extends AbstractSessionOperations implements C private final StatementColumnValuePreparer valuePreparer; private final Metadata metadata; - HelenusSession( - Session session, - String usingKeyspace, - CodecRegistry registry, - boolean showCql, - PrintStream printStream, - SessionRepositoryBuilder sessionRepositoryBuilder, - Executor executor, - boolean dropSchemaOnClose, - ConsistencyLevel consistencyLevel, - boolean defaultQueryIdempotency, - Class unitOfWorkClass, - MetricRegistry metricRegistry, - Tracer tracer) { + Session session, + String usingKeyspace, + CodecRegistry registry, + boolean showCql, + PrintStream printStream, + SessionRepositoryBuilder sessionRepositoryBuilder, + Executor executor, + boolean dropSchemaOnClose, + ConsistencyLevel consistencyLevel, + boolean defaultQueryIdempotency, + Class unitOfWorkClass, + MetricRegistry metricRegistry, + Tracer tracer) { this.session = session; this.registry = registry == null ? CodecRegistry.DEFAULT_INSTANCE : registry; this.usingKeyspace = - Objects.requireNonNull( - usingKeyspace, "keyspace needs to be selected before creating session"); + Objects.requireNonNull( + usingKeyspace, "keyspace needs to be selected before creating session"); this.showCql = showCql; this.printStream = printStream; this.sessionRepository = sessionRepositoryBuilder.build(); @@ -177,42 +174,53 @@ public final class HelenusSession extends AbstractSessionOperations implements C } @Override - public boolean getDefaultQueryIdempotency() { return defaultQueryIdempotency; } + public boolean getDefaultQueryIdempotency() { + return defaultQueryIdempotency; + } - public Metadata getMetadata() { return metadata; } + public Metadata getMetadata() { + return metadata; + } public synchronized T begin() { - return begin(null); + return begin(null); } public synchronized T begin(T parent) { try { Class clazz = unitOfWorkClass; - Constructor ctor = clazz.getConstructor(HelenusSession.class, UnitOfWork.class); + Constructor ctor = + clazz.getConstructor(HelenusSession.class, UnitOfWork.class); UnitOfWork uow = ctor.newInstance(this, parent); if (parent != null) { parent.addNestedUnitOfWork(uow); } return (T) uow.begin(); - } - catch (NoSuchMethodException | InvocationTargetException | InstantiationException | IllegalAccessException e) { - throw new HelenusException(String.format("Unable to instantiate {} as a UnitOfWork.", unitOfWorkClass.getSimpleName()), e); + } catch (NoSuchMethodException + | InvocationTargetException + | InstantiationException + | IllegalAccessException e) { + throw new HelenusException( + String.format( + "Unable to instantiate {} as a UnitOfWork.", unitOfWorkClass.getSimpleName()), + e); } } public SelectOperation select(E pojo) { - Objects.requireNonNull(pojo, "supplied object must be a dsl for a registered entity but cannot be null"); + Objects.requireNonNull( + pojo, "supplied object must be a dsl for a registered entity but cannot be null"); ColumnValueProvider valueProvider = getValueProvider(); HelenusEntity entity = Helenus.resolve(pojo); Class entityClass = entity.getMappingInterface(); return new SelectOperation( - this, - entity, - (r) -> { - Map map = new ValueProviderMap(r, valueProvider, entity); - return (E) Helenus.map(entityClass, map); - }); + this, + entity, + (r) -> { + Map map = new ValueProviderMap(r, valueProvider, entity); + return (E) Helenus.map(entityClass, map); + }); } public SelectOperation select(Class entityClass) { @@ -221,12 +229,12 @@ public final class HelenusSession extends AbstractSessionOperations implements C HelenusEntity entity = Helenus.entity(entityClass); return new SelectOperation( - this, - entity, - (r) -> { - Map map = new ValueProviderMap(r, valueProvider, entity); - return (E) Helenus.map(entityClass, map); - }); + this, + entity, + (r) -> { + Map map = new ValueProviderMap(r, valueProvider, entity); + return (E) Helenus.map(entityClass, map); + }); } public SelectOperation select() { @@ -239,7 +247,8 @@ public final class HelenusSession extends AbstractSessionOperations implements C } public SelectOperation selectAll(E pojo) { - Objects.requireNonNull(pojo, "supplied object must be a dsl for a registered entity but cannot be null"); + Objects.requireNonNull( + pojo, "supplied object must be a dsl for a registered entity but cannot be null"); HelenusEntity entity = Helenus.resolve(pojo); return new SelectOperation(this, entity); } @@ -411,51 +420,62 @@ public final class HelenusSession extends AbstractSessionOperations implements C public UpdateOperation update(Drafted drafted) { if (drafted instanceof AbstractEntityDraft == false) { - throw new HelenusMappingException("update of draft objects that don't inherit from AbstractEntityDraft is not yet supported"); + throw new HelenusMappingException( + "update of draft objects that don't inherit from AbstractEntityDraft is not yet supported"); } - AbstractEntityDraft draft = (AbstractEntityDraft)drafted; + AbstractEntityDraft draft = (AbstractEntityDraft) drafted; UpdateOperation update = new UpdateOperation(this, draft); Map map = draft.toMap(); Set mutatedProperties = draft.mutated(); HelenusEntity entity = Helenus.entity(draft.getEntityClass()); // Add all the mutated values contained in the draft. - entity.getOrderedProperties().forEach(property -> { - switch (property.getColumnType()) { - case PARTITION_KEY: - case CLUSTERING_COLUMN: - break; - default: - String propertyName = property.getPropertyName(); - if (mutatedProperties.contains(propertyName)) { - Object value = map.get(propertyName); - Getter getter = new Getter() { - @Override - public Object get() { - throw new DslPropertyException(new HelenusPropertyNode(property, Optional.empty())); + entity + .getOrderedProperties() + .forEach( + property -> { + switch (property.getColumnType()) { + case PARTITION_KEY: + case CLUSTERING_COLUMN: + break; + default: + String propertyName = property.getPropertyName(); + if (mutatedProperties.contains(propertyName)) { + Object value = map.get(propertyName); + Getter getter = + new Getter() { + @Override + public Object get() { + throw new DslPropertyException( + new HelenusPropertyNode(property, Optional.empty())); + } + }; + update.set(getter, value); + } } - }; - update.set(getter, value); - } - } - }); + }); // Add the partition and clustering keys if they were in the draft (normally the case). - entity.getOrderedProperties().forEach(property -> { - switch (property.getColumnType()) { - case PARTITION_KEY: - case CLUSTERING_COLUMN: - String propertyName = property.getPropertyName(); - Object value = map.get(propertyName); - Getter getter = new Getter() { - @Override - public Object get() { - throw new DslPropertyException(new HelenusPropertyNode(property, Optional.empty())); - } - }; - update.where(getter, eq(value)); - } - }); + entity + .getOrderedProperties() + .forEach( + property -> { + switch (property.getColumnType()) { + case PARTITION_KEY: + case CLUSTERING_COLUMN: + String propertyName = property.getPropertyName(); + Object value = map.get(propertyName); + Getter getter = + new Getter() { + @Override + public Object get() { + throw new DslPropertyException( + new HelenusPropertyNode(property, Optional.empty())); + } + }; + update.where(getter, eq(value)); + } + }); return update; } @@ -478,9 +498,14 @@ public final class HelenusSession extends AbstractSessionOperations implements C } public InsertOperation insert(T pojo) { - Objects.requireNonNull(pojo, "supplied object must be either an instance of the entity class or a dsl for it, but cannot be null"); + Objects.requireNonNull( + pojo, + "supplied object must be either an instance of the entity class or a dsl for it, but cannot be null"); HelenusEntity entity = null; - try { entity = Helenus.resolve(pojo); } catch (HelenusMappingException e) {} + try { + entity = Helenus.resolve(pojo); + } catch (HelenusMappingException e) { + } if (entity != null) { return new InsertOperation(this, entity.getMappingInterface(), true); } else { @@ -488,7 +513,9 @@ public final class HelenusSession extends AbstractSessionOperations implements C } } - public InsertOperation insert(Drafted draft) { return insert(draft.build(), draft.mutated()); } + public InsertOperation insert(Drafted draft) { + return insert(draft.build(), draft.mutated()); + } private InsertOperation insert(T pojo, Set mutations) { Objects.requireNonNull(pojo, "pojo is empty"); @@ -512,9 +539,14 @@ public final class HelenusSession extends AbstractSessionOperations implements C } public InsertOperation upsert(T pojo) { - Objects.requireNonNull(pojo, "supplied object must be either an instance of the entity class or a dsl for it, but cannot be null"); + Objects.requireNonNull( + pojo, + "supplied object must be either an instance of the entity class or a dsl for it, but cannot be null"); HelenusEntity entity = null; - try { entity = Helenus.resolve(pojo); } catch (HelenusMappingException e) {} + try { + entity = Helenus.resolve(pojo); + } catch (HelenusMappingException e) { + } if (entity != null) { return new InsertOperation(this, entity.getMappingInterface(), false); } else { @@ -587,5 +619,4 @@ public final class HelenusSession extends AbstractSessionOperations implements C break; } } - } diff --git a/src/main/java/net/helenus/core/PostCommitFunction.java b/src/main/java/net/helenus/core/PostCommitFunction.java index 22835c5..3ba2b8b 100644 --- a/src/main/java/net/helenus/core/PostCommitFunction.java +++ b/src/main/java/net/helenus/core/PostCommitFunction.java @@ -1,30 +1,29 @@ package net.helenus.core; - -import java.util.Objects; import java.util.*; +import java.util.Objects; public class PostCommitFunction implements java.util.function.Function { - private final UnitOfWork uow; - private final List postCommit; + private final UnitOfWork uow; + private final List postCommit; - PostCommitFunction(UnitOfWork uow, List postCommit) { - this.uow = uow; - this.postCommit = postCommit; - } + PostCommitFunction(UnitOfWork uow, List postCommit) { + this.uow = uow; + this.postCommit = postCommit; + } - public void andThen(CommitThunk after) { - Objects.requireNonNull(after); - if (postCommit == null) { - after.apply(); - } else { - postCommit.add(after); - } + public void andThen(CommitThunk after) { + Objects.requireNonNull(after); + if (postCommit == null) { + after.apply(); + } else { + postCommit.add(after); } + } - @Override - public R apply(T t) { - return null; - } + @Override + public R apply(T t) { + return null; + } } diff --git a/src/main/java/net/helenus/core/SessionInitializer.java b/src/main/java/net/helenus/core/SessionInitializer.java index d7f0d7e..27a0312 100644 --- a/src/main/java/net/helenus/core/SessionInitializer.java +++ b/src/main/java/net/helenus/core/SessionInitializer.java @@ -25,7 +25,6 @@ import java.util.*; import java.util.concurrent.Executor; import java.util.concurrent.Executors; import java.util.function.Consumer; - import net.helenus.core.reflect.DslExportable; import net.helenus.mapping.HelenusEntity; import net.helenus.mapping.HelenusEntityType; @@ -131,12 +130,12 @@ public final class SessionInitializer extends AbstractSessionOperations { } public SessionInitializer idempotentQueryExecution(boolean idempotent) { - this.idempotent = idempotent; - return this; + this.idempotent = idempotent; + return this; } public boolean getDefaultQueryIdempotency() { - return idempotent; + return idempotent; } @Override @@ -185,9 +184,10 @@ public final class SessionInitializer extends AbstractSessionOperations { PackageUtil.getClasses(packageName) .stream() .filter(c -> c.isInterface() && !c.isAnnotation()) - .forEach(clazz -> { - initList.add(Either.right(clazz)); - }); + .forEach( + clazz -> { + initList.add(Either.right(clazz)); + }); } catch (IOException | ClassNotFoundException e) { throw new HelenusException("fail to add package " + packageName, e); } @@ -267,18 +267,19 @@ public final class SessionInitializer extends AbstractSessionOperations { Objects.requireNonNull(usingKeyspace, "please define keyspace by 'use' operator"); - initList.forEach((either) -> { - Class iface = null; - if (either.isLeft()) { + initList.forEach( + (either) -> { + Class iface = null; + if (either.isLeft()) { iface = MappingUtil.getMappingInterface(either.getLeft()); - } else { + } else { iface = either.getRight(); - } + } - DslExportable dsl = (DslExportable) Helenus.dsl(iface); - dsl.setMetadata(session.getCluster().getMetadata()); - sessionRepository.add(dsl); - }); + DslExportable dsl = (DslExportable) Helenus.dsl(iface); + dsl.setMetadata(session.getCluster().getMetadata()); + sessionRepository.add(dsl); + }); TableOperations tableOps = new TableOperations(this, dropUnusedColumns, dropUnusedIndexes); UserTypeOperations userTypeOps = new UserTypeOperations(this, dropUnusedColumns); diff --git a/src/main/java/net/helenus/core/UnitOfWork.java b/src/main/java/net/helenus/core/UnitOfWork.java index 0cfa536..f27b76c 100644 --- a/src/main/java/net/helenus/core/UnitOfWork.java +++ b/src/main/java/net/helenus/core/UnitOfWork.java @@ -15,44 +15,42 @@ */ package net.helenus.core; -import net.helenus.support.Either; import java.util.Map; import java.util.Set; public interface UnitOfWork extends AutoCloseable { - /** - * Marks the beginning of a transactional section of work. Will write a record to the shared - * write-ahead log. - * - * @return the handle used to commit or abort the work. - */ - UnitOfWork begin(); + /** + * Marks the beginning of a transactional section of work. Will write a record to the shared + * write-ahead log. + * + * @return the handle used to commit or abort the work. + */ + UnitOfWork begin(); - UnitOfWork addNestedUnitOfWork(UnitOfWork uow); + UnitOfWork addNestedUnitOfWork(UnitOfWork uow); - /** - * Checks to see if the work performed between calling begin and now can be committed or not. - * - * @return a function from which to chain work that only happens when commit is successful - * @throws E when the work overlaps with other concurrent writers. - */ - PostCommitFunction commit() throws E; + /** + * Checks to see if the work performed between calling begin and now can be committed or not. + * + * @return a function from which to chain work that only happens when commit is successful + * @throws E when the work overlaps with other concurrent writers. + */ + PostCommitFunction commit() throws E; - /** - * Explicitly abort the work within this unit of work. Any nested aborted unit of work - * will trigger the entire unit of work to commit. - */ - void abort(); + /** + * Explicitly abort the work within this unit of work. Any nested aborted unit of work will + * trigger the entire unit of work to commit. + */ + void abort(); + boolean hasAborted(); - boolean hasAborted(); + boolean hasCommitted(); - boolean hasCommitted(); + //Either> cacheLookup(String key); + Set cacheLookup(String key); - //Either> cacheLookup(String key); - Set cacheLookup(String key); - - Map> getCache(); + Map> getCache(); } diff --git a/src/main/java/net/helenus/core/UnitOfWorkImpl.java b/src/main/java/net/helenus/core/UnitOfWorkImpl.java index b9aab3b..52cae59 100644 --- a/src/main/java/net/helenus/core/UnitOfWorkImpl.java +++ b/src/main/java/net/helenus/core/UnitOfWorkImpl.java @@ -19,9 +19,8 @@ import net.helenus.support.HelenusException; class UnitOfWorkImpl extends AbstractUnitOfWork { - @SuppressWarnings("unchecked") - public UnitOfWorkImpl(HelenusSession session, UnitOfWork parent) { - super(session, (AbstractUnitOfWork) parent); - } - + @SuppressWarnings("unchecked") + public UnitOfWorkImpl(HelenusSession session, UnitOfWork parent) { + super(session, (AbstractUnitOfWork) parent); + } } diff --git a/src/main/java/net/helenus/core/cache/EntityIdentifyingFacet.java b/src/main/java/net/helenus/core/cache/EntityIdentifyingFacet.java index 8bf3b8b..b4a7cf9 100644 --- a/src/main/java/net/helenus/core/cache/EntityIdentifyingFacet.java +++ b/src/main/java/net/helenus/core/cache/EntityIdentifyingFacet.java @@ -4,11 +4,7 @@ import net.helenus.mapping.HelenusProperty; public class EntityIdentifyingFacet extends Facet { - public EntityIdentifyingFacet(HelenusProperty prop) { + public EntityIdentifyingFacet(HelenusProperty prop) {} - } - - public EntityIdentifyingFacet(HelenusProperty[]... props) { - - } + public EntityIdentifyingFacet(HelenusProperty[]... props) {} } diff --git a/src/main/java/net/helenus/core/cache/Facet.java b/src/main/java/net/helenus/core/cache/Facet.java index eb521b1..31fdb2c 100644 --- a/src/main/java/net/helenus/core/cache/Facet.java +++ b/src/main/java/net/helenus/core/cache/Facet.java @@ -1,7 +1,6 @@ package net.helenus.core.cache; -public class Facet { -} +public class Facet {} /* An Entity is identifiable via one or more Facets diff --git a/src/main/java/net/helenus/core/operation/AbstractFilterOptionalOperation.java b/src/main/java/net/helenus/core/operation/AbstractFilterOptionalOperation.java index 8ed14af..6abfd81 100644 --- a/src/main/java/net/helenus/core/operation/AbstractFilterOptionalOperation.java +++ b/src/main/java/net/helenus/core/operation/AbstractFilterOptionalOperation.java @@ -19,11 +19,11 @@ import java.util.LinkedHashMap; import java.util.LinkedList; import java.util.List; import java.util.Map; - import net.helenus.core.*; import net.helenus.mapping.HelenusProperty; -public abstract class AbstractFilterOptionalOperation> +public abstract class AbstractFilterOptionalOperation< + E, O extends AbstractFilterOptionalOperation> extends AbstractOptionalOperation { protected Map> filters = null; diff --git a/src/main/java/net/helenus/core/operation/AbstractFilterStreamOperation.java b/src/main/java/net/helenus/core/operation/AbstractFilterStreamOperation.java index 755e220..b78daf1 100644 --- a/src/main/java/net/helenus/core/operation/AbstractFilterStreamOperation.java +++ b/src/main/java/net/helenus/core/operation/AbstractFilterStreamOperation.java @@ -19,11 +19,11 @@ import java.util.LinkedHashMap; import java.util.LinkedList; import java.util.List; import java.util.Map; - import net.helenus.core.*; import net.helenus.mapping.HelenusProperty; -public abstract class AbstractFilterStreamOperation> +public abstract class AbstractFilterStreamOperation< + E, O extends AbstractFilterStreamOperation> extends AbstractStreamOperation { protected Map> filters = null; diff --git a/src/main/java/net/helenus/core/operation/AbstractOperation.java b/src/main/java/net/helenus/core/operation/AbstractOperation.java index 46739b9..8eb7961 100644 --- a/src/main/java/net/helenus/core/operation/AbstractOperation.java +++ b/src/main/java/net/helenus/core/operation/AbstractOperation.java @@ -17,8 +17,6 @@ package net.helenus.core.operation; import com.codahale.metrics.Timer; import com.datastax.driver.core.ResultSet; - -import java.util.Objects; import java.util.concurrent.CompletableFuture; import net.helenus.core.AbstractSessionOperations; import net.helenus.core.UnitOfWork; @@ -40,7 +38,6 @@ public abstract class AbstractOperation> return new PreparedOperation(prepareStatement(), this); } - public E sync() { final Timer.Context context = requestLatency.time(); try { @@ -72,5 +69,4 @@ public abstract class AbstractOperation> if (uow == null) return async(); return CompletableFuture.supplyAsync(() -> sync(uow)); } - } diff --git a/src/main/java/net/helenus/core/operation/AbstractStatementOperation.java b/src/main/java/net/helenus/core/operation/AbstractStatementOperation.java index 6a287f0..5177724 100644 --- a/src/main/java/net/helenus/core/operation/AbstractStatementOperation.java +++ b/src/main/java/net/helenus/core/operation/AbstractStatementOperation.java @@ -32,7 +32,8 @@ import net.helenus.support.HelenusException; import org.slf4j.Logger; import org.slf4j.LoggerFactory; -public abstract class AbstractStatementOperation> extends Operation { +public abstract class AbstractStatementOperation> + extends Operation { final Logger logger = LoggerFactory.getLogger(getClass()); @@ -55,7 +56,6 @@ public abstract class AbstractStatementOperation> extends AbstractStatementOperation { @@ -113,5 +104,4 @@ public abstract class AbstractStreamOperation>supplyAsync(() -> sync(uow)); } - } diff --git a/src/main/java/net/helenus/core/operation/BoundOptionalOperation.java b/src/main/java/net/helenus/core/operation/BoundOptionalOperation.java index 4230bab..c3f5332 100644 --- a/src/main/java/net/helenus/core/operation/BoundOptionalOperation.java +++ b/src/main/java/net/helenus/core/operation/BoundOptionalOperation.java @@ -34,7 +34,9 @@ public final class BoundOptionalOperation } @Override - public Optional transform(ResultSet resultSet) { return delegate.transform(resultSet); } + public Optional transform(ResultSet resultSet) { + return delegate.transform(resultSet); + } @Override public Statement buildStatement(boolean cached) { diff --git a/src/main/java/net/helenus/core/operation/BoundStreamOperation.java b/src/main/java/net/helenus/core/operation/BoundStreamOperation.java index 4210b04..7a3cf2c 100644 --- a/src/main/java/net/helenus/core/operation/BoundStreamOperation.java +++ b/src/main/java/net/helenus/core/operation/BoundStreamOperation.java @@ -26,14 +26,17 @@ public final class BoundStreamOperation private final BoundStatement boundStatement; private final AbstractStreamOperation delegate; - public BoundStreamOperation(BoundStatement boundStatement, AbstractStreamOperation operation) { + public BoundStreamOperation( + BoundStatement boundStatement, AbstractStreamOperation operation) { super(operation.sessionOps); this.boundStatement = boundStatement; this.delegate = operation; } @Override - public String getStatementCacheKey() { return delegate.getStatementCacheKey(); } + public String getStatementCacheKey() { + return delegate.getStatementCacheKey(); + } @Override public Stream transform(ResultSet resultSet) { @@ -41,5 +44,7 @@ public final class BoundStreamOperation } @Override - public Statement buildStatement(boolean cached) { return boundStatement; } + public Statement buildStatement(boolean cached) { + return boundStatement; + } } diff --git a/src/main/java/net/helenus/core/operation/InsertOperation.java b/src/main/java/net/helenus/core/operation/InsertOperation.java index f05bf87..3ee0c8a 100644 --- a/src/main/java/net/helenus/core/operation/InsertOperation.java +++ b/src/main/java/net/helenus/core/operation/InsertOperation.java @@ -19,10 +19,9 @@ import com.datastax.driver.core.ResultSet; import com.datastax.driver.core.querybuilder.BuiltStatement; import com.datastax.driver.core.querybuilder.Insert; import com.datastax.driver.core.querybuilder.QueryBuilder; +import com.google.common.base.Joiner; import java.util.*; import java.util.function.Function; - -import com.google.common.base.Joiner; import net.helenus.core.AbstractSessionOperations; import net.helenus.core.Getter; import net.helenus.core.Helenus; @@ -41,7 +40,8 @@ public final class InsertOperation extends AbstractOperation> values = new ArrayList>(); + private final List> values = + new ArrayList>(); private final T pojo; private final Class resultType; private boolean ifNotExists; @@ -57,7 +57,8 @@ public final class InsertOperation extends AbstractOperation resultType, boolean ifNotExists) { + public InsertOperation( + AbstractSessionOperations sessionOperations, Class resultType, boolean ifNotExists) { super(sessionOperations); this.ifNotExists = ifNotExists; @@ -172,7 +173,7 @@ public final class InsertOperation extends AbstractOperation> converter = - prop.getReadConverter(sessionOps.getSessionRepository()); + prop.getReadConverter(sessionOps.getSessionRepository()); if (converter.isPresent()) { backingMap.put(key, converter.get().apply(backingMap.get(key))); } @@ -232,23 +233,25 @@ public final class InsertOperation extends AbstractOperation keys = new ArrayList<>(values.size()); values.forEach( - t -> { - HelenusPropertyNode prop = t._1; - switch (prop.getProperty().getColumnType()) { - case PARTITION_KEY: - case CLUSTERING_COLUMN: - keys.add(prop.getColumnName() + "==" + t._2.toString()); - break; - default: - break; - } - }); + t -> { + HelenusPropertyNode prop = t._1; + switch (prop.getProperty().getColumnType()) { + case PARTITION_KEY: + case CLUSTERING_COLUMN: + keys.add(prop.getColumnName() + "==" + t._2.toString()); + break; + default: + break; + } + }); return entity.getName() + ": " + Joiner.on(",").join(keys); } @Override public T sync(UnitOfWork uow) { - if (uow == null) { return sync(); } + if (uow == null) { + return sync(); + } T result = super.sync(uow); Class iface = entity.getMappingInterface(); if (resultType == iface) { @@ -261,5 +264,4 @@ public final class InsertOperation extends AbstractOperation { - protected final AbstractSessionOperations sessionOps; - protected final Meter uowCacheHits; - protected final Meter uowCacheMiss; - protected final Timer requestLatency; + protected final AbstractSessionOperations sessionOps; + protected final Meter uowCacheHits; + protected final Meter uowCacheMiss; + protected final Timer requestLatency; - Operation(AbstractSessionOperations sessionOperations) { - this.sessionOps = sessionOperations; - MetricRegistry metrics = sessionOperations.getMetricRegistry(); - this.uowCacheHits = metrics.meter("net.helenus.UOW-cache-hits"); - this.uowCacheMiss = metrics.meter("net.helenus.UOW-cache-miss"); - this.requestLatency = metrics.timer("net.helenus.request-latency"); + Operation(AbstractSessionOperations sessionOperations) { + this.sessionOps = sessionOperations; + MetricRegistry metrics = sessionOperations.getMetricRegistry(); + this.uowCacheHits = metrics.meter("net.helenus.UOW-cache-hits"); + this.uowCacheMiss = metrics.meter("net.helenus.UOW-cache-miss"); + this.requestLatency = metrics.timer("net.helenus.request-latency"); + } + + public ResultSet execute( + AbstractSessionOperations session, + UnitOfWork uow, + TraceContext traceContext, + boolean showValues, + boolean cached) { + + // Start recording in a Zipkin sub-span our execution time to perform this operation. + Tracer tracer = session.getZipkinTracer(); + Span span = null; + if (tracer != null && traceContext != null) { + span = tracer.newChild(traceContext); } - public ResultSet execute(AbstractSessionOperations session, UnitOfWork uow, TraceContext traceContext, boolean showValues, boolean cached) { + try { - // Start recording in a Zipkin sub-span our execution time to perform this operation. - Tracer tracer = session.getZipkinTracer(); - Span span = null; - if (tracer != null && traceContext != null) { - span = tracer.newChild(traceContext); - } + if (span != null) { + span.name("cassandra"); + span.start(); + } - try { + Statement statement = options(buildStatement(cached)); + ResultSetFuture futureResultSet = session.executeAsync(statement, showValues); + return futureResultSet.get(); - if (span != null) { - span.name("cassandra"); - span.start(); - } + } catch (InterruptedException | ExecutionException e) { - Statement statement = options(buildStatement(cached)); - ResultSetFuture futureResultSet = session.executeAsync(statement, showValues); - return futureResultSet.get(); + throw new RuntimeException(e); - } catch (InterruptedException | ExecutionException e) { + } finally { - throw new RuntimeException(e); - - } finally { - - if (span != null) { - span.finish(); - } - - } + if (span != null) { + span.finish(); + } } + } - public Statement options(Statement statement) { return statement; } + public Statement options(Statement statement) { + return statement; + } - public Statement buildStatement(boolean cached) { return null; } - - public Set getIdentifyingFacets() { return null; } + public Statement buildStatement(boolean cached) { + return null; + } + public Set getIdentifyingFacets() { + return null; + } } diff --git a/src/main/java/net/helenus/core/operation/PreparedStreamOperation.java b/src/main/java/net/helenus/core/operation/PreparedStreamOperation.java index b622874..cd0f6be 100644 --- a/src/main/java/net/helenus/core/operation/PreparedStreamOperation.java +++ b/src/main/java/net/helenus/core/operation/PreparedStreamOperation.java @@ -17,7 +17,6 @@ package net.helenus.core.operation; import com.datastax.driver.core.BoundStatement; import com.datastax.driver.core.PreparedStatement; -import java.util.regex.Pattern; public final class PreparedStreamOperation { diff --git a/src/main/java/net/helenus/core/operation/SelectFirstOperation.java b/src/main/java/net/helenus/core/operation/SelectFirstOperation.java index ff88555..d5cf01e 100644 --- a/src/main/java/net/helenus/core/operation/SelectFirstOperation.java +++ b/src/main/java/net/helenus/core/operation/SelectFirstOperation.java @@ -38,7 +38,9 @@ public final class SelectFirstOperation } @Override - public String getStatementCacheKey() { return delegate.getStatementCacheKey(); } + public String getStatementCacheKey() { + return delegate.getStatementCacheKey(); + } @Override public BuiltStatement buildStatement(boolean cached) { diff --git a/src/main/java/net/helenus/core/operation/SelectFirstTransformingOperation.java b/src/main/java/net/helenus/core/operation/SelectFirstTransformingOperation.java index 1d95256..8ef4f60 100644 --- a/src/main/java/net/helenus/core/operation/SelectFirstTransformingOperation.java +++ b/src/main/java/net/helenus/core/operation/SelectFirstTransformingOperation.java @@ -36,7 +36,9 @@ public final class SelectFirstTransformingOperation } @Override - public String getStatementCacheKey() { return delegate.getStatementCacheKey(); } + public String getStatementCacheKey() { + return delegate.getStatementCacheKey(); + } @Override public BuiltStatement buildStatement(boolean cached) { diff --git a/src/main/java/net/helenus/core/operation/SelectOperation.java b/src/main/java/net/helenus/core/operation/SelectOperation.java index 63619b6..4204c29 100644 --- a/src/main/java/net/helenus/core/operation/SelectOperation.java +++ b/src/main/java/net/helenus/core/operation/SelectOperation.java @@ -22,15 +22,12 @@ import com.datastax.driver.core.querybuilder.QueryBuilder; import com.datastax.driver.core.querybuilder.Select; import com.datastax.driver.core.querybuilder.Select.Selection; import com.datastax.driver.core.querybuilder.Select.Where; +import com.google.common.collect.Iterables; import java.util.*; import java.util.function.Function; import java.util.stream.Stream; import java.util.stream.StreamSupport; - -import com.google.common.base.Joiner; -import com.google.common.collect.Iterables; import net.helenus.core.*; -import net.helenus.core.cache.Facet; import net.helenus.core.cache.EntityIdentifyingFacet; import net.helenus.core.reflect.HelenusPropertyNode; import net.helenus.mapping.HelenusEntity; @@ -56,7 +53,7 @@ public final class SelectOperation extends AbstractFilterStreamOperation() { + new Function() { @Override public E apply(Row source) { @@ -188,20 +185,25 @@ public final class SelectOperation extends AbstractFilterStreamOperation facets = new HashSet<>(filters.size()); // Check to see if this select statement has enough information to build one or // more identifying facets. - entity.getIdentityFacets().forEach((facetName, facet) -> { - EntityIdentifyingFacet boundFacet = null; - if (!facet.isFullyBound()) { - boundFacet = new EntityIdentifyingFacet(facet); - for (HelenusProperty prop : facet.getUnboundEntityProperties()) { - Filter filter = filters.get(facet.getProperty()); - if (filter == null) { break; } - boundFacet.setValueForProperty(prop, filter.toString()); - } - } - if (boundFacet != null && boundFacet.isFullyBound()) { - facets.add(boundFacet); - } - }); + entity + .getIdentityFacets() + .forEach( + (facetName, facet) -> { + EntityIdentifyingFacet boundFacet = null; + if (!facet.isFullyBound()) { + boundFacet = new EntityIdentifyingFacet(facet); + for (HelenusProperty prop : facet.getUnboundEntityProperties()) { + Filter filter = filters.get(facet.getProperty()); + if (filter == null) { + break; + } + boundFacet.setValueForProperty(prop, filter.toString()); + } + } + if (boundFacet != null && boundFacet.isFullyBound()) { + facets.add(boundFacet); + } + }); return facets; } @@ -223,14 +225,14 @@ public final class SelectOperation extends AbstractFilterStreamOperation extends AbstractFilterStreamOperation transform(ResultSet resultSet) { if (rowMapper != null) { - return StreamSupport.stream(Spliterators.spliteratorUnknownSize(resultSet.iterator(), Spliterator.ORDERED), false).map(rowMapper); + return StreamSupport.stream( + Spliterators.spliteratorUnknownSize(resultSet.iterator(), Spliterator.ORDERED), false) + .map(rowMapper); } else { return (Stream) - StreamSupport.stream(Spliterators.spliteratorUnknownSize(resultSet.iterator(), Spliterator.ORDERED),false); + StreamSupport.stream( + Spliterators.spliteratorUnknownSize(resultSet.iterator(), Spliterator.ORDERED), + false); } } diff --git a/src/main/java/net/helenus/core/operation/SelectTransformingOperation.java b/src/main/java/net/helenus/core/operation/SelectTransformingOperation.java index 8883f7e..f93ca41 100644 --- a/src/main/java/net/helenus/core/operation/SelectTransformingOperation.java +++ b/src/main/java/net/helenus/core/operation/SelectTransformingOperation.java @@ -36,7 +36,9 @@ public final class SelectTransformingOperation } @Override - public String getStatementCacheKey() { return delegate.getStatementCacheKey(); } + public String getStatementCacheKey() { + return delegate.getStatementCacheKey(); + } @Override public BuiltStatement buildStatement(boolean cached) { @@ -47,5 +49,4 @@ public final class SelectTransformingOperation public Stream transform(ResultSet resultSet) { return delegate.transform(resultSet).map(fn); } - } diff --git a/src/main/java/net/helenus/core/operation/UpdateOperation.java b/src/main/java/net/helenus/core/operation/UpdateOperation.java index ba358d6..b918bc6 100644 --- a/src/main/java/net/helenus/core/operation/UpdateOperation.java +++ b/src/main/java/net/helenus/core/operation/UpdateOperation.java @@ -15,15 +15,13 @@ */ package net.helenus.core.operation; -import java.util.*; -import java.util.function.Function; - import com.datastax.driver.core.ResultSet; import com.datastax.driver.core.querybuilder.Assignment; import com.datastax.driver.core.querybuilder.BuiltStatement; import com.datastax.driver.core.querybuilder.QueryBuilder; import com.datastax.driver.core.querybuilder.Update; - +import java.util.*; +import java.util.function.Function; import net.helenus.core.*; import net.helenus.core.reflect.HelenusPropertyNode; import net.helenus.mapping.HelenusEntity; @@ -43,19 +41,21 @@ public final class UpdateOperation extends AbstractFilterOperation draft) { + public UpdateOperation( + AbstractSessionOperations sessionOperations, AbstractEntityDraft draft) { super(sessionOperations); this.draft = draft; this.draftMap = draft.toMap(); } - public UpdateOperation(AbstractSessionOperations sessionOperations, HelenusPropertyNode p, Object v) { + public UpdateOperation( + AbstractSessionOperations sessionOperations, HelenusPropertyNode p, Object v) { super(sessionOperations); this.draft = null; this.draftMap = null; @@ -152,7 +152,7 @@ public final class UpdateOperation extends AbstractFilterOperation list = (List)draftMap.get(key); + List list = (List) draftMap.get(key); list.add(0, value); } @@ -194,7 +194,7 @@ public final class UpdateOperation extends AbstractFilterOperation list = (List)draftMap.get(key); + List list = (List) draftMap.get(key); if (idx < 0) { list.add(0, value); } else if (idx > list.size()) { @@ -222,7 +222,7 @@ public final class UpdateOperation extends AbstractFilterOperation list = (List)draftMap.get(key); + List list = (List) draftMap.get(key); list.add(value); } @@ -579,7 +579,9 @@ public final class UpdateOperation extends AbstractFilterOperation extends AbstractFilterOperation implements InvocationHandler { } private HelenusEntity init(Metadata metadata) { - HelenusEntity entity = new HelenusMappingEntity(iface, metadata); + HelenusEntity entity = new HelenusMappingEntity(iface, metadata); - for (HelenusProperty prop : entity.getOrderedProperties()) { + for (HelenusProperty prop : entity.getOrderedProperties()) { - map.put(prop.getGetterMethod(), prop); + map.put(prop.getGetterMethod(), prop); - AbstractDataType type = prop.getDataType(); - Class javaType = prop.getJavaType(); + AbstractDataType type = prop.getDataType(); + Class javaType = prop.getJavaType(); - if (type instanceof UDTDataType && !UDTValue.class.isAssignableFrom(javaType)) { + if (type instanceof UDTDataType && !UDTValue.class.isAssignableFrom(javaType)) { - Object childDsl = - Helenus.dsl( - javaType, - classLoader, - Optional.of(new HelenusPropertyNode(prop, parent)), - metadata); + Object childDsl = + Helenus.dsl( + javaType, + classLoader, + Optional.of(new HelenusPropertyNode(prop, parent)), + metadata); - udtMap.put(prop.getGetterMethod(), childDsl); - } - - if (type instanceof DTDataType) { - DTDataType dataType = (DTDataType) type; - - if (dataType.getDataType() instanceof TupleType - && !TupleValue.class.isAssignableFrom(javaType)) { - - Object childDsl = - Helenus.dsl( - javaType, - classLoader, - Optional.of(new HelenusPropertyNode(prop, parent)), - metadata); - - tupleMap.put(prop.getGetterMethod(), childDsl); - } - } + udtMap.put(prop.getGetterMethod(), childDsl); } - return entity; + if (type instanceof DTDataType) { + DTDataType dataType = (DTDataType) type; + + if (dataType.getDataType() instanceof TupleType + && !TupleValue.class.isAssignableFrom(javaType)) { + + Object childDsl = + Helenus.dsl( + javaType, + classLoader, + Optional.of(new HelenusPropertyNode(prop, parent)), + metadata); + + tupleMap.put(prop.getGetterMethod(), childDsl); + } + } + } + + return entity; } @Override @@ -127,10 +127,10 @@ public class DslInvocationHandler implements InvocationHandler { } if (DslExportable.SET_METADATA_METHOD.equals(methodName) - && args.length == 1 - && args[0] instanceof Metadata) { + && args.length == 1 + && args[0] instanceof Metadata) { if (metadata == null) { - this.setMetadata((Metadata) args[0]); + this.setMetadata((Metadata) args[0]); } return null; } diff --git a/src/main/java/net/helenus/core/reflect/MapperInvocationHandler.java b/src/main/java/net/helenus/core/reflect/MapperInvocationHandler.java index 27d48cf..4845c11 100644 --- a/src/main/java/net/helenus/core/reflect/MapperInvocationHandler.java +++ b/src/main/java/net/helenus/core/reflect/MapperInvocationHandler.java @@ -23,7 +23,6 @@ import java.lang.reflect.Method; import java.lang.reflect.Proxy; import java.util.Collections; import java.util.Map; - import net.helenus.core.Helenus; import net.helenus.mapping.annotation.Transient; import net.helenus.support.HelenusException; @@ -94,7 +93,7 @@ public class MapperInvocationHandler implements InvocationHandler, Serializab } if ("dsl".equals(methodName)) { - return Helenus.dsl(iface); + return Helenus.dsl(iface); } if (MapExportable.TO_MAP_METHOD.equals(methodName)) { diff --git a/src/main/java/net/helenus/mapping/HelenusEntity.java b/src/main/java/net/helenus/mapping/HelenusEntity.java index ef63078..fa84c7b 100644 --- a/src/main/java/net/helenus/mapping/HelenusEntity.java +++ b/src/main/java/net/helenus/mapping/HelenusEntity.java @@ -15,10 +15,9 @@ */ package net.helenus.mapping; -import net.helenus.core.cache.EntityIdentifyingFacet; - import java.util.Collection; import java.util.Map; +import net.helenus.core.cache.EntityIdentifyingFacet; public interface HelenusEntity { @@ -35,5 +34,4 @@ public interface HelenusEntity { HelenusProperty getProperty(String name); Map getIdentityFacets(); - } diff --git a/src/main/java/net/helenus/mapping/HelenusMappingEntity.java b/src/main/java/net/helenus/mapping/HelenusMappingEntity.java index eaee3fe..b7fd8c3 100644 --- a/src/main/java/net/helenus/mapping/HelenusMappingEntity.java +++ b/src/main/java/net/helenus/mapping/HelenusMappingEntity.java @@ -20,7 +20,6 @@ import com.google.common.collect.ImmutableList; import com.google.common.collect.ImmutableMap; import java.lang.reflect.Method; import java.util.*; - import net.helenus.config.HelenusSettings; import net.helenus.core.Helenus; import net.helenus.core.annotation.Cacheable; @@ -60,21 +59,22 @@ public final class HelenusMappingEntity implements HelenusEntity { Map methods = new HashMap(); for (Method m : iface.getDeclaredMethods()) { - methods.put(m.getName(), m); + methods.put(m.getName(), m); } for (Class c : ClassUtils.getAllInterfaces(iface)) { - if (c.getDeclaredAnnotation(Table.class) != null || c.getDeclaredAnnotation(InheritedTable.class) != null) { + if (c.getDeclaredAnnotation(Table.class) != null + || c.getDeclaredAnnotation(InheritedTable.class) != null) { for (Method m : c.getDeclaredMethods()) { - Method o = methods.get(m.getName()); - if (o != null) { - // Prefer overridden method implementation. - if (o.getDeclaringClass().isAssignableFrom(m.getDeclaringClass())) { - methods.put(m.getName(), m); - } - } else { - methods.put(m.getName(), m); + Method o = methods.get(m.getName()); + if (o != null) { + // Prefer overridden method implementation. + if (o.getDeclaringClass().isAssignableFrom(m.getDeclaringClass())) { + methods.put(m.getName(), m); } + } else { + methods.put(m.getName(), m); + } } } } @@ -110,27 +110,34 @@ public final class HelenusMappingEntity implements HelenusEntity { cacheable = (null != iface.getDeclaredAnnotation(Cacheable.class)); ImmutableMap.Builder allFacetsBuilder = ImmutableMap.builder(); - ImmutableMap.Builder ancillaryFacetsBuilder = ImmutableMap.builder(); + ImmutableMap.Builder ancillaryFacetsBuilder = + ImmutableMap.builder(); EntityIdentifyingFacet primaryFacet = null; List primaryProperties = new ArrayList(4); for (HelenusProperty prop : propsLocal) { - switch(prop.getColumnType()) { + switch (prop.getColumnType()) { case PARTITION_KEY: case CLUSTERING_COLUMN: - primaryProperties.add(prop); - break; + primaryProperties.add(prop); + break; default: - if (primaryProperties != null) { - primaryFacet = new EntityIdentifyingFacet(keyspace, table, schemaVersion, primaryProperties.toArray(new HelenusProperty[props.size()])); - allFacetsBuilder.put("*", primaryFacet); - primaryProperties = null; - } - Optional optionalIndexName = prop.getIndexName(); - if (optionalIndexName.isPresent()) { - EntityIdentifyingFacet facet = new EntityIdentifyingFacet(keyspace, table, schemaVersion, prop); - ancillaryFacetsBuilder.put(prop.getPropertyName(), facet); - } - } + if (primaryProperties != null) { + primaryFacet = + new EntityIdentifyingFacet( + keyspace, + table, + schemaVersion, + primaryProperties.toArray(new HelenusProperty[props.size()])); + allFacetsBuilder.put("*", primaryFacet); + primaryProperties = null; + } + Optional optionalIndexName = prop.getIndexName(); + if (optionalIndexName.isPresent()) { + EntityIdentifyingFacet facet = + new EntityIdentifyingFacet(keyspace, table, schemaVersion, prop); + ancillaryFacetsBuilder.put(prop.getPropertyName(), facet); + } + } } this.primaryIdentityFacet = primaryFacet; this.ancillaryIdentityFacets = ancillaryFacetsBuilder.build(); diff --git a/src/main/java/net/helenus/mapping/javatype/AbstractCollectionJavaType.java b/src/main/java/net/helenus/mapping/javatype/AbstractCollectionJavaType.java index 61a764a..a882398 100644 --- a/src/main/java/net/helenus/mapping/javatype/AbstractCollectionJavaType.java +++ b/src/main/java/net/helenus/mapping/javatype/AbstractCollectionJavaType.java @@ -2,6 +2,7 @@ package net.helenus.mapping.javatype; public abstract class AbstractCollectionJavaType extends AbstractJavaType { - public static boolean isCollectionType() { return true; } - + public static boolean isCollectionType() { + return true; + } } diff --git a/src/main/java/net/helenus/mapping/javatype/AbstractJavaType.java b/src/main/java/net/helenus/mapping/javatype/AbstractJavaType.java index 806fc2e..abd7e21 100644 --- a/src/main/java/net/helenus/mapping/javatype/AbstractJavaType.java +++ b/src/main/java/net/helenus/mapping/javatype/AbstractJavaType.java @@ -33,7 +33,9 @@ import net.helenus.support.HelenusMappingException; public abstract class AbstractJavaType { - public static boolean isCollectionType() { return false; } + public static boolean isCollectionType() { + return false; + } public abstract Class getJavaClass(); diff --git a/src/main/java/net/helenus/mapping/javatype/MapJavaType.java b/src/main/java/net/helenus/mapping/javatype/MapJavaType.java index 4d6e38f..2a5d187 100644 --- a/src/main/java/net/helenus/mapping/javatype/MapJavaType.java +++ b/src/main/java/net/helenus/mapping/javatype/MapJavaType.java @@ -18,7 +18,6 @@ package net.helenus.mapping.javatype; import com.datastax.driver.core.*; import java.lang.reflect.Method; import java.lang.reflect.Type; -import java.util.AbstractCollection; import java.util.Map; import java.util.Optional; import java.util.function.Function; diff --git a/src/main/java/net/helenus/mapping/type/AbstractCollectionDataType.java b/src/main/java/net/helenus/mapping/type/AbstractCollectionDataType.java index d138263..7e76d92 100644 --- a/src/main/java/net/helenus/mapping/type/AbstractCollectionDataType.java +++ b/src/main/java/net/helenus/mapping/type/AbstractCollectionDataType.java @@ -4,10 +4,11 @@ import net.helenus.mapping.ColumnType; public abstract class AbstractCollectionDataType extends AbstractDataType { - public AbstractCollectionDataType(ColumnType columnType) { - super(columnType); - } - - public boolean isCollectionType() { return true; } + public AbstractCollectionDataType(ColumnType columnType) { + super(columnType); + } + public boolean isCollectionType() { + return true; + } } diff --git a/src/main/java/net/helenus/mapping/type/AbstractDataType.java b/src/main/java/net/helenus/mapping/type/AbstractDataType.java index 8be0fd1..cc65d68 100644 --- a/src/main/java/net/helenus/mapping/type/AbstractDataType.java +++ b/src/main/java/net/helenus/mapping/type/AbstractDataType.java @@ -55,6 +55,7 @@ public abstract class AbstractDataType { "wrong column type " + columnType + " for UserDefinedType in columnName " + columnName); } - public boolean isCollectionType() { return false; } - + public boolean isCollectionType() { + return false; + } } diff --git a/src/main/java/net/helenus/mapping/type/UDTKeyMapDataType.java b/src/main/java/net/helenus/mapping/type/UDTKeyMapDataType.java index b9724aa..e04685c 100644 --- a/src/main/java/net/helenus/mapping/type/UDTKeyMapDataType.java +++ b/src/main/java/net/helenus/mapping/type/UDTKeyMapDataType.java @@ -18,8 +18,6 @@ package net.helenus.mapping.type; import com.datastax.driver.core.DataType; import com.datastax.driver.core.UserType; import com.datastax.driver.core.schemabuilder.*; - -import java.util.AbstractCollection; import java.util.List; import net.helenus.mapping.ColumnType; import net.helenus.mapping.IdentityName; diff --git a/src/main/java/net/helenus/mapping/value/BeanColumnValueProvider.java b/src/main/java/net/helenus/mapping/value/BeanColumnValueProvider.java index 2381514..a87fd15 100644 --- a/src/main/java/net/helenus/mapping/value/BeanColumnValueProvider.java +++ b/src/main/java/net/helenus/mapping/value/BeanColumnValueProvider.java @@ -33,7 +33,7 @@ public enum BeanColumnValueProvider implements ColumnValueProvider { try { value = getter.invoke(bean, new Object[] {}); } catch (InvocationTargetException e) { - if (e.getCause() != null ) { + if (e.getCause() != null) { throw new HelenusException("getter threw an exception", e.getCause()); } } catch (ReflectiveOperationException e) { diff --git a/src/test/java/net/helenus/test/integration/core/collection/CollectionTest.java b/src/test/java/net/helenus/test/integration/core/collection/CollectionTest.java index 5b437eb..e838ad3 100644 --- a/src/test/java/net/helenus/test/integration/core/collection/CollectionTest.java +++ b/src/test/java/net/helenus/test/integration/core/collection/CollectionTest.java @@ -67,11 +67,8 @@ public class CollectionTest extends AbstractEmbeddedCassandraTest { // read full object - Customer actual = session.select(customer) - .where(customer::id, eq(id)) - .single() - .sync() - .orElse(null); + Customer actual = + session.select(customer).where(customer::id, eq(id)).single().sync().orElse(null); Assert.assertEquals(id, actual.id()); Assert.assertEquals(aliases, actual.aliases()); Assert.assertNull(actual.names()); @@ -90,11 +87,8 @@ public class CollectionTest extends AbstractEmbeddedCassandraTest { session.update().set(customer::aliases, expected).where(customer::id, eq(id)).sync(); - actual = session.select(customer) - .where(customer::id, eq(id)) - .single() - .sync() - .orElse(null); + actual = + session.select(customer).where(customer::id, eq(id)).single().sync().orElse(null); Assert.assertEquals(id, actual.id()); Assert.assertEquals(expected, actual.aliases()); @@ -170,11 +164,8 @@ public class CollectionTest extends AbstractEmbeddedCassandraTest { // read full object - Customer actual = session.select(customer) - .where(customer::id, eq(id)) - .single() - .sync() - .orElse(null); + Customer actual = + session.select(customer).where(customer::id, eq(id)).single().sync().orElse(null); Assert.assertEquals(id, actual.id()); Assert.assertEquals(names, actual.names()); @@ -200,11 +191,8 @@ public class CollectionTest extends AbstractEmbeddedCassandraTest { session.update().set(customer::names, expected).where(customer::id, eq(id)).sync(); - actual = session.select(customer) - .where(customer::id, eq(id)) - .single() - .sync() - .orElse(null); + actual = + session.select(customer).where(customer::id, eq(id)).single().sync().orElse(null); Assert.assertEquals(id, actual.id()); Assert.assertEquals(expected, actual.names()); @@ -306,10 +294,8 @@ public class CollectionTest extends AbstractEmbeddedCassandraTest { // read full object - Customer actual = session.select(customer) - .where(customer::id, eq(id)).single() - .sync() - .orElse(null); + Customer actual = + session.select(customer).where(customer::id, eq(id)).single().sync().orElse(null); Assert.assertEquals(id, actual.id()); Assert.assertEquals(props, actual.properties()); @@ -343,11 +329,8 @@ public class CollectionTest extends AbstractEmbeddedCassandraTest { session.update().set(customer::properties, expected).where(customer::id, eq(id)).sync(); - actual = session.select(customer) - .where(customer::id, eq(id)) - .single() - .sync() - .orElse(null); + actual = + session.select(customer).where(customer::id, eq(id)).single().sync().orElse(null); Assert.assertEquals(id, actual.id()); Assert.assertEquals(expected, actual.properties()); diff --git a/src/test/java/net/helenus/test/integration/core/draft/EntityDraftBuilderTest.java b/src/test/java/net/helenus/test/integration/core/draft/EntityDraftBuilderTest.java index 2be5381..3296488 100644 --- a/src/test/java/net/helenus/test/integration/core/draft/EntityDraftBuilderTest.java +++ b/src/test/java/net/helenus/test/integration/core/draft/EntityDraftBuilderTest.java @@ -18,76 +18,73 @@ package net.helenus.test.integration.core.draft; import java.util.ArrayList; import java.util.HashMap; import java.util.HashSet; - +import net.helenus.core.Helenus; +import net.helenus.core.HelenusSession; +import net.helenus.test.integration.build.AbstractEmbeddedCassandraTest; import org.junit.Assert; import org.junit.BeforeClass; import org.junit.Test; -import net.helenus.core.Helenus; -import net.helenus.core.HelenusSession; -import net.helenus.test.integration.build.AbstractEmbeddedCassandraTest; - -import static net.helenus.core.Query.eq; public class EntityDraftBuilderTest extends AbstractEmbeddedCassandraTest { + static Supply supply; + static HelenusSession session; - static Supply supply; - static HelenusSession session; + @BeforeClass + public static void beforeTest() { + session = Helenus.init(getSession()).showCql().add(Supply.class).autoCreateDrop().get(); + supply = session.dsl(Supply.class); + } + @Test + public void testFoo() throws Exception { + Supply.Draft draft = null; - @BeforeClass - public static void beforeTest() { - session = Helenus.init(getSession()) - .showCql() - .add(Supply.class) - .autoCreateDrop() - .get(); - supply = session.dsl(Supply.class); - } - - @Test - public void testFoo() throws Exception { - Supply.Draft draft = null; - - draft = Supply.draft("APAC") - .code("WIDGET-002") - .description("Our second Widget!") - .demand(new HashMap() {{ + draft = + Supply.draft("APAC") + .code("WIDGET-002") + .description("Our second Widget!") + .demand( + new HashMap() { + { put("APAC", 100L); put("EMEA", 10000L); put("NORAM", 2000000L); - }}) - .shipments(new HashSet() {{ + } + }) + .shipments( + new HashSet() { + { add("HMS Puddle in transit to APAC, 100 units."); add("Frigate Jimmy in transit to EMEA, 10000 units."); - }}) - .suppliers(new ArrayList() {{ + } + }) + .suppliers( + new ArrayList() { + { add("Puddle, Inc."); add("Jimmy Town, LTD."); - }}); + } + }); - Supply s1 = session.insert(draft) - .sync(); + Supply s1 = session.insert(draft).sync(); - // List - Supply s2 = session.update(s1.update()) - .prepend(supply::suppliers, "Pignose Supply, LLC.") - .sync(); - Assert.assertEquals(s2.suppliers().get(0), "Pignose Supply, LLC."); + // List + Supply s2 = + session + .update(s1.update()) + .prepend(supply::suppliers, "Pignose Supply, LLC.") + .sync(); + Assert.assertEquals(s2.suppliers().get(0), "Pignose Supply, LLC."); - // Set - String shipment = "Pignose, on the way! (1M units)"; - Supply s3 = session.update(s2.update()) - .add(supply::shipments, shipment) - .sync(); - Assert.assertTrue(s3.shipments().contains(shipment)); + // Set + String shipment = "Pignose, on the way! (1M units)"; + Supply s3 = session.update(s2.update()).add(supply::shipments, shipment).sync(); + Assert.assertTrue(s3.shipments().contains(shipment)); - // Map - Supply s4 = session.update(s3.update()) - .put(supply::demand, "NORAM", 10L) - .sync(); - Assert.assertEquals((long)s4.demand().get("NORAM"), 10L); - - } + // Map + Supply s4 = session.update(s3.update()).put(supply::demand, "NORAM", 10L).sync(); + Assert.assertEquals((long) s4.demand().get("NORAM"), 10L); + } } diff --git a/src/test/java/net/helenus/test/integration/core/draft/Inventory.java b/src/test/java/net/helenus/test/integration/core/draft/Inventory.java index 52448d2..a8b5945 100644 --- a/src/test/java/net/helenus/test/integration/core/draft/Inventory.java +++ b/src/test/java/net/helenus/test/integration/core/draft/Inventory.java @@ -1,78 +1,93 @@ package net.helenus.test.integration.core.draft; import java.util.UUID; - import net.helenus.core.AbstractAuditedEntityDraft; import net.helenus.core.Helenus; import net.helenus.core.reflect.MapExportable; import net.helenus.mapping.annotation.*; - @Table public interface Inventory { - static Inventory inventory = Helenus.dsl(Inventory.class); + static Inventory inventory = Helenus.dsl(Inventory.class); - @PartitionKey UUID id(); - @Column("emea") @Types.Counter long EMEA(); - @Column("noram") @Types.Counter long NORAM(); - @Column("apac") @Types.Counter long APAC(); + @PartitionKey + UUID id(); - @Transient static Draft draft(UUID id) { return new Draft(id); } + @Column("emea") + @Types.Counter + long EMEA(); - @Transient default Draft update() { return new Draft(this); } + @Column("noram") + @Types.Counter + long NORAM(); + @Column("apac") + @Types.Counter + long APAC(); - class Draft extends AbstractAuditedEntityDraft { + @Transient + static Draft draft(UUID id) { + return new Draft(id); + } - // Entity/Draft pattern-enabling methods: - Draft(UUID id) { - super(null); + @Transient + default Draft update() { + return new Draft(this); + } - // Primary Key: - set(inventory::id, id); - } + class Draft extends AbstractAuditedEntityDraft { - Draft(Inventory inventory) { - super((MapExportable) inventory); - } - - public Class getEntityClass() { return Inventory.class; } - - protected String getCurrentAuditor() { return "unknown"; } - - // Immutable properties: - public UUID id() { - return this.get(inventory::id, UUID.class); - } - - public long EMEA() { - return this.get(inventory::EMEA, long.class); - } - - public Draft EMEA(long count) { - mutate(inventory::EMEA, count); - return this; - } - - public long APAC() { - return this.get(inventory::APAC, long.class); - } - - public Draft APAC(long count) { - mutate(inventory::APAC, count); - return this; - } - - public long NORAM() { - return this.get(inventory::NORAM, long.class); - } - - public Draft NORAM(long count) { - mutate(inventory::NORAM, count); - return this; - } + // Entity/Draft pattern-enabling methods: + Draft(UUID id) { + super(null); + // Primary Key: + set(inventory::id, id); } + Draft(Inventory inventory) { + super((MapExportable) inventory); + } + + public Class getEntityClass() { + return Inventory.class; + } + + protected String getCurrentAuditor() { + return "unknown"; + } + + // Immutable properties: + public UUID id() { + return this.get(inventory::id, UUID.class); + } + + public long EMEA() { + return this.get(inventory::EMEA, long.class); + } + + public Draft EMEA(long count) { + mutate(inventory::EMEA, count); + return this; + } + + public long APAC() { + return this.get(inventory::APAC, long.class); + } + + public Draft APAC(long count) { + mutate(inventory::APAC, count); + return this; + } + + public long NORAM() { + return this.get(inventory::NORAM, long.class); + } + + public Draft NORAM(long count) { + mutate(inventory::NORAM, count); + return this; + } + } } diff --git a/src/test/java/net/helenus/test/integration/core/draft/Supply.java b/src/test/java/net/helenus/test/integration/core/draft/Supply.java index 91f1689..72c0ba7 100644 --- a/src/test/java/net/helenus/test/integration/core/draft/Supply.java +++ b/src/test/java/net/helenus/test/integration/core/draft/Supply.java @@ -1,129 +1,145 @@ package net.helenus.test.integration.core.draft; -import java.lang.reflect.Proxy; +import com.datastax.driver.core.utils.UUIDs; import java.util.List; import java.util.Map; import java.util.Set; import java.util.UUID; - -import com.datastax.driver.core.utils.UUIDs; - import net.helenus.core.AbstractEntityDraft; import net.helenus.core.Helenus; import net.helenus.core.reflect.MapExportable; import net.helenus.mapping.annotation.*; - @Table public interface Supply { - static Supply supply = Helenus.dsl(Supply.class); + static Supply supply = Helenus.dsl(Supply.class); - @PartitionKey UUID id(); - @ClusteringColumn(ordinal=0) default String region() { return "NORAM"; } + @PartitionKey + UUID id(); - @Index(caseSensitive = false) String code(); - @Index String description(); // @IndexText == lucene index - @Index Map demand(); - @Index List suppliers(); - @Index Set shipments(); + @ClusteringColumn(ordinal = 0) + default String region() { + return "NORAM"; + } - @Transient static Draft draft(String region) { return new Draft(region); } + @Index(caseSensitive = false) + String code(); - @Transient default Draft update() { return new Draft(this); } + @Index + String description(); // @IndexText == lucene index + @Index + Map demand(); - class Draft extends AbstractEntityDraft { + @Index + List suppliers(); - // Entity/Draft pattern-enabling methods: - Draft(String region) { - super(null); + @Index + Set shipments(); - // Primary Key: - set(supply::id, UUIDs.timeBased()); - set(supply::region, region); - } + @Transient + static Draft draft(String region) { + return new Draft(region); + } - Draft(Supply supply) { - super((MapExportable) supply); - } + @Transient + default Draft update() { + return new Draft(this); + } - public Class getEntityClass() { return Supply.class; } + class Draft extends AbstractEntityDraft { - // Immutable properties: - public UUID id() { - return this.get(supply::id, UUID.class); - } - - public String region() { - return this.get(supply::region, String.class); - } - - // Mutable properties: - public String code() { - return this.get(supply::code, String.class); - } - - public Draft code(String code) { - mutate(supply::code, code); - return this; - } - - public Draft setCode(String code) { - return code(code); - } - - public String description() { - return this.get(supply::description, String.class); - } - - public Draft description(String description) { - mutate(supply::description, description); - return this; - } - - public Draft setDescription(String description) { - return description(description); - } - - public Map demand() { - return this.>get(supply::demand, Map.class); - } - - public Draft demand(Map demand) { - mutate(supply::demand, demand); - return this; - } - - public Draft setDemand(Map demand) { - return demand(demand); - } - - public List suppliers() { - return this.>get(supply::suppliers, List.class); - } - - public Draft suppliers(List suppliers) { - mutate(supply::suppliers, suppliers); - return this; - } - - public Draft setSuppliers(List suppliers) { - return suppliers(suppliers); - } - - public Set shipments() { - return this.>get(supply::shipments, Set.class); - } - - public Draft shipments(Set shipments) { - mutate(supply::shipments, shipments); - return this; - } - - public Draft setshipments(Set shipments) { - return shipments(shipments); - } + // Entity/Draft pattern-enabling methods: + Draft(String region) { + super(null); + // Primary Key: + set(supply::id, UUIDs.timeBased()); + set(supply::region, region); } + + Draft(Supply supply) { + super((MapExportable) supply); + } + + public Class getEntityClass() { + return Supply.class; + } + + // Immutable properties: + public UUID id() { + return this.get(supply::id, UUID.class); + } + + public String region() { + return this.get(supply::region, String.class); + } + + // Mutable properties: + public String code() { + return this.get(supply::code, String.class); + } + + public Draft code(String code) { + mutate(supply::code, code); + return this; + } + + public Draft setCode(String code) { + return code(code); + } + + public String description() { + return this.get(supply::description, String.class); + } + + public Draft description(String description) { + mutate(supply::description, description); + return this; + } + + public Draft setDescription(String description) { + return description(description); + } + + public Map demand() { + return this.>get(supply::demand, Map.class); + } + + public Draft demand(Map demand) { + mutate(supply::demand, demand); + return this; + } + + public Draft setDemand(Map demand) { + return demand(demand); + } + + public List suppliers() { + return this.>get(supply::suppliers, List.class); + } + + public Draft suppliers(List suppliers) { + mutate(supply::suppliers, suppliers); + return this; + } + + public Draft setSuppliers(List suppliers) { + return suppliers(suppliers); + } + + public Set shipments() { + return this.>get(supply::shipments, Set.class); + } + + public Draft shipments(Set shipments) { + mutate(supply::shipments, shipments); + return this; + } + + public Draft setshipments(Set shipments) { + return shipments(shipments); + } + } } diff --git a/src/test/java/net/helenus/test/integration/core/hierarchy/HierarchyTest.java b/src/test/java/net/helenus/test/integration/core/hierarchy/HierarchyTest.java index 0aaeceb..dce32b3 100644 --- a/src/test/java/net/helenus/test/integration/core/hierarchy/HierarchyTest.java +++ b/src/test/java/net/helenus/test/integration/core/hierarchy/HierarchyTest.java @@ -65,7 +65,8 @@ public class HierarchyTest extends AbstractEmbeddedCassandraTest { .value(cat::nickname, "garfield") .value(cat::eatable, false) .sync(); - Optional animal = session.select(Cat.class).where(cat::nickname, eq("garfield")).single().sync(); + Optional animal = + session.select(Cat.class).where(cat::nickname, eq("garfield")).single().sync(); Assert.assertTrue(animal.isPresent()); Cat cat = animal.get(); diff --git a/src/test/java/net/helenus/test/integration/core/hierarchy/Mammal.java b/src/test/java/net/helenus/test/integration/core/hierarchy/Mammal.java index 203ea66..950287a 100644 --- a/src/test/java/net/helenus/test/integration/core/hierarchy/Mammal.java +++ b/src/test/java/net/helenus/test/integration/core/hierarchy/Mammal.java @@ -20,6 +20,7 @@ import net.helenus.mapping.annotation.InheritedTable; @InheritedTable public interface Mammal extends Animal { - default boolean warmBlodded() { return true; } - + default boolean warmBlodded() { + return true; + } } diff --git a/src/test/java/net/helenus/test/integration/core/simple/SimpleUserTest.java b/src/test/java/net/helenus/test/integration/core/simple/SimpleUserTest.java index 9773ad1..da7997d 100644 --- a/src/test/java/net/helenus/test/integration/core/simple/SimpleUserTest.java +++ b/src/test/java/net/helenus/test/integration/core/simple/SimpleUserTest.java @@ -18,21 +18,17 @@ package net.helenus.test.integration.core.simple; import static net.helenus.core.Query.eq; import com.datastax.driver.core.ResultSet; +import java.util.*; import net.helenus.core.Helenus; import net.helenus.core.HelenusSession; import net.helenus.core.Operator; import net.helenus.core.operation.UpdateOperation; -import net.helenus.core.reflect.Drafted; -import net.helenus.mapping.HelenusEntity; import net.helenus.support.Fun; import net.helenus.test.integration.build.AbstractEmbeddedCassandraTest; import org.junit.Assert; import org.junit.BeforeClass; import org.junit.Test; -import java.util.*; - - public class SimpleUserTest extends AbstractEmbeddedCassandraTest { static User user; @@ -102,11 +98,7 @@ public class SimpleUserTest extends AbstractEmbeddedCassandraTest { // select as object - actual = session.select(user) - .where(user::id, eq(100L)) - .single() - .sync() - .orElse(null); + actual = session.select(user).where(user::id, eq(100L)).single().sync().orElse(null); assertUsers(newUser, actual); // select by columns @@ -192,11 +184,7 @@ public class SimpleUserTest extends AbstractEmbeddedCassandraTest { Assert.assertEquals("_albert", name); - User u2 = session.select(user) - .where(user::id, eq(100L)) - .single() - .sync() - .orElse(null); + User u2 = session.select(user).where(user::id, eq(100L)).single().sync().orElse(null); Assert.assertEquals(Long.valueOf(100L), u2.id()); Assert.assertEquals("albert", u2.name()); @@ -204,31 +192,27 @@ public class SimpleUserTest extends AbstractEmbeddedCassandraTest { // User greg = - session - .insert(user) - .value(user::name, "greg") - .value(user::age, 44) - .value(user::type, UserType.USER) - .value(user::id, 1234L) - .sync(); + session + .insert(user) + .value(user::name, "greg") + .value(user::age, 44) + .value(user::type, UserType.USER) + .value(user::id, 1234L) + .sync(); Optional maybeGreg = - session - .select(user) - .where(user::id, eq(1234L)) - .single() - .sync(); + session.select(user).where(user::id, eq(1234L)).single().sync(); // INSERT session - .update() - .set(user::name, null) - .set(user::age, null) - .set(user::type, null) - .where(user::id, eq(100L)) - .zipkinContext(null) - .sync(); + .update() + .set(user::name, null) + .set(user::age, null) + .set(user::type, null) + .where(user::id, eq(100L)) + .zipkinContext(null) + .sync(); Fun.Tuple3 tuple = session @@ -252,21 +236,16 @@ public class SimpleUserTest extends AbstractEmbeddedCassandraTest { public void testZipkin() throws Exception { session - .update() - .set(user::name, null) - .set(user::age, null) - .set(user::type, null) - .where(user::id, eq(100L)) - .zipkinContext(null) - .sync(); - + .update() + .set(user::name, null) + .set(user::age, null) + .set(user::type, null) + .where(user::id, eq(100L)) + .zipkinContext(null) + .sync(); UpdateOperation update = session.update(); - update - .set(user::name, null) - .zipkinContext(null) - .sync(); - + update.set(user::name, null).zipkinContext(null).sync(); } private void assertUsers(User expected, User actual) { diff --git a/src/test/java/net/helenus/test/integration/core/tuplecollection/TupleMapTest.java b/src/test/java/net/helenus/test/integration/core/tuplecollection/TupleMapTest.java index 143e4f0..a77276d 100644 --- a/src/test/java/net/helenus/test/integration/core/tuplecollection/TupleMapTest.java +++ b/src/test/java/net/helenus/test/integration/core/tuplecollection/TupleMapTest.java @@ -42,11 +42,8 @@ public class TupleMapTest extends TupleCollectionTest { // read full object - Book actual = session.select(book) - .where(book::id, Query.eq(id)) - .single() - .sync() - .orElse(null); + Book actual = + session.select(book).where(book::id, Query.eq(id)).single().sync().orElse(null); Assert.assertEquals(id, actual.id()); assertEqualMaps(writers, actual.writers()); Assert.assertNull(actual.reviewers()); @@ -77,11 +74,7 @@ public class TupleMapTest extends TupleCollectionTest { session.update().set(book::writers, expected).where(book::id, Query.eq(id)).sync(); - actual = session.select(book) - .where(book::id, Query.eq(id)) - .single() - .sync() - .orElse(null); + actual = session.select(book).where(book::id, Query.eq(id)).single().sync().orElse(null); Assert.assertEquals(id, actual.id()); assertEqualMaps(expected, actual.writers()); @@ -96,7 +89,8 @@ public class TupleMapTest extends TupleCollectionTest { expected.put(third, unk); session.update().put(book::writers, third, unk).where(book::id, Query.eq(id)).sync(); - actualMap = session.select(book::writers).where(book::id, Query.eq(id)).sync().findFirst().get()._1; + actualMap = + session.select(book::writers).where(book::id, Query.eq(id)).sync().findFirst().get()._1; assertEqualMaps(expected, actualMap); // putAll operation diff --git a/src/test/java/net/helenus/test/integration/core/unitofwork/AndThenOrderTest.java b/src/test/java/net/helenus/test/integration/core/unitofwork/AndThenOrderTest.java index 37ba96a..2b3ebcd 100644 --- a/src/test/java/net/helenus/test/integration/core/unitofwork/AndThenOrderTest.java +++ b/src/test/java/net/helenus/test/integration/core/unitofwork/AndThenOrderTest.java @@ -15,6 +15,9 @@ */ package net.helenus.test.integration.core.unitofwork; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; import net.helenus.core.Helenus; import net.helenus.core.HelenusSession; import net.helenus.core.UnitOfWork; @@ -23,102 +26,125 @@ import org.junit.Assert; import org.junit.BeforeClass; import org.junit.Test; -import java.util.ArrayList; -import java.util.Arrays; -import java.util.List; - - public class AndThenOrderTest extends AbstractEmbeddedCassandraTest { - static HelenusSession session; + static HelenusSession session; - @BeforeClass - public static void beforeTest() { - session = Helenus.init(getSession()) - .showCql() - .autoCreateDrop() - .get(); - } + @BeforeClass + public static void beforeTest() { + session = Helenus.init(getSession()).showCql().autoCreateDrop().get(); + } - @Test - public void testAndThenOrdering() throws Exception { - List q = new ArrayList(5); - UnitOfWork uow1, uow2, uow3, uow4, uow5; + @Test + public void testAndThenOrdering() throws Exception { + List q = new ArrayList(5); + UnitOfWork uow1, uow2, uow3, uow4, uow5; - uow5 = session.begin(); - uow3 = session.begin(uow5); - uow1 = session.begin(uow3); - uow1.commit().andThen(() -> { q.add("1"); }); - uow2 = session.begin(uow3); - uow2.commit().andThen(() -> { q.add("2"); }); - uow3.commit().andThen(() -> { q.add("3"); }); - uow4 = session.begin(uow5); - uow4.commit().andThen(() -> { q.add("4"); }); - uow5.commit().andThen(() -> { q.add("5"); }); + uow5 = session.begin(); + uow3 = session.begin(uow5); + uow1 = session.begin(uow3); + uow1.commit() + .andThen( + () -> { + q.add("1"); + }); + uow2 = session.begin(uow3); + uow2.commit() + .andThen( + () -> { + q.add("2"); + }); + uow3.commit() + .andThen( + () -> { + q.add("3"); + }); + uow4 = session.begin(uow5); + uow4.commit() + .andThen( + () -> { + q.add("4"); + }); + uow5.commit() + .andThen( + () -> { + q.add("5"); + }); - System.out.println(q); - Assert.assertTrue(Arrays.equals(q.toArray(new String[5]), new String[] {"1", "2", "3", "4", "5"})); + System.out.println(q); + Assert.assertTrue( + Arrays.equals(q.toArray(new String[5]), new String[] {"1", "2", "3", "4", "5"})); + } - } + @Test + public void testExceptionWithinAndThen() throws Exception { + List q = new ArrayList(5); + UnitOfWork uow1, uow2, uow3, uow4, uow5; - @Test - public void testExceptionWithinAndThen() throws Exception { - List q = new ArrayList(5); - UnitOfWork uow1, uow2, uow3, uow4, uow5; - - uow5 = session.begin(); - uow4 = session.begin(uow5); - try { - uow3 = session.begin(uow4); - uow1 = session.begin(uow3); - uow1.commit().andThen(() -> { + uow5 = session.begin(); + uow4 = session.begin(uow5); + try { + uow3 = session.begin(uow4); + uow1 = session.begin(uow3); + uow1.commit() + .andThen( + () -> { q.add("1"); - }); - uow2 = session.begin(uow3); - uow2.commit().andThen(() -> { + }); + uow2 = session.begin(uow3); + uow2.commit() + .andThen( + () -> { q.add("2"); - }); - uow3.commit().andThen(() -> { + }); + uow3.commit() + .andThen( + () -> { q.add("3"); - }); - uow4.commit().andThen(() -> { + }); + uow4.commit() + .andThen( + () -> { q.add("4"); + }); + throw new Exception(); + } catch (Exception e) { + uow4.abort(); + } + uow5.commit() + .andThen( + () -> { + q.add("5"); }); - throw new Exception(); - } catch(Exception e) { - uow4.abort(); - } - uow5.commit().andThen(() -> { q.add("5"); }); - System.out.println(q); - Assert.assertTrue(q.isEmpty() == true); + System.out.println(q); + Assert.assertTrue(q.isEmpty() == true); + } + @Test + public void testClosableWillAbortWhenNotCommitted() throws Exception { + UnitOfWork unitOfWork; + try (UnitOfWork uow = session.begin()) { + unitOfWork = uow; + Assert.assertFalse(uow.hasAborted()); } + Assert.assertTrue(unitOfWork.hasAborted()); + } - @Test - public void testClosableWillAbortWhenNotCommitted() throws Exception { - UnitOfWork unitOfWork; - try(UnitOfWork uow = session.begin()) { - unitOfWork = uow; - Assert.assertFalse(uow.hasAborted()); - } - Assert.assertTrue(unitOfWork.hasAborted()); - - } - - @Test - public void testClosable() throws Exception { - UnitOfWork unitOfWork; - try(UnitOfWork uow = session.begin()) { - unitOfWork = uow; - Assert.assertFalse(uow.hasAborted()); - uow.commit().andThen(() -> { + @Test + public void testClosable() throws Exception { + UnitOfWork unitOfWork; + try (UnitOfWork uow = session.begin()) { + unitOfWork = uow; + Assert.assertFalse(uow.hasAborted()); + uow.commit() + .andThen( + () -> { Assert.assertFalse(uow.hasAborted()); Assert.assertTrue(uow.hasCommitted()); - }); - } - Assert.assertFalse(unitOfWork.hasAborted()); - Assert.assertTrue(unitOfWork.hasCommitted()); + }); } - + Assert.assertFalse(unitOfWork.hasAborted()); + Assert.assertTrue(unitOfWork.hasCommitted()); + } } diff --git a/src/test/java/net/helenus/test/integration/core/unitofwork/Directory.java b/src/test/java/net/helenus/test/integration/core/unitofwork/Directory.java index 4984d9d..70cf9e9 100644 --- a/src/test/java/net/helenus/test/integration/core/unitofwork/Directory.java +++ b/src/test/java/net/helenus/test/integration/core/unitofwork/Directory.java @@ -15,16 +15,13 @@ */ package net.helenus.test.integration.core.unitofwork; -import net.helenus.mapping.annotation.*; - import com.datastax.driver.core.DataType.Name; import java.util.Set; - +import net.helenus.mapping.annotation.*; @UDT public interface Directory extends FilesystemNode { - @Types.Set(Name.TIMEUUID) - Set inodes(); - + @Types.Set(Name.TIMEUUID) + Set inodes(); } diff --git a/src/test/java/net/helenus/test/integration/core/unitofwork/File.java b/src/test/java/net/helenus/test/integration/core/unitofwork/File.java index 4e8ff6f..a0ccebd 100644 --- a/src/test/java/net/helenus/test/integration/core/unitofwork/File.java +++ b/src/test/java/net/helenus/test/integration/core/unitofwork/File.java @@ -17,11 +17,9 @@ package net.helenus.test.integration.core.unitofwork; import net.helenus.mapping.annotation.*; - @UDT public interface File extends FilesystemNode { - @Column - byte[] data(); - + @Column + byte[] data(); } diff --git a/src/test/java/net/helenus/test/integration/core/unitofwork/FileAttributes.java b/src/test/java/net/helenus/test/integration/core/unitofwork/FileAttributes.java index 6d2db6d..f2eaccc 100644 --- a/src/test/java/net/helenus/test/integration/core/unitofwork/FileAttributes.java +++ b/src/test/java/net/helenus/test/integration/core/unitofwork/FileAttributes.java @@ -20,6 +20,5 @@ import net.helenus.mapping.annotation.UDT; @UDT public interface FileAttributes { - String owner(); - + String owner(); } diff --git a/src/test/java/net/helenus/test/integration/core/unitofwork/FilesystemNode.java b/src/test/java/net/helenus/test/integration/core/unitofwork/FilesystemNode.java index a57f296..7907fbd 100644 --- a/src/test/java/net/helenus/test/integration/core/unitofwork/FilesystemNode.java +++ b/src/test/java/net/helenus/test/integration/core/unitofwork/FilesystemNode.java @@ -15,20 +15,18 @@ */ package net.helenus.test.integration.core.unitofwork; -import net.helenus.mapping.annotation.*; - import java.util.UUID; +import net.helenus.mapping.annotation.*; @Table("fs") public interface FilesystemNode { - @PartitionKey - UUID inode(); + @PartitionKey + UUID inode(); - @ClusteringColumn - String name(); - - @Column - FileAttributes attr(); + @ClusteringColumn + String name(); + @Column + FileAttributes attr(); } diff --git a/src/test/java/net/helenus/test/integration/core/unitofwork/UnitOfWorkTest.java b/src/test/java/net/helenus/test/integration/core/unitofwork/UnitOfWorkTest.java index 762c668..4e45fff 100644 --- a/src/test/java/net/helenus/test/integration/core/unitofwork/UnitOfWorkTest.java +++ b/src/test/java/net/helenus/test/integration/core/unitofwork/UnitOfWorkTest.java @@ -15,7 +15,10 @@ */ package net.helenus.test.integration.core.unitofwork; +import static net.helenus.core.Query.eq; + import com.datastax.driver.core.utils.UUIDs; +import java.util.UUID; import net.bytebuddy.utility.RandomString; import net.helenus.core.Helenus; import net.helenus.core.HelenusSession; @@ -29,167 +32,161 @@ import org.junit.Assert; import org.junit.BeforeClass; import org.junit.Test; -import java.util.UUID; - -import static net.helenus.core.Query.eq; - - @Table @Cacheable interface Widget { - @PartitionKey - UUID id(); - @Column - String name(); -} + @PartitionKey + UUID id(); + @Column + String name(); +} public class UnitOfWorkTest extends AbstractEmbeddedCassandraTest { - static Widget widget; - static HelenusSession session; + static Widget widget; + static HelenusSession session; + @BeforeClass + public static void beforeTest() { + session = Helenus.init(getSession()).showCql().add(Widget.class).autoCreateDrop().get(); + widget = session.dsl(Widget.class); + } - @BeforeClass - public static void beforeTest() { - session = Helenus.init(getSession()) - .showCql() - .add(Widget.class) - .autoCreateDrop() - .get(); - widget = session.dsl(Widget.class); - } + @Test + public void testSelectAfterSelect() throws Exception { + Widget w1, w2; + UUID key = UUIDs.timeBased(); - @Test - public void testSelectAfterSelect() throws Exception { - Widget w1, w2; - UUID key = UUIDs.timeBased(); + // This should inserted Widget, but not cache it. + session + .insert(widget) + .value(widget::id, key) + .value(widget::name, RandomString.make(20)) + .sync(); - // This should inserted Widget, but not cache it. - session.insert(widget) - .value(widget::id, key) - .value(widget::name, RandomString.make(20)) - .sync(); + try (UnitOfWork uow = session.begin()) { - try (UnitOfWork uow = session.begin()) { + // This should read from the database and return a Widget. + w1 = + session.select(widget).where(widget::id, eq(key)).single().sync(uow).orElse(null); - // This should read from the database and return a Widget. - w1 = session.select(widget) - .where(widget::id, eq(key)) - .single() - .sync(uow) - .orElse(null); - - // This should read from the cache and get the same instance of a Widget. - w2 = session.select(widget) - .where(widget::id, eq(key)) - .single() - .sync(uow) - .orElse(null); - - uow.commit() - .andThen(() -> { - Assert.assertEquals(w1, w2); - }); - } - - } - - @Test - public void testSelectAfterNestedSelect() throws Exception { - Widget w1, w2, w3, w4; - UUID key1 = UUIDs.timeBased(); - UUID key2 = UUIDs.timeBased(); - - // This should inserted Widget, and not cache it in uow1. - try (UnitOfWork uow1 = session.begin()) { - w1 = session.insert(widget) - .value(widget::id, key1) - .value(widget::name, RandomString.make(20)) - .sync(uow1); - - try (UnitOfWork uow2 = session.begin(uow1)) { - - // This should read from uow1's cache and return the same Widget. - w2 = session.select(widget) - .where(widget::id, eq(key1)) - .single() - .sync(uow2) - .orElse(null); + // This should read from the cache and get the same instance of a Widget. + w2 = + session.select(widget).where(widget::id, eq(key)).single().sync(uow).orElse(null); + uow.commit() + .andThen( + () -> { Assert.assertEquals(w1, w2); - - w3 = session.insert(widget) - .value(widget::id, key2) - .value(widget::name, RandomString.make(20)) - .sync(uow2); - - uow2.commit() - .andThen(() -> { - Assert.assertEquals(w1, w2); - }); - } - - // This should read from the cache and get the same instance of a Widget. - w4 = session.select(widget) - .where(widget::id, eq(key2)) - .single() - .sync(uow1) - .orElse(null); - - uow1.commit() - .andThen(() -> { - Assert.assertEquals(w3, w4); - }); - } - + }); } -/* - @Test - public void testSelectAfterInsertProperlyCachesEntity() throws Exception { - Widget w1, w2, w3, w4; - UUID key = UUIDs.timeBased(); + } - try (UnitOfWork uow = session.begin()) { + @Test + public void testSelectAfterNestedSelect() throws Exception { + Widget w1, w2, w3, w4; + UUID key1 = UUIDs.timeBased(); + UUID key2 = UUIDs.timeBased(); - // This should cache the inserted Widget. - w1 = session.insert(widget) - .value(widget::id, key) - .value(widget::name, RandomString.make(20)) - .sync(uow); + // This should inserted Widget, and not cache it in uow1. + try (UnitOfWork uow1 = session.begin()) { + w1 = + session + .insert(widget) + .value(widget::id, key1) + .value(widget::name, RandomString.make(20)) + .sync(uow1); - // This should read from the cache and get the same instance of a Widget. - w2 = session.select(widget) - .where(widget::id, eq(key)) - .single() - .sync(uow) - .orElse(null); + try (UnitOfWork uow2 = session.begin(uow1)) { - uow.commit() - .andThen(() -> { - Assert.assertEquals(w1, w2); - }); - } - - // This should read the widget from the session cache and maintain object identity. - w3 = session.select(widget) - .where(widget::id, eq(key)) + // This should read from uow1's cache and return the same Widget. + w2 = + session + .select(widget) + .where(widget::id, eq(key1)) .single() - .sync() + .sync(uow2) .orElse(null); - Assert.assertEquals(w1, w3); + Assert.assertEquals(w1, w2); - // This should read the widget from the database, no object identity but values should match. - w4 = session.select(widget) - .where(widget::id, eq(key)) - .ignoreCache() - .single() - .sync() - .orElse(null); + w3 = + session + .insert(widget) + .value(widget::id, key2) + .value(widget::name, RandomString.make(20)) + .sync(uow2); - Assert.assertNotEquals(w1, w4); - Assert.assertTrue(w1.equals(w4)); + uow2.commit() + .andThen( + () -> { + Assert.assertEquals(w1, w2); + }); + } + + // This should read from the cache and get the same instance of a Widget. + w4 = + session + .select(widget) + .where(widget::id, eq(key2)) + .single() + .sync(uow1) + .orElse(null); + + uow1.commit() + .andThen( + () -> { + Assert.assertEquals(w3, w4); + }); } -*/ + } + /* + @Test + public void testSelectAfterInsertProperlyCachesEntity() throws Exception { + Widget w1, w2, w3, w4; + UUID key = UUIDs.timeBased(); + + try (UnitOfWork uow = session.begin()) { + + // This should cache the inserted Widget. + w1 = session.insert(widget) + .value(widget::id, key) + .value(widget::name, RandomString.make(20)) + .sync(uow); + + // This should read from the cache and get the same instance of a Widget. + w2 = session.select(widget) + .where(widget::id, eq(key)) + .single() + .sync(uow) + .orElse(null); + + uow.commit() + .andThen(() -> { + Assert.assertEquals(w1, w2); + }); + } + + // This should read the widget from the session cache and maintain object identity. + w3 = session.select(widget) + .where(widget::id, eq(key)) + .single() + .sync() + .orElse(null); + + Assert.assertEquals(w1, w3); + + // This should read the widget from the database, no object identity but values should match. + w4 = session.select(widget) + .where(widget::id, eq(key)) + .ignoreCache() + .single() + .sync() + .orElse(null); + + Assert.assertNotEquals(w1, w4); + Assert.assertTrue(w1.equals(w4)); + } + */ } diff --git a/src/test/java/net/helenus/test/unit/core/dsl/Account.java b/src/test/java/net/helenus/test/unit/core/dsl/Account.java index 2725f16..10fc9fd 100644 --- a/src/test/java/net/helenus/test/unit/core/dsl/Account.java +++ b/src/test/java/net/helenus/test/unit/core/dsl/Account.java @@ -19,7 +19,6 @@ import java.util.Date; import java.util.Map; import java.util.Set; import net.helenus.core.reflect.Drafted; -import net.helenus.mapping.HelenusEntity; import net.helenus.mapping.annotation.*; @Table