diff --git a/src/main/java/net/helenus/core/operation/InsertOperation.java b/src/main/java/net/helenus/core/operation/InsertOperation.java index 3ee0c8a..53abaee 100644 --- a/src/main/java/net/helenus/core/operation/InsertOperation.java +++ b/src/main/java/net/helenus/core/operation/InsertOperation.java @@ -83,9 +83,18 @@ public final class InsertOperation extends AbstractOperation keys = (mutations == null) ? null : mutations; for (HelenusProperty prop : properties) { + boolean addProp = false; - if (keys == null || keys.contains(prop.getPropertyName())) { + switch (prop.getColumnType()) { + case PARTITION_KEY: + case CLUSTERING_COLUMN: + addProp = true; + break; + default: + addProp = (keys == null || keys.contains(prop.getPropertyName())); + } + if (addProp) { Object value = BeanColumnValueProvider.INSTANCE.getColumnValue(pojo, -1, prop); value = sessionOps.getValuePreparer().prepareColumnValue(value, prop);