Search in sources :

Example 1 with EntityKey

use of io.openk9.entity.manager.cache.model.EntityKey in project openk9 by smclab.

the class EntityManagerBus method run.

@SneakyThrows
public void run() {
    while (true) {
        Payload request = _entityManagerQueue.take();
        TransactionContext transactionContext = _hazelcastInstance.newTransactionContext();
        transactionContext.beginTransaction();
        try {
            TransactionalMap<EntityKey, Entity> entityTransactionalMap = transactionContext.getMap("entityMap");
            TransactionalMap<EntityRelationKey, EntityRelation> transactionalEntityRelationMap = transactionContext.getMap("entityRelationMap");
            TransactionalMultiMap<DocumentKey, String> documentEntityMap = transactionContext.getMultiMap("documentEntityMap");
            EntityManagerRequest payload = request.getPayload();
            _loggerAggregator.emitLog("process ingestionId", payload.getIngestionId());
            long tenantId = payload.getTenantId();
            String ingestionId = payload.getIngestionId();
            List<EntityRequest> entities = request.getEntities();
            Map<EntityKey, Entity> localEntityMap = new HashMap<>(entities.size());
            for (EntityRequest entityRequest : entities) {
                String name = entityRequest.getName();
                String type = entityRequest.getType();
                String cacheId = Long.toString(_entityFlakeId.newId());
                EntityKey key = EntityKey.of(tenantId, name, type, cacheId, ingestionId);
                Entity entity = new Entity(null, cacheId, tenantId, name, type, null, ingestionId, false, true, entityRequest.getContext());
                entityTransactionalMap.set(key, entity);
                localEntityMap.put(key, entity);
                for (EntityRequest entityRequest2 : entities) {
                    for (RelationRequest relation : entityRequest2.getRelations()) {
                        if (relation.getTo().equals(entityRequest.getTmpId())) {
                            relation.setTo(entity.getCacheId());
                        }
                    }
                }
            }
            for (EntityRequest entity : entities) {
                List<RelationRequest> relations = entity.getRelations();
                if (relations == null || relations.isEmpty()) {
                    continue;
                }
                Collection<Entity> values = localEntityMap.values();
                Entity current = values.stream().filter(e -> e.getName().equals(entity.getName()) && e.getType().equals(entity.getType())).findFirst().orElse(null);
                if (current == null) {
                    continue;
                }
                for (RelationRequest relation : relations) {
                    String to = relation.getTo();
                    String name = relation.getName();
                    for (Entity value : values) {
                        if (value.getCacheId().equals(to)) {
                            long entityRelationId = _entityRelationFlakeId.newId();
                            EntityRelation entityRelation = new EntityRelation(entityRelationId, current.getCacheId(), ingestionId, name, value.getCacheId());
                            transactionalEntityRelationMap.set(EntityRelationKey.of(entityRelationId, current.getCacheId(), ingestionId), entityRelation);
                        }
                    }
                }
            }
            if (!localEntityMap.isEmpty()) {
                DocumentKey key = DocumentKey.of(payload.getDatasourceId(), payload.getContentId(), tenantId);
                for (Entity value : localEntityMap.values()) {
                    documentEntityMap.put(key, value.getCacheId());
                }
            }
        } catch (Exception e) {
            _log.error(e.getMessage(), e);
            transactionContext.rollbackTransaction();
        } finally {
            transactionContext.commitTransaction();
        }
    }
}
Also used : Entity(io.openk9.entity.manager.cache.model.Entity) HashMap(java.util.HashMap) RelationRequest(io.openk9.entity.manager.dto.RelationRequest) EntityRelationKey(io.openk9.entity.manager.cache.model.EntityRelationKey) EntityKey(io.openk9.entity.manager.cache.model.EntityKey) EntityRelation(io.openk9.entity.manager.cache.model.EntityRelation) EntityRequest(io.openk9.entity.manager.dto.EntityRequest) TransactionContext(com.hazelcast.transaction.TransactionContext) DocumentKey(io.openk9.entity.manager.cache.model.DocumentKey) Payload(io.openk9.entity.manager.dto.Payload) EntityManagerRequest(io.openk9.entity.manager.dto.EntityManagerRequest) SneakyThrows(lombok.SneakyThrows)

