• Home
• com.cloudimpl.outstack
• outstack-spring
• 4.0.544
• source code
• DynamodbEventRepository.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.cloudimpl.outstack.spring.repo.DynamodbEventRepository Maven / Gradle / Ivy

/*
 * Copyright 2021 nuwan.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.cloudimpl.outstack.spring.repo;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;
import com.amazonaws.services.dynamodbv2.document.*;
import com.amazonaws.services.dynamodbv2.document.internal.IteratorSupport;
import com.amazonaws.services.dynamodbv2.document.spec.QuerySpec;
import com.amazonaws.services.dynamodbv2.document.utils.ValueMap;
import com.amazonaws.services.dynamodbv2.model.AttributeValue;
import com.amazonaws.services.dynamodbv2.model.Delete;
import com.amazonaws.services.dynamodbv2.model.Put;
import com.amazonaws.services.dynamodbv2.model.QueryRequest;
import com.amazonaws.services.dynamodbv2.model.QueryResult;
import com.amazonaws.services.dynamodbv2.model.ReturnConsumedCapacity;
import com.amazonaws.services.dynamodbv2.model.ReturnValuesOnConditionCheckFailure;
import com.amazonaws.services.dynamodbv2.model.TransactWriteItem;
import com.amazonaws.services.dynamodbv2.model.TransactWriteItemsRequest;
import com.amazonaws.services.dynamodbv2.model.Update;
import com.cloudimpl.outstack.common.GsonCodec;
import com.cloudimpl.outstack.runtime.EntityCheckpoint;
import com.cloudimpl.outstack.runtime.EntityContextProvider;
import com.cloudimpl.outstack.runtime.EntityIdHelper;
import com.cloudimpl.outstack.runtime.EventRepositoy;
import com.cloudimpl.outstack.runtime.EventStream;
import com.cloudimpl.outstack.runtime.ResourceHelper;
import com.cloudimpl.outstack.runtime.ResultSet;
import com.cloudimpl.outstack.runtime.domainspec.ChildEntity;
import com.cloudimpl.outstack.runtime.domainspec.Entity;
import com.cloudimpl.outstack.runtime.domainspec.Event;
import com.cloudimpl.outstack.runtime.domainspec.Query;
import com.cloudimpl.outstack.runtime.domainspec.RootEntity;
import com.cloudimpl.outstack.runtime.domainspec.TenantRequirement;
import com.cloudimpl.outstack.runtime.repo.EventRepoUtil;
import com.cloudimpl.outstack.runtime.repo.RepositoryException;
import com.cloudimpl.outstack.spring.component.SpringApplicationConfigManager.Provider;
import java.text.MessageFormat;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;

/**
 *
 * @author nuwan
 * @param 
 */
public class DynamodbEventRepository extends EventRepositoy {

    private final Provider.ProviderConfigs configs;
    private final DynamoDB dynamodb;
    private final AmazonDynamoDB client;
    private String entityTable;
    private String eventTable;
    public static final String partitionKeyColumn = "partitionKey";
    public static final String rangeKeyColumn = "rangeKey";
    public static final String jsonColumn = "json";
    public static final String lastSeqColumn = "lastSeq";
    public static final String eventTypeColumn = "eventType";
    public static final String eventOwnerColumn = "eventOwner";
    public static final String ownerIdColumn = "ownerId";
    //private final int partitionCount;
    private final String tableName;
    ThreadLocal> txContext = ThreadLocal.withInitial(() -> new LinkedList<>());

    public DynamodbEventRepository(AmazonDynamoDB client, DynamoDB dynamodb, Class rootType, ResourceHelper resourceHelper, Provider.ProviderConfigs configs) {
        super(rootType, resourceHelper);
        this.dynamodb = dynamodb;
        this.client = client;
        //this.loadTables();
        this.configs = configs;
        this.tableName = this.configs.getOption(rootType.getSimpleName() + "Table").or(()->this.configs.getOption("defaultTable")).get();
        System.out.println("table name " + this.tableName + " pick for root type: " + rootType.getSimpleName());
        //this.partitionCount = Integer.valueOf(configs.getOption("partitionCount").get());
    }