Example 2 with EntityKey

use of io.openk9.entity.manager.cache.model.EntityKey in project openk9 by smclab.

the class CreateDocumentEntitiesRunnable method run_.

@Override
public void run_() {
    _log.info("start CreateEntitiesRunnable");
    IMap<EntityKey, Entity> entityIMap = MapUtil.getEntityMap(_hazelcastInstance);
    MultiMap<DocumentKey, String> documentEntityMapMap = MapUtil.getDocumentEntityMapMap(_hazelcastInstance);
    EntityGraphService entityGraphService = CDI.current().select(EntityGraphService.class).get();
    for (DocumentKey documentKey : documentEntityMapMap.localKeySet()) {
        Collection<String> entityCacheIds = documentEntityMapMap.get(documentKey);
        String[] arr = entityCacheIds.toArray(String[]::new);
        Collection<Object> project = entityIMap.project(Projections.singleAttribute("indexable"), Predicates.and(Predicates.in("cacheId", arr), Predicates.notEqual("id", null), Predicates.notEqual("graphId", null)));
        if (project.size() == entityCacheIds.size()) {
            Collection<Entity> entities = entityIMap.values(Predicates.in("cacheId", arr));
            DocumentGraph document = entityGraphService.insertDocument(DocumentGraph.of(null, documentKey.getDatasourceId(), documentKey.getTenantId(), documentKey.getContentId()));
            for (Entity entity : entities) {
                entityGraphService.createDocumentRelationship(entity.getId(), document.getId(), "related_to");
            }
            documentEntityMapMap.remove(documentKey);
        }
    }
}
Also used : Entity(io.openk9.entity.manager.cache.model.Entity) EntityGraphService(io.openk9.entity.manager.service.graph.EntityGraphService) EntityKey(io.openk9.entity.manager.cache.model.EntityKey) DocumentKey(io.openk9.entity.manager.cache.model.DocumentKey) DocumentGraph(io.openk9.entity.manager.model.graph.DocumentGraph)

Example 3 with EntityKey

use of io.openk9.entity.manager.cache.model.EntityKey in project openk9 by smclab.

the class CreateEntitiesRunnable method run_.

@Override
public void run_() {
    _log.info("start CreateEntitiesRunnable");
    IMap<EntityKey, Entity> entityIMap = MapUtil.getEntityMap(_hazelcastInstance);
    IMap<AssociableEntityKey, Entity> associableEntityMap = MapUtil.getAssociableEntityMap(_hazelcastInstance);
    Set<EntityKey> entityKeys = entityIMap.localKeySet(Predicates.and(Predicates.equal("id", null), Predicates.equal("graphId", null)));
    EntityGraphConfig config = CDI.current().select(EntityGraphConfig.class).get();
    EntityNameCleanerProvider entityNameCleanerProvider = CDI.current().select(EntityNameCleanerProvider.class).get();
    EntityService entityService = CDI.current().select(EntityService.class).get();
    EntityGraphService entityGraphService = CDI.current().select(EntityGraphService.class).get();
    Map<EntityKey, Entity> localEntityMap = entityIMap.getAll(entityKeys);
    Collection<Entity> localEntityValues = localEntityMap.values();
    Set<EntityKey> localEntityKeys = localEntityMap.keySet();
    List<Member> collect = _hazelcastInstance.getCluster().getMembers().stream().filter(member -> !member.localMember()).collect(Collectors.toList());
    String[] ingestionIds = localEntityKeys.stream().map(EntityKey::getIngestionId).distinct().toArray(String[]::new);
    IExecutorService entityExecutor = _hazelcastInstance.getExecutorService("entityExecutor");
    Map<Member, Future<Map<EntityKey, Entity>>> memberFutureMap = entityExecutor.submitToMembers(new GetEntitiesCallable(ingestionIds), collect);
    Map<EntityKey, Entity> otherEntityKeyEntityMap = memberFutureMap.values().stream().map(FutureUtil::makeCompletableFuture).map(CompletableFuture::join).reduce((a, b) -> {
        Map<EntityKey, Entity> map = new HashMap<>();
        map.putAll(a);
        map.putAll(b);
        return map;
    }).orElseGet(Map::of);
    Stream<EntityMember> otherEntityMemberStream = otherEntityKeyEntityMap.values().stream().map(entity -> EntityMember.of(entity, false));
    Stream<EntityMember> localEntityMemberStream = localEntityValues.stream().map(entity -> EntityMember.of(entity, true));
    Map<String, List<EntityMember>> entitiesGroupingByIngestionId = Stream.concat(localEntityMemberStream, otherEntityMemberStream).collect(Collectors.groupingBy(entityMember -> entityMember.getEntity().getIngestionId()));
    Collection<List<EntityMember>> values = entitiesGroupingByIngestionId.values();
    Map<EntityKey, Entity> entityMap = new HashMap<>();
    for (List<EntityMember> ingestionIdEntities : values) {
        Map<AssociableEntityKey, Entity> localAssociableEntityMap = new HashMap<>();
        List<EntityCandidates> entityCandidateList = new ArrayList<>();
        for (EntityMember ingestionIdEntity : ingestionIdEntities) {
            Entity innerEntity = ingestionIdEntity.getEntity();
            entityCandidateList.add(getEntityCandidates(entityNameCleanerProvider, entityService, ingestionIdEntity, innerEntity));
        }
        List<Mono<Entity>> completableFutureList = entityCandidateList.stream().filter(entityCandidates -> entityCandidates.getEntity().isLocal()).map(entityCandidates -> Mono.fromSupplier(_getAndCreateEntityDisambiguate(config, entityNameCleanerProvider, entityService, entityGraphService, entityCandidateList, entityCandidates, entityCandidates.getEntity())).subscribeOn(Schedulers.boundedElastic())).collect(Collectors.toList());
        Mono<List<Entity>> zip = Mono.zip(completableFutureList, a -> {
            List<Entity> entities = new ArrayList<>();
            for (Object o : a) {
                entities.add((Entity) o);
            }
            return entities;
        }).defaultIfEmpty(List.of());
        for (Entity currentEntityRequest : zip.block()) {
            localAssociableEntityMap.put(AssociableEntityKey.of(currentEntityRequest.getCacheId(), currentEntityRequest.getIngestionId()), currentEntityRequest);
            entityMap.put(EntityKey.of(currentEntityRequest.getTenantId(), currentEntityRequest.getName(), currentEntityRequest.getType(), currentEntityRequest.getCacheId(), currentEntityRequest.getIngestionId()), currentEntityRequest);
        }
        associableEntityMap.setAll(localAssociableEntityMap);
    }
    entityIMap.setAll(entityMap);
}
Also used : Arrays(java.util.Arrays) Member(com.hazelcast.cluster.Member) FutureUtil(io.openk9.entity.manager.util.FutureUtil) EntityKey(io.openk9.entity.manager.cache.model.EntityKey) SymbolicName(org.neo4j.cypherdsl.core.SymbolicName) Future(java.util.concurrent.Future) Map(java.util.Map) ToString(lombok.ToString) Property(org.neo4j.cypherdsl.core.Property) Cypher(org.neo4j.cypherdsl.core.Cypher) EntityIndex(io.openk9.entity.manager.model.index.EntityIndex) EntityGraph(io.openk9.entity.manager.model.graph.EntityGraph) Collection(java.util.Collection) CDI(javax.enterprise.inject.spi.CDI) Set(java.util.Set) Collectors(java.util.stream.Collectors) Serializable(java.io.Serializable) Objects(java.util.Objects) List(java.util.List) Stream(java.util.stream.Stream) Builder(lombok.Builder) Optional(java.util.Optional) Cypher.literalOf(org.neo4j.cypherdsl.core.Cypher.literalOf) HazelcastInstanceAware(com.hazelcast.core.HazelcastInstanceAware) AssociableEntityKey(io.openk9.entity.manager.cache.model.AssociableEntityKey) EntityGraphConfig(io.openk9.entity.manager.config.EntityGraphConfig) Logger(org.jboss.logging.Logger) HashMap(java.util.HashMap) CompletableFuture(java.util.concurrent.CompletableFuture) Supplier(java.util.function.Supplier) ArrayList(java.util.ArrayList) Statement(org.neo4j.cypherdsl.core.Statement) Schedulers(reactor.core.scheduler.Schedulers) EntityNameCleanerProvider(io.openk9.entity.manager.cleaner.EntityNameCleanerProvider) Node(org.neo4j.cypherdsl.core.Node) HazelcastInstance(com.hazelcast.core.HazelcastInstance) QueryBuilder(org.elasticsearch.index.query.QueryBuilder) EntityNameCleaner(io.openk9.entity.manager.cleaner.EntityNameCleaner) Entity(io.openk9.entity.manager.cache.model.Entity) Functions(org.neo4j.cypherdsl.core.Functions) Mono(reactor.core.publisher.Mono) IExecutorService(com.hazelcast.core.IExecutorService) GetEntitiesCallable(io.openk9.entity.manager.action.GetEntitiesCallable) AliasedExpression(org.neo4j.cypherdsl.core.AliasedExpression) Predicates(com.hazelcast.query.Predicates) EntityService(io.openk9.entity.manager.service.index.EntityService) Data(lombok.Data) AllArgsConstructor(lombok.AllArgsConstructor) EntityGraphService(io.openk9.entity.manager.service.graph.EntityGraphService) Collections(java.util.Collections) MapUtil(io.openk9.entity.manager.util.MapUtil) IMap(com.hazelcast.map.IMap) Entity(io.openk9.entity.manager.cache.model.Entity) EntityGraphService(io.openk9.entity.manager.service.graph.EntityGraphService) HashMap(java.util.HashMap) ArrayList(java.util.ArrayList) ToString(lombok.ToString) IExecutorService(com.hazelcast.core.IExecutorService) EntityKey(io.openk9.entity.manager.cache.model.EntityKey) AssociableEntityKey(io.openk9.entity.manager.cache.model.AssociableEntityKey) CompletableFuture(java.util.concurrent.CompletableFuture) GetEntitiesCallable(io.openk9.entity.manager.action.GetEntitiesCallable) List(java.util.List) ArrayList(java.util.ArrayList) EntityGraphConfig(io.openk9.entity.manager.config.EntityGraphConfig) Member(com.hazelcast.cluster.Member) EntityService(io.openk9.entity.manager.service.index.EntityService) Mono(reactor.core.publisher.Mono) EntityNameCleanerProvider(io.openk9.entity.manager.cleaner.EntityNameCleanerProvider) Future(java.util.concurrent.Future) CompletableFuture(java.util.concurrent.CompletableFuture) AssociableEntityKey(io.openk9.entity.manager.cache.model.AssociableEntityKey) Map(java.util.Map) HashMap(java.util.HashMap) IMap(com.hazelcast.map.IMap)