    @Override
    protected void startTransaction() {
        List list = txContext.get();
        list.clear();

    }

    @Override
    protected void endTransaction() {
        TransactWriteItemsRequest transaction = new TransactWriteItemsRequest()
                .withTransactItems(txContext.get())
                .withReturnConsumedCapacity(ReturnConsumedCapacity.TOTAL);

        // Execute the transaction and process the result.
        try {
            client.transactWriteItems(transaction);
            System.out.println("Transaction Successful");

        } catch (Exception ex) {
            throw new RepositoryException(ex);
        }
    }

    @Override
    protected void saveRootEntityBrnIfNotExist(RootEntity e) {
        String rn = resourceHelper.getFQBrn(e.getBRN());
        String prefix = rn.substring(0, rn.lastIndexOf("/"));
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(prefix));
        item.put(this.rangeKeyColumn, new AttributeValue(rn));
        item.put(this.jsonColumn, new AttributeValue(GsonCodec.encode(e)));
        item.put(this.lastSeqColumn, new AttributeValue().withN(String.valueOf(e.getMeta().getLastSeq())));

        Put createEntity = new Put()
                .withTableName(this.tableName)
                .withItem(item)
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD)
                .withConditionExpression("attribute_not_exists(" + this.rangeKeyColumn + ")");
        txContext.get().add(new TransactWriteItem().withPut(createEntity));
    }

    @Override
    protected void saveRootEntityTrnIfNotExist(RootEntity e) {
        String trn = resourceHelper.getFQTrn(e.getTRN());
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(trn));
        item.put(this.rangeKeyColumn, new AttributeValue(trn));
        item.put(this.jsonColumn, new AttributeValue(GsonCodec.encode(e)));
        item.put(this.lastSeqColumn, new AttributeValue().withN(String.valueOf(e.getMeta().getLastSeq())));

        Put createEntity = new Put()
                .withTableName(this.tableName)
                .withItem(item)
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD)
                .withConditionExpression("attribute_not_exists(" + this.partitionKeyColumn + ")");

        txContext.get().add(new TransactWriteItem().withPut(createEntity));
    }

    @Override
    protected void saveRootEntityBrnIfExist(long lastSeq, RootEntity e) {
        String rn = resourceHelper.getFQBrn(e.getBRN());
        String prefix = rn.substring(0, rn.lastIndexOf("/"));
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(prefix));
        item.put(this.rangeKeyColumn, new AttributeValue(rn));

        Map expressionAttributeValues = new HashMap<>();
        expressionAttributeValues.put(":lastSeq", new AttributeValue().withN(String.valueOf(e.getMeta().getLastSeq())));
        expressionAttributeValues.put(":checkLastSeq", new AttributeValue().withN(String.valueOf(lastSeq)));
        expressionAttributeValues.put(":json", new AttributeValue(GsonCodec.encode(e)));

        Update updateEntity = new Update()
                .withTableName(this.tableName)
                .withKey(item)
                .withUpdateExpression("SET json = :json , lastSeq = :lastSeq")
                .withExpressionAttributeValues(expressionAttributeValues)
                .withConditionExpression("lastSeq = :checkLastSeq and attribute_exists(" + this.rangeKeyColumn + ")")
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD);
        txContext.get().add(new TransactWriteItem().withUpdate(updateEntity));
    }

    @Override
    protected void saveRootEntityTrnIfExist(long lastSeq, RootEntity e) {
        String trn = resourceHelper.getFQTrn(e.getTRN());
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(trn));
        item.put(this.rangeKeyColumn, new AttributeValue(trn));

        Map expressionAttributeValues = new HashMap<>();
        expressionAttributeValues.put(":lastSeq", new AttributeValue().withN(String.valueOf(e.getMeta().getLastSeq())));
        expressionAttributeValues.put(":checkLastSeq", new AttributeValue().withN(String.valueOf(lastSeq)));
        expressionAttributeValues.put(":json", new AttributeValue(GsonCodec.encode(e)));

        Update updateEntity = new Update()
                .withTableName(this.tableName)
                .withKey(item)
                .withUpdateExpression("SET json = :json , lastSeq = :lastSeq")
                .withExpressionAttributeValues(expressionAttributeValues)
                .withConditionExpression("lastSeq = :checkLastSeq and attribute_exists(" + this.rangeKeyColumn + ")")
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD);
        txContext.get().add(new TransactWriteItem().withUpdate(updateEntity));
    }

    @Override
    protected void saveChildEntityBrnIfNotExist(ChildEntity e) {
        String partitionKey = resourceHelper.getFQTrn(e.getRootTRN());
        String rangeKey = resourceHelper.getFQBrn(e.getBRN());
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(partitionKey));
        item.put(this.rangeKeyColumn, new AttributeValue(rangeKey));
        item.put(this.jsonColumn, new AttributeValue(GsonCodec.encode(e)));
        item.put(this.lastSeqColumn, new AttributeValue().withN(String.valueOf(e.getMeta().getLastSeq())));

        Put createEntity = new Put()
                .withTableName(this.tableName)
                .withItem(item)
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD)
                .withConditionExpression("attribute_not_exists(" + this.rangeKeyColumn + ")");
        txContext.get().add(new TransactWriteItem().withPut(createEntity));
    }

    @Override
    protected void saveChildEntityTrnIfNotExist(ChildEntity e) {
        String partitionKey = resourceHelper.getFQTrn(e.getRootTRN());
        String rangeKey = resourceHelper.getFQTrn(e.getTRN());
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(partitionKey));
        item.put(this.rangeKeyColumn, new AttributeValue(rangeKey));
        item.put(this.jsonColumn, new AttributeValue(GsonCodec.encode(e)));
        item.put(this.lastSeqColumn, new AttributeValue().withN(String.valueOf(e.getMeta().getLastSeq())));

        Put createEntity = new Put()
                .withTableName(this.tableName)
                .withItem(item)
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD)
                .withConditionExpression("attribute_not_exists(" + this.rangeKeyColumn + ")");
        txContext.get().add(new TransactWriteItem().withPut(createEntity));
    }

    @Override
    protected void saveChildEntityBrnIfExist(long lastSeq,ChildEntity e) {
        String partitionKey = resourceHelper.getFQTrn(e.getRootTRN());
        String rangeKey = resourceHelper.getFQBrn(e.getBRN());
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(partitionKey));
        item.put(this.rangeKeyColumn, new AttributeValue(rangeKey));

        Map expressionAttributeValues = new HashMap<>();
        expressionAttributeValues.put(":lastSeq", new AttributeValue().withN(String.valueOf(e.getMeta().getLastSeq())));
        expressionAttributeValues.put(":checkLastSeq", new AttributeValue().withN(String.valueOf(lastSeq)));
        expressionAttributeValues.put(":json", new AttributeValue(GsonCodec.encode(e)));

        Update updateEntity = new Update()
                .withTableName(this.tableName)
                .withKey(item)
                .withUpdateExpression("SET json = :json , lastSeq = :lastSeq")
                .withExpressionAttributeValues(expressionAttributeValues)
                .withConditionExpression("lastSeq = :checkLastSeq and attribute_exists(" + this.rangeKeyColumn + ")")
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD);

        txContext.get().add(new TransactWriteItem().withUpdate(updateEntity));
    }

    @Override
    protected void saveChildEntityTrnIfExist(long lastSeq,ChildEntity e) {
        String partitionKey = resourceHelper.getFQTrn(e.getRootTRN());
        String rangeKey = resourceHelper.getFQTrn(e.getTRN());
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(partitionKey));
        item.put(this.rangeKeyColumn, new AttributeValue(rangeKey));

        Map expressionAttributeValues = new HashMap<>();
        expressionAttributeValues.put(":lastSeq", new AttributeValue().withN(String.valueOf(e.getMeta().getLastSeq())));
        expressionAttributeValues.put(":checkLastSeq", new AttributeValue().withN(String.valueOf(lastSeq)));
        expressionAttributeValues.put(":json", new AttributeValue(GsonCodec.encode(e)));

        Update updateEntity = new Update()
                .withTableName(this.tableName)
                .withKey(item)
                .withUpdateExpression("SET json = :json , lastSeq = :lastSeq")
                .withExpressionAttributeValues(expressionAttributeValues)
                .withConditionExpression("lastSeq = :checkLastSeq and attribute_exists(" + this.rangeKeyColumn + ")")
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD);

        txContext.get().add(new TransactWriteItem().withUpdate(updateEntity));
    }

    @Override
    protected void deleteRootEntityBrnById(RootEntity e,boolean deleteOnlyEntity) {
        String rn = resourceHelper.getFQBrn(RootEntity.makeRN(e.getClass(), version, e.entityId(), e.getTenantId()));
        String prefix = rn.substring(0, rn.lastIndexOf("/"));
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(prefix));
        item.put(this.rangeKeyColumn, new AttributeValue(rn));
        Delete delete = new Delete()
                .withTableName(tableName)
                .withKey(item);
        txContext.get().add(new TransactWriteItem().withDelete(delete));
    }

    @Override
    protected void deleteRootEntityTrnById(Class rootType, String id, String tenantId) {
        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
    }

    @Override
    protected > void deleteChildEntityBrnById(ChildEntity e,boolean deleteOnlyEntity) {
        String partitionKey = resourceHelper.getFQTrn(RootEntity.makeTRN(rootType, version, e.id(),e.getTenantId()));
        String rangeKey = resourceHelper.getFQBrn(ChildEntity.makeRN(rootType, version, e.rootId(), e.getClass(), e.id(), e.getTenantId()));
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(partitionKey));
        item.put(this.rangeKeyColumn, new AttributeValue(rangeKey));
        Delete delete = new Delete()
                .withTableName(tableName)
                .withKey(item);
        txContext.get().add(new TransactWriteItem().withDelete(delete));
    }

    @Override
    protected > void deleteChildEntityTrnById(Class rootType, String id, Class childType, String childId, String tenantId) {
        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
    }

    @Override
    protected Optional _getCheckpoint(String rootTrn) {

        String partitionKey = resourceHelper.getFQTrn(rootTrn);
        Map expressionAttributesNames = new HashMap<>();
        expressionAttributesNames.put("#partitionKey", "partitionKey");
        expressionAttributesNames.put("#rangeKey", "rangeKey");
        expressionAttributesNames.put("#json", "json");

        Map expressionAttributeValues = new HashMap<>();
        expressionAttributeValues.put(":partitionKey", new AttributeValue().withS(partitionKey));
        expressionAttributeValues.put(":rangeKey", new AttributeValue().withS(EntityCheckpoint.class.getSimpleName()));

        QueryRequest queryRequest = new QueryRequest()
                .withTableName(this.tableName)
                //.withAttributesToGet("json")
                .withKeyConditionExpression("partitionKey = :partitionKey AND rangeKey = :rangeKey")
                //  .withExpressionAttributeNames(expressionAttributesNames)
                .withExpressionAttributeValues(expressionAttributeValues);

        QueryResult queryResult = client.query(queryRequest);
        List> items = queryResult.getItems();
        if (items.size() > 0) {
            return Optional.of(GsonCodec.decode(EntityCheckpoint.class, items.get(0).get("json").getS()));
        } else {
            return Optional.of(new EntityCheckpoint(rootTrn).setSeq(0));
        }
    }

    @Override
    protected void addEvent(Event event) {
        String trn = resourceHelper.getFQTrn(event.getRootEntityTRN() + "/events");
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(trn));
        item.put(this.rangeKeyColumn, new AttributeValue(String.format("%019d", event.getSeqNum())));
        item.put(this.jsonColumn, new AttributeValue(GsonCodec.encodeWithType(event)));
        item.put(this.lastSeqColumn, new AttributeValue().withN(String.valueOf(event.getSeqNum())));
        item.put(this.eventTypeColumn, new AttributeValue(event.getClass().getSimpleName()));
        item.put(this.eventOwnerColumn, new AttributeValue(event.getOwner().getSimpleName()));
        item.put(this.ownerIdColumn, new AttributeValue(event.id()));

        Put createEvent = new Put()
                .withTableName(this.tableName)
                .withItem(item)
                .withConditionExpression("attribute_not_exists(" + this.rangeKeyColumn + ")")
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD);
        txContext.get().add(new TransactWriteItem().withPut(createEvent));
    }

    private String getRootBrn(Class rootType, String tenantId) {
        String trn = null;
        TenantRequirement tenantReq = Entity.checkTenantRequirement(rootType);
        switch (tenantReq) {
            case REQUIRED:
                trn = resourcePrefix("brn") + ":tenant/" + tenantId + "/" + version + "/" + rootType.getSimpleName();
                break;
            case OPTIONAL:
                if (tenantId != null) {
                    trn = resourcePrefix("brn") + ":tenant/" + tenantId + "/" + version + "/" + rootType.getSimpleName();
                } else {
                    trn = resourcePrefix("brn") + ":" + version + "/" + rootType.getSimpleName();
                }
                break;
            default:
                trn = resourcePrefix("brn") + ":" + version + "/" + rootType.getSimpleName();
                break;
        }
        return trn;
    }

    private > String getChildBrn(Class rootType, String id, Class childType, String tenantId) {
        EntityIdHelper.validateTechnicalId(id);
        String trn = null;
        TenantRequirement tenantReq = Entity.checkTenantRequirement(rootType);
        switch (tenantReq) {
            case REQUIRED:
                trn = resourcePrefix("brn") + ":tenant/" + tenantId + "/" + version + "/" + rootType.getSimpleName() + "/" + id + "/" + childType.getSimpleName() + "/";
                break;
            case OPTIONAL:
                if (tenantId != null) {
                    trn = resourcePrefix("brn") + ":tenant/" + tenantId + "/" + version + "/" + rootType.getSimpleName() + "/" + id + "/" + childType.getSimpleName() + "/";
                } else {
                    trn = resourcePrefix("brn") + ":" + version + "/" + rootType.getSimpleName() + "/" + id + "/" + childType.getSimpleName() + "/";
                }
                break;
            default:
                trn = resourcePrefix("brn") + ":" + version + "/" + rootType.getSimpleName() + "/" + id + "/" + childType.getSimpleName() + "/";
                break;
        }
        return trn;
    }

    @Override
    public ResultSet getAllByRootType(Class rootType, String tenantId, Query.PagingRequest paging) {
        String partitionKey = getRootBrn(rootType, tenantId);
        //    Map expressionAttributesNames = new HashMap<>();
        //    expressionAttributesNames.put("#partitionKey", "partitionKey");

        Table table = this.dynamodb.getTable(this.tableName);
        QuerySpec querySpec = new QuerySpec().withKeyConditionExpression("partitionKey = :partitionKey").withValueMap(new ValueMap().withString(":partitionKey", partitionKey)).withConsistentRead(true);
        ItemCollection query = table.query(querySpec);
        Iterator iterator = query.iterator();
        Iterable iterable = () -> iterator;
        List items = StreamSupport.stream(iterable.spliterator(), false).map(attr -> GsonCodec.decode(rootType, attr.getString("json"))).filter(i->EventRepoUtil.onFilter(i, paging.getParams())).collect(Collectors.toList());

        return EventRepoUtil.onPageable(items, paging);
    }

    @Override
    public ResultSet getAllByRootType(Class rootType, Collection tenantId, Query.PagingRequest paging) {
        throw new UnsupportedOperationException("Not supported.");
    }

    @Override
    public  K executeRawQuery(String rawQuery) {
        throw new UnsupportedOperationException("Not supported.");
    }

    @Override
    public Optional getRootById(Class rootType, String id, String tenantId) {
        String pKey;
        String rKey;
        if (EntityIdHelper.isTechnicalId(id)) {
            pKey = resourceHelper.getFQTrn(RootEntity.makeTRN(rootType, version, id, tenantId));
            rKey = pKey;
        } else {
            pKey = getRootBrn(rootType, tenantId);
            rKey = resourceHelper.getFQBrn(RootEntity.makeRN(rootType, version, id, tenantId));
        }

        //      Map expressionAttributesNames = new HashMap<>();
        //      expressionAttributesNames.put("#partitionKey", "partitionKey");
        //      expressionAttributesNames.put("#rangeKey", "rangeKey");
        Map expressionAttributeValues = new HashMap<>();
        expressionAttributeValues.put(":partitionKey", new AttributeValue().withS(pKey));
        expressionAttributeValues.put(":rangeKey", new AttributeValue().withS(rKey));

        QueryRequest queryRequest = new QueryRequest()
                .withTableName(this.tableName)
                //    .withAttributesToGet("json")
                .withKeyConditionExpression("partitionKey = :partitionKey AND rangeKey = :rangeKey")
                //          .withExpressionAttributeNames(expressionAttributesNames)
                .withExpressionAttributeValues(expressionAttributeValues);

        QueryResult queryResult = client.query(queryRequest);
        List> items = queryResult.getItems();
        if (items.size() > 0) {
            return Optional.of(GsonCodec.decode(rootType, items.get(0).get("json").getS()));
        } else {
            return Optional.empty();
        }
    }

    @Override
    public > Optional getChildById(Class rootType, String id, Class childType, String childId, String tenantId) {
        EntityIdHelper.validateTechnicalId(id);
        String pKey = resourceHelper.getFQTrn(RootEntity.makeTRN(rootType, version, id, tenantId));
        String rKey;
        if (EntityIdHelper.isTechnicalId(childId)) {
            rKey = resourceHelper.getFQTrn(ChildEntity.makeTRN(rootType, version, id, childType, childId, tenantId));
        } else {
            rKey = resourceHelper.getFQBrn(ChildEntity.makeRN(rootType, version, id, childType, childId, tenantId));
        }
        //   Map expressionAttributesNames = new HashMap<>();
        //    expressionAttributesNames.put("#partitionKey", "partitionKey");
        //    expressionAttributesNames.put("#rangeKey", "rangeKey");

        Map expressionAttributeValues = new HashMap<>();
        expressionAttributeValues.put(":partitionKey", new AttributeValue().withS(pKey));
        expressionAttributeValues.put(":rangeKey", new AttributeValue().withS(rKey));

        QueryRequest queryRequest = new QueryRequest()
                .withTableName(this.tableName)
                //        .withAttributesToGet("json")
                .withKeyConditionExpression("partitionKey = :partitionKey AND rangeKey = :rangeKey")
                //            .withExpressionAttributeNames(expressionAttributesNames)
                .withExpressionAttributeValues(expressionAttributeValues);

        QueryResult queryResult = client.query(queryRequest);
        List> items = queryResult.getItems();
        if (items.size() > 0) {
            return Optional.of(GsonCodec.decode(childType, items.get(0).get("json").getS()));
        } else {
            return Optional.empty();
        }
    }

    @Override
    public > ResultSet getAllChildByType(Class rootType, String id, Class childType, String tenantId, Query.PagingRequest paging) {
        String pKey = resourceHelper.getFQTrn(RootEntity.makeTRN(rootType, version, id, tenantId));
        String rkey = getChildBrn(rootType, id, childType, tenantId);
        //   Map expressionAttributesNames = new HashMap<>();
        //   expressionAttributesNames.put("#partitionKey", "partitionKey");
        //   expressionAttributesNames.put("#rangeKey", "rangeKey");

        Table table = this.dynamodb.getTable(this.tableName);
        QuerySpec querySpec = new QuerySpec().withKeyConditionExpression("partitionKey = :partitionKey AND begins_with(rangeKey ,:rangeKey)").withValueMap(new ValueMap().withString(":partitionKey", pKey).withString(":rangeKey", rkey)).withConsistentRead(true);
        ItemCollection query = table.query(querySpec);
        Iterator iterator = query.iterator();
        Iterable iterable = () -> iterator;

        List items = StreamSupport.stream(iterable.spliterator(), false).map(attr -> GsonCodec.decode(childType, attr.getString("json"))).filter(i->EventRepoUtil.onFilter(i, paging.getParams())).collect(Collectors.toList());

        return EventRepoUtil.onPageable(items, paging);
    }

    @Override
    public > ResultSet getAllChildByType(Class rootType, String id, Class childType, Collection tenantId, Query.PagingRequest paging) {
        throw new UnsupportedOperationException("Not supported.");
    }

    @Override
    public ResultSet> getEventsByRootId(Class rootType, String rootId, String tenantId, Query.PagingRequest paging) {
        String trn = resourceHelper.getFQTrn(RootEntity.makeTRN(rootType, version, rootId, tenantId) + "/events");
   //     Map expressionAttributesNames = new HashMap<>();
   //     expressionAttributesNames.put("#partitionKey", "partitionKey");
   //     expressionAttributesNames.put("#eventOwner", "eventOwner");

        Table table = this.dynamodb.getTable(this.tableName);
        QuerySpec querySpec = new QuerySpec().withKeyConditionExpression("partitionKey = :partitionKey AND rangeKey > :rangeKey")
                .withFilterExpression("eventOwner = :eventOwner")
                .withValueMap(new ValueMap()
                        .withString(":partitionKey", trn).withString(":rangeKey", String.format("%019d", 0)).withString(":eventOwner", rootType.getSimpleName()))
                .withConsistentRead(true)
                .withScanIndexForward(Boolean.FALSE);

        ItemCollection query = table.query(querySpec);
        Iterator iterator = query.iterator();
        Iterable iterable = () -> iterator;
        List items = StreamSupport.stream(iterable.spliterator(), false).map(attr -> GsonCodec.decode(attr.getString("json"))).filter(i->EventRepoUtil.onFilter(i, paging.getParams())).collect(Collectors.toList());

        return EventRepoUtil.onPageable(items, paging);
    }

    @Override
    public > ResultSet> getEventsByChildId(Class rootType, String id, Class childType, String childId, String tenantId, Query.PagingRequest paging) {
        String trn = resourceHelper.getFQTrn(RootEntity.makeTRN(rootType, version, id, tenantId) + "/events");
    //    Map expressionAttributesNames = new HashMap<>();
    //    expressionAttributesNames.put("#partitionKey", "partitionKey");
    //    expressionAttributesNames.put("#eventOwner", "eventOwner");
        if(!EntityIdHelper.isTechnicalId(childId))
        {
            childId = getChildById(rootType, id, childType, childId, tenantId).get().id();
        }

        Table table = this.dynamodb.getTable(this.tableName);
        QuerySpec querySpec = new QuerySpec().withKeyConditionExpression("partitionKey = :partitionKey AND rangeKey > :rangeKey")
                .withFilterExpression("eventOwner = :eventOwner AND ownerId = :ownerId")
                .withValueMap(new ValueMap()
                        .withString(":partitionKey", trn).withString(":rangeKey", String.format("%019d", 0)).withString(":ownerId", childId).withString(":eventOwner", childType.getSimpleName()))
                .withConsistentRead(true)
                .withScanIndexForward(Boolean.FALSE);

        ItemCollection query = table.query(querySpec);
        Iterator iterator = query.iterator();
        Iterable iterable = () -> iterator;
        List items = StreamSupport.stream(iterable.spliterator(), false).map(attr -> GsonCodec.decode(attr.getString("json"))).filter(i->EventRepoUtil.onFilter(i, paging.getParams())).collect(Collectors.toList());

        return EventRepoUtil.onPageable(items, paging);
    }

    @Override
    protected void updateCheckpoint(EntityCheckpoint checkpoint) {
        String partitionKey = resourceHelper.getFQTrn(checkpoint.getRootTrn());
        HashMap item = new HashMap<>();
        item.put(this.partitionKeyColumn, new AttributeValue(partitionKey));
        item.put(this.rangeKeyColumn, new AttributeValue(EntityCheckpoint.class.getSimpleName()));
        item.put(this.jsonColumn, new AttributeValue(GsonCodec.encode(checkpoint)));

        Put createEntity = new Put()
                .withTableName(this.tableName)
                .withItem(item)
                .withReturnValuesOnConditionCheckFailure(ReturnValuesOnConditionCheckFailure.ALL_OLD);
        txContext.get().add(new TransactWriteItem().withPut(createEntity));
    }

    @Override
    public boolean isIdExist(String id,String tenantId) {
        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
    }

}