Example 4 with EntityKey

use of io.openk9.entity.manager.cache.model.EntityKey in project openk9 by smclab.

the class CreateRelationRunnable method run_.

@Override
public void run_() {
    IMap<EntityKey, Entity> entityIMap = MapUtil.getEntityMap(_hazelcastInstance);
    Set<EntityKey> entityKeys = entityIMap.localKeySet(Predicates.and(Predicates.notEqual("id", null), Predicates.notEqual("graphId", null)));
    _log.info("entityKeys: " + entityKeys.size());
    Map<EntityKey, Entity> entityIMapAll = entityIMap.getAll(entityKeys);
    _log.info("entityIMapAll: " + entityIMapAll.size());
    IMap<EntityRelationKey, EntityRelation> entityRelationMap = MapUtil.getEntityRelationMap(_hazelcastInstance);
    Map<String, String> collect = entityIMapAll.values().stream().collect(Collectors.toMap(Entity::getCacheId, Entity::getId));
    String[] cacheIds = entityKeys.stream().map(EntityKey::getCacheId).distinct().toArray(String[]::new);
    Map<EntityRelationKey, EntityRelation> entries = entityRelationMap.getAll(entityRelationMap.keySet(Predicates.in("__key.entityId", cacheIds)));
    _log.info("entityRelations: " + entries.size());
    EntityGraphService entityGraphService = CDI.current().select(EntityGraphService.class).get();
    List<EntityRelationKey> entityRelationKeysToDelete = new ArrayList<>();
    for (Map.Entry<EntityRelationKey, EntityRelation> entry : entries.entrySet()) {
        EntityRelationKey key = entry.getKey();
        EntityRelation value = entry.getValue();
        String from = collect.get(value.getEntityCacheId());
        String to = collect.get(value.getTo());
        if (from != null && to != null) {
            try {
                entityGraphService.createRelationship(from, to, value.getName());
                entityRelationKeysToDelete.add(key);
            } catch (Exception e) {
                _log.error(e.getMessage(), e);
            }
        }
    }
    try {
        Pipelining pipelining = new Pipelining<>(10);
        for (EntityRelationKey entityRelationKey : entityRelationKeysToDelete) {
            pipelining.add(entityRelationMap.removeAsync(entityRelationKey));
        }
        pipelining.results();
    } catch (Exception e) {
        _log.error(e.getMessage(), e);
    }
}
Also used : Entity(io.openk9.entity.manager.cache.model.Entity) EntityGraphService(io.openk9.entity.manager.service.graph.EntityGraphService) ArrayList(java.util.ArrayList) EntityRelationKey(io.openk9.entity.manager.cache.model.EntityRelationKey) EntityKey(io.openk9.entity.manager.cache.model.EntityKey) EntityRelation(io.openk9.entity.manager.cache.model.EntityRelation) Pipelining(com.hazelcast.core.Pipelining) Map(java.util.Map) IMap(com.hazelcast.map.IMap)

Aggregations

Entity (io.openk9.entity.manager.cache.model.Entity)4 EntityKey (io.openk9.entity.manager.cache.model.EntityKey)4 EntityGraphService (io.openk9.entity.manager.service.graph.EntityGraphService)3 IMap (com.hazelcast.map.IMap)2 DocumentKey (io.openk9.entity.manager.cache.model.DocumentKey)2 EntityRelation (io.openk9.entity.manager.cache.model.EntityRelation)2 EntityRelationKey (io.openk9.entity.manager.cache.model.EntityRelationKey)2 ArrayList (java.util.ArrayList)2 HashMap (java.util.HashMap)2 Map (java.util.Map)2 Member (com.hazelcast.cluster.Member)1 HazelcastInstance (com.hazelcast.core.HazelcastInstance)1 HazelcastInstanceAware (com.hazelcast.core.HazelcastInstanceAware)1 IExecutorService (com.hazelcast.core.IExecutorService)1 Pipelining (com.hazelcast.core.Pipelining)1 Predicates (com.hazelcast.query.Predicates)1 TransactionContext (com.hazelcast.transaction.TransactionContext)1 GetEntitiesCallable (io.openk9.entity.manager.action.GetEntitiesCallable)1 AssociableEntityKey (io.openk9.entity.manager.cache.model.AssociableEntityKey)1 EntityNameCleaner (io.openk9.entity.manager.cleaner.EntityNameCleaner)1