g.dugas/semrack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add fields API
Some checks failed
Maven build / build (push) Failing after 1m14s
Details

Browse Source
This commit is contained in:
Guillaume Dugas
2026-02-17 16:19:33 +01:00
parent def2e4140f
commit cdbe2dfd67
57 changed files with 537 additions and 497 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
modules/semrack-api-rest/runtime/src/main/java/fr/codeanddata/semrack/api/rest/SemrackApi.java
View File

@@ -1,50 +1,54 @@
package fr.codeanddata.semrack.api.rest;
import fr.codeanddata.semrack.core.models.PushDocumentRequest;
import fr.codeanddata.semrack.core.models.SearchRequest;
import fr.codeanddata.semrack.core.models.SearchResult;
import fr.codeanddata.semrack.core.models.SemrackDocument;
import fr.codeanddata.semrack.core.repositories.SemdocRepository;
import fr.codeanddata.semrack.core.models.*;
import fr.codeanddata.semrack.core.repositories.DocumentRepository;
import io.smallrye.mutiny.Uni;
import jakarta.inject.Inject;
import jakarta.ws.rs.GET;
import jakarta.ws.rs.POST;
import jakarta.ws.rs.PUT;
import jakarta.ws.rs.Path;
import jakarta.ws.rs.Produces;
import jakarta.ws.rs.core.MediaType;
import org.jboss.resteasy.reactive.RestPath;
@Path("/semrack/documents")
//@Authenticated
public class SemrackApi {
@Inject
SemdocRepository repository;
DocumentRepository repository;
@POST
@Produces(MediaType.APPLICATION_JSON)
public Uni<SemrackDocument> pushDocument(PushDocumentRequest semrackDocument) {
public Uni<Document> pushDocument(PushDocument semrackDocument) {
return repository.pushDocument(semrackDocument);
}
@POST
@Path("/search")
@Produces(MediaType.APPLICATION_JSON)
public Uni<SearchResult> searchDocument(SearchRequest query) {
public Uni<SearchResult> searchDocument(Search query) {
return repository.searchDocument(query);
}
@GET
@Path("{uid}")
@Produces(MediaType.APPLICATION_JSON)
public Uni<SemrackDocument> getDocument(@RestPath String uid) {
public Uni<Document> getDocument(@RestPath String uid) {
return repository.get(uid);
}
@POST
@Path("{uid}")
@Produces(MediaType.APPLICATION_JSON)
public Uni<SemrackDocument> updateDocument(@RestPath String uid, PushDocumentRequest semrackDocument) {
public Uni<Document> getDocument(@RestPath String uid, StorageGet request) {
return repository.get(uid, request);
}
@POST
@Path("{uid}/update")
@Produces(MediaType.APPLICATION_JSON)
public Uni<Document> updateDocument(@RestPath String uid, PushDocument semrackDocument) {
return repository.pushDocument(semrackDocument);
}
}

13
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/Index.java Normal file
View File

@@ -0,0 +1,13 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.IndexSearchResult;
import fr.codeanddata.semrack.core.models.Search;
import io.smallrye.mutiny.Uni;
public interface Index {
Uni<Long> count(Search request);
Uni<Boolean> exist(Search query);
Uni<Void> index(String documentId);
Uni<IndexSearchResult> search(Search search);
Uni<Void> clear(String documentId);
}

2
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/SemrackLookupExpression.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/LookupExpression.java
View File

@@ -4,7 +4,7 @@ import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import jakarta.enterprise.inject.spi.CDI;
public interface SemrackLookupExpression<T> {
public interface LookupExpression<T> {
String apply(Object params);
default T convert(Object params) {

8
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/ReadInterceptor.java Normal file
View File

@@ -0,0 +1,8 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.ReadContext;
import io.smallrye.mutiny.Uni;
public interface ReadInterceptor {
Uni<Void> interceptSemdocRead(ReadContext context);
}

8
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/SearchInterceptor.java Normal file
View File

@@ -0,0 +1,8 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.SearchContext;
import io.smallrye.mutiny.Uni;
public interface SearchInterceptor {
Uni<Void> interceptSemdocSearch(SearchContext context);
}

13
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/SemdocIndex.java
View File

@@ -1,13 +0,0 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.IndexSearchResult;
import fr.codeanddata.semrack.core.models.SearchRequest;
import io.smallrye.mutiny.Uni;
public interface SemdocIndex {
Uni<Long> count(SearchRequest request);
Uni<Boolean> exist(SearchRequest query);
Uni<Void> index(String documentId);
Uni<IndexSearchResult> search(SearchRequest searchRequest);
Uni<Void> clear(String documentId);
}

8
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/SemdocReadInterceptor.java
View File

@@ -1,8 +0,0 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.SemdocReadContext;
import io.smallrye.mutiny.Uni;
public interface SemdocReadInterceptor {
Uni<Void> interceptSemdocRead(SemdocReadContext context);
}

8
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/SemdocSearchInterceptor.java
View File

@@ -1,8 +0,0 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.SemdocSearchContext;
import io.smallrye.mutiny.Uni;
public interface SemdocSearchInterceptor {
Uni<Void> interceptSemdocSearch(SemdocSearchContext context);
}

12
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/SemdocStorage.java
View File

@@ -1,12 +0,0 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.SemrackDocument;
import io.smallrye.mutiny.Uni;
import java.util.List;
public interface SemdocStorage {
Uni<SemrackDocument> get(String uid);
Uni<List<SemrackDocument>> get(List<String> uids);
Uni<SemrackDocument> storeDocument(SemrackDocument document);
}

8
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/SemdocWriteInterceptor.java
View File

@@ -1,8 +0,0 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.SemdocWriteContext;
import io.smallrye.mutiny.Uni;
public interface SemdocWriteInterceptor {
Uni<Void> interceptSemdocWrite(SemdocWriteContext context);
}

13
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/Storage.java Normal file
View File

@@ -0,0 +1,13 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.Document;
import fr.codeanddata.semrack.core.models.StorageGet;
import io.smallrye.mutiny.Uni;
import java.util.List;
public interface Storage {
Uni<Document> get(String uid, StorageGet request);
Uni<List<Document>> get(List<String> uids, StorageGet request);
Uni<Document> storeDocument(Document document);
}

8
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/WriteInterceptor.java Normal file
View File

@@ -0,0 +1,8 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.WriteContext;
import io.smallrye.mutiny.Uni;
public interface WriteInterceptor {
Uni<Void> interceptSemdocWrite(WriteContext context);
}

2
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/enums/SemrackSortDirection.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/enums/SortDirection.java
View File

@@ -1,5 +1,5 @@
package fr.codeanddata.semrack.core.enums;
public enum SemrackSortDirection {
public enum SortDirection {
asc, desc
}

8
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/interceptors/AnnotateWriteInterceptor.java
View File

@@ -2,8 +2,8 @@ package fr.codeanddata.semrack.core.interceptors;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import fr.codeanddata.semrack.core.SemdocWriteInterceptor;
import fr.codeanddata.semrack.core.models.SemdocWriteContext;
import fr.codeanddata.semrack.core.WriteInterceptor;
import fr.codeanddata.semrack.core.models.WriteContext;
import io.smallrye.mutiny.Uni;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
@@ -11,7 +11,7 @@ import jakarta.inject.Inject;
import java.util.Map;
@ApplicationScoped
public class AnnotateWriteInterceptor implements SemdocWriteInterceptor {
public class AnnotateWriteInterceptor implements WriteInterceptor {
public static final String ANNOTATE_KEY = "annotate";
public static final String CUSTOM_PREFIX = "custom/";
@@ -20,7 +20,7 @@ public class AnnotateWriteInterceptor implements SemdocWriteInterceptor {
ObjectMapper objectMapper;
@Override
public Uni<Void> interceptSemdocWrite(SemdocWriteContext context) {
public Uni<Void> interceptSemdocWrite(WriteContext context) {
final Map<String, Object> directives = context.getDirectives();
if (directives.containsKey(ANNOTATE_KEY)) {

18
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/interceptors/PublishWriteInterceptor.java
View File

@@ -1,8 +1,8 @@
package fr.codeanddata.semrack.core.interceptors;
import fr.codeanddata.semrack.core.SemdocWriteInterceptor;
import fr.codeanddata.semrack.core.models.SemdocWriteContext;
import fr.codeanddata.semrack.core.models.SemrackDocument;
import fr.codeanddata.semrack.core.WriteInterceptor;
import fr.codeanddata.semrack.core.models.WriteContext;
import fr.codeanddata.semrack.core.models.Document;
import fr.codeanddata.semrack.core.utils.UIDGenerator;
import io.smallrye.mutiny.Uni;
import jakarta.enterprise.context.ApplicationScoped;
@@ -12,7 +12,7 @@ import java.util.Map;
import java.util.Optional;
@ApplicationScoped
public class PublishWriteInterceptor implements SemdocWriteInterceptor {
public class PublishWriteInterceptor implements WriteInterceptor {
final String PUBLICATION_PREFIX = "publication/";
@@ -20,18 +20,18 @@ public class PublishWriteInterceptor implements SemdocWriteInterceptor {
UIDGenerator uidGenerator;
@Override
public Uni<Void> interceptSemdocWrite(SemdocWriteContext context) {
public Uni<Void> interceptSemdocWrite(WriteContext context) {
final Map<String, Object> directives = context.getDirectives();
final SemrackDocument current = context.getCurrentDocument();
final SemrackDocument next = context.getNextDocument();
final Document current = context.getCurrentDocument();
final Document next = context.getNextDocument();
int version = (int) Optional.ofNullable(current)
.map(SemrackDocument::getAnnotations)
.map(Document::getAnnotations)
.map(annotations -> annotations.getOrDefault(PUBLICATION_PREFIX + "version", 0))
.orElse(0);
next.getAnnotations().put(PUBLICATION_PREFIX + "reference", Optional.ofNullable(current)
.map(SemrackDocument::getAnnotations)
.map(Document::getAnnotations)
.map(annotations -> annotations.getOrDefault(PUBLICATION_PREFIX + "reference", next.getUid()))
.orElse(next.getUid()));

8
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/mappers/SemrackDocumentMapper.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/mappers/DocumentMapper.java
View File

@@ -1,14 +1,14 @@
package fr.codeanddata.semrack.core.mappers;
import fr.codeanddata.semrack.core.models.SemrackDocument;
import fr.codeanddata.semrack.core.models.PushDocumentRequest;
import fr.codeanddata.semrack.core.models.Document;
import fr.codeanddata.semrack.core.models.PushDocument;
import org.mapstruct.Mapper;
import org.mapstruct.Mapping;
import org.mapstruct.MappingConstants;
@Mapper(componentModel = MappingConstants.ComponentModel.CDI)
public abstract class SemrackDocumentMapper {
public abstract class DocumentMapper {
@Mapping(source = "uid", target = "uid")
@Mapping(source = "metadata", target = "metadata")
abstract SemrackDocument toDocument(PushDocumentRequest document);
abstract Document toDocument(PushDocument document);
}

6
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/SemrackDocument.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/Document.java
View File

@@ -11,7 +11,7 @@ import java.util.Map;
@AllArgsConstructor
@NoArgsConstructor
@RegisterForReflection(serialization = true)
public class SemrackDocument {
public class Document {
/**
* L'identifiant unique du document.
@@ -29,4 +29,8 @@ public class SemrackDocument {
*/
Map<String, Object> metadata;
/**
*
*/
Map<String, Object> fields;
}

2
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/SemrackPagination.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/Pagination.java
View File

@@ -7,7 +7,7 @@ import lombok.*;
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class SemrackPagination {
public class Pagination {
Integer page;
Integer size;
}

2
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/PushDocumentRequest.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/PushDocument.java
View File

@@ -9,7 +9,7 @@ import java.util.Map;
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class PushDocumentRequest {
public class PushDocument {
/**
* L'identifiant unique du document.
*/

8
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/SemdocReadContext.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/ReadContext.java
View File

@@ -2,15 +2,11 @@ package fr.codeanddata.semrack.core.models;
import lombok.*;
import java.util.Map;
@Getter
@Setter
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class SemdocReadContext {
SemrackDocument currentDocument;
public class ReadContext {
Document currentDocument;
}

10
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/SearchRequest.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/Search.java
View File

@@ -10,9 +10,11 @@ import java.util.Map;
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class SearchRequest {
public class Search {
Map<String, Object> filter;
List<SemrackSort> sort;
SemrackPagination paginate;
List<String> fields;
List<Sort> sort;
Pagination paginate;
Boolean annotations;
Boolean metadata;
Map<String, String> fields;
}

2
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/SemdocSearchContext.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/SearchContext.java
View File

@@ -9,7 +9,7 @@ import java.util.Map;
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class SemdocSearchContext {
public class SearchContext {
Map<String, Object> search;
}

2
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/SearchResult.java
View File

@@ -10,6 +10,6 @@ import java.util.List;
@AllArgsConstructor
@NoArgsConstructor
public class SearchResult {
List<SemrackDocument> documents;
List<Document> documents;
PaginationInfo pagination;
}

6
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/SemrackSort.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/Sort.java
View File

@@ -1,6 +1,6 @@
package fr.codeanddata.semrack.core.models;
import fr.codeanddata.semrack.core.enums.SemrackSortDirection;
import fr.codeanddata.semrack.core.enums.SortDirection;
import lombok.*;
@Getter
@@ -8,7 +8,7 @@ import lombok.*;
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class SemrackSort {
public class Sort {
String field;
SemrackSortDirection direction;
SortDirection direction;
}

16
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/StorageGet.java Normal file
View File

@@ -0,0 +1,16 @@
package fr.codeanddata.semrack.core.models;
import lombok.*;
import java.util.Map;
@Getter
@Setter
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class StorageGet {
Boolean metadataSource;
Boolean annotationsSource;
Map<String, String> fields;
}

6
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/SemdocWriteContext.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/models/WriteContext.java
View File

@@ -9,19 +9,19 @@ import java.util.Map;
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class SemdocWriteContext {
public class WriteContext {
/**
* Previous stored document
*/
SemrackDocument currentDocument;
Document currentDocument;
/**
* Document to be store. It initialized with :
* - the currentDocument uid if exists, or a new one
* - the metadata to be persisted
*/
SemrackDocument nextDocument;
Document nextDocument;
/**
* The directives to be applied

36
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/repositories/SemdocRepository.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/repositories/DocumentRepository.java
View File

@@ -1,9 +1,9 @@
package fr.codeanddata.semrack.core.repositories;
import fr.codeanddata.semrack.core.SemdocWriteInterceptor;
import fr.codeanddata.semrack.core.WriteInterceptor;
import fr.codeanddata.semrack.core.exceptions.SemrackRuntimeException;
import fr.codeanddata.semrack.core.models.*;
import fr.codeanddata.semrack.core.storages.SemdocStorageProxy;
import fr.codeanddata.semrack.core.storages.StorageProxy;
import fr.codeanddata.semrack.core.utils.UIDGenerator;
import io.smallrye.mutiny.Uni;
import jakarta.enterprise.context.ApplicationScoped;
@@ -16,35 +16,43 @@ import java.util.Optional;
import java.util.concurrent.atomic.AtomicReference;
@ApplicationScoped
public class SemdocRepository {
public class DocumentRepository {
@Inject
SemdocStorageProxy semdocStorage;
StorageProxy semdocStorage;
@Inject
@Any
Instance<SemdocWriteInterceptor> writeInterceptors;
Instance<WriteInterceptor> writeInterceptors;
@Inject
UIDGenerator uidGenerator;
public Uni<SearchResult> searchDocument(SearchRequest query) {
public Uni<SearchResult> searchDocument(Search query) {
return semdocStorage.searchDocument(query);
}
public Uni<SemrackDocument> get(String documentId) {
return semdocStorage.readDocument(documentId);
public Uni<Document> get(String documentId) {
return get(documentId, StorageGet.builder().build());
}
public Uni<SemrackDocument> pushDocument(PushDocumentRequest pushDocument) {
return Optional.ofNullable(pushDocument.getUid()).map(semdocStorage::readDocument).orElse(Uni.createFrom().nullItem())
public Uni<Document> get(String documentId, StorageGet request) {
return semdocStorage.readDocument(documentId, request);
}
public Uni<Document> pushDocument(PushDocument pushDocument) {
return Optional.ofNullable(pushDocument.getUid()).map(uid -> semdocStorage.readDocument(uid, StorageGet.builder()
.annotationsSource(true)
.metadataSource(true)
.build()))
.orElse(Uni.createFrom().nullItem())
.chain(currentDocument -> {
final SemdocWriteContext context = SemdocWriteContext.builder()
final WriteContext context = WriteContext.builder()
.directives(pushDocument.getDirectives())
.currentDocument(currentDocument)
.nextDocument(SemrackDocument.builder()
.uid(Optional.ofNullable(currentDocument).map(SemrackDocument::getUid).orElse(uidGenerator.apply(null)))
.nextDocument(Document.builder()
.uid(Optional.ofNullable(currentDocument).map(Document::getUid).orElse(uidGenerator.apply(null)))
.metadata(pushDocument.getMetadata())
.annotations(new HashMap<>())
.build())
@@ -57,7 +65,7 @@ public class SemdocRepository {
});
}
public Uni<SemrackDocument> getOrCreate(SearchRequest query, PushDocumentRequest toCreate) {
public Uni<Document> getOrCreate(Search query, PushDocument toCreate) {
return searchDocument(query)
.chain(searchResult -> {
if (searchResult.getDocuments().isEmpty()) {

35
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/services/IndexService.java Normal file
View File

@@ -0,0 +1,35 @@
package fr.codeanddata.semrack.core.services;
import fr.codeanddata.semrack.core.Index;
import fr.codeanddata.semrack.core.models.IndexSearchResult;
import fr.codeanddata.semrack.core.models.Search;
import io.smallrye.mutiny.Uni;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
@ApplicationScoped
public class IndexService {
@Inject
Index index;
public Uni<Void> index(String documentId) {
return index.index(documentId);
}
public Uni<IndexSearchResult> search(Search search) {
return index.search(search);
}
public Uni<Long> count(Search request) {
return index.count(request);
}
public Uni<Boolean> exist(Search query) {
return index.exist(query);
}
public Uni<Void> clear(String documentId) {
return index.clear(documentId);
}
}

10
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/services/SemrackLookupService.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/services/LookupService.java
View File

@@ -1,6 +1,6 @@
package fr.codeanddata.semrack.core.services;
import fr.codeanddata.semrack.core.SemrackLookupExpression;
import fr.codeanddata.semrack.core.LookupExpression;
import io.quarkus.runtime.StartupEvent;
import io.smallrye.common.annotation.Identifier;
import jakarta.enterprise.context.ApplicationScoped;
@@ -14,13 +14,13 @@ import java.util.Map;
import java.util.Set;
@ApplicationScoped
public class SemrackLookupService {
public class LookupService {
@Inject
@Any
Instance<SemrackLookupExpression<?>> operators;
Instance<LookupExpression<?>> operators;
final Map<String, SemrackLookupExpression<?>> operatorsIndex = new HashMap<>();
final Map<String, LookupExpression<?>> operatorsIndex = new HashMap<>();
void startup(@Observes StartupEvent event) {
operators.stream()
@@ -42,7 +42,7 @@ public class SemrackLookupService {
if (! operatorsIndex.containsKey(lookupKey)) {
throw new RuntimeException("Unknown lookup expression '" + lookupKey + "'");
} else {
final SemrackLookupExpression<?> lookup = operatorsIndex.get(lookupKey);
final LookupExpression<?> lookup = operatorsIndex.get(lookupKey);
final Object lookupParams = lookupExpression.get(lookupKey);
return lookup.apply(lookupParams);
}

35
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/services/SemrackIndexService.java
View File

@@ -1,35 +0,0 @@
package fr.codeanddata.semrack.core.services;
import fr.codeanddata.semrack.core.SemdocIndex;
import fr.codeanddata.semrack.core.models.IndexSearchResult;
import fr.codeanddata.semrack.core.models.SearchRequest;
import io.smallrye.mutiny.Uni;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
@ApplicationScoped
public class SemrackIndexService {
@Inject
SemdocIndex semdocIndex;
public Uni<Void> index(String documentId) {
return semdocIndex.index(documentId);
}
public Uni<IndexSearchResult> search(SearchRequest searchRequest) {
return semdocIndex.search(searchRequest);
}
public Uni<Long> count(SearchRequest request) {
return semdocIndex.count(request);
}
public Uni<Boolean> exist(SearchRequest query) {
return semdocIndex.exist(query);
}
public Uni<Void> clear(String documentId) {
return semdocIndex.clear(documentId);
}
}

46
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/storages/SemdocStorageProxy.java
View File

@@ -1,46 +0,0 @@
package fr.codeanddata.semrack.core.storages;
import fr.codeanddata.semrack.core.SemdocIndex;
import fr.codeanddata.semrack.core.SemdocStorage;
import fr.codeanddata.semrack.core.models.SearchRequest;
import fr.codeanddata.semrack.core.models.SearchResult;
import fr.codeanddata.semrack.core.models.SemrackDocument;
import io.smallrye.mutiny.Uni;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
@ApplicationScoped
public class SemdocStorageProxy {
@Inject
SemdocStorage storage;
@Inject
SemdocIndex index;
public Uni<SemrackDocument> readDocument(String uid) {
return storage.get(uid);
}
public Uni<SearchResult> searchDocument(SearchRequest query) {
return index.search(query)
.chain(searchResult -> storage.get(searchResult.getUids())
.map(documents -> SearchResult.builder()
.documents(documents)
.pagination(searchResult.getPagination())
.build()));
}
public Uni<Long> countDocuments(SearchRequest request) {
return index.count(request);
}
public Uni<Boolean> documentsExist(SearchRequest query) {
return index.exist(query);
}
public Uni<SemrackDocument> storeDocument(SemrackDocument document) {
return storage.storeDocument(document)
.call(x -> index.index(x.getUid()));
}
}

55
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/storages/StorageProxy.java Normal file
View File

@@ -0,0 +1,55 @@
package fr.codeanddata.semrack.core.storages;
import fr.codeanddata.semrack.core.Index;
import fr.codeanddata.semrack.core.Storage;
import fr.codeanddata.semrack.core.models.Search;
import fr.codeanddata.semrack.core.models.SearchResult;
import fr.codeanddata.semrack.core.models.Document;
import fr.codeanddata.semrack.core.models.StorageGet;
import io.smallrye.mutiny.Uni;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
@ApplicationScoped
public class StorageProxy {
@Inject
Storage storage;
@Inject
Index index;
public Uni<Document> readDocument(String uid) {
return readDocument(uid, StorageGet.builder().build());
}
public Uni<Document> readDocument(String uid, StorageGet request) {
return storage.get(uid, request);
}
public Uni<SearchResult> searchDocument(Search query) {
return index.search(query)
.chain(searchResult -> storage.get(searchResult.getUids(), StorageGet.builder()
.annotationsSource(query.getAnnotations())
.metadataSource(query.getMetadata())
.fields(query.getFields())
.build())
.map(documents -> SearchResult.builder()
.documents(documents)
.pagination(searchResult.getPagination())
.build()));
}
public Uni<Long> countDocuments(Search request) {
return index.count(request);
}
public Uni<Boolean> documentsExist(Search query) {
return index.exist(query);
}
public Uni<Document> storeDocument(Document document) {
return storage.storeDocument(document)
.call(x -> index.index(x.getUid()));
}
}

2
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/utils/SemdocUtils.java → modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/utils/DocumentUtils.java
View File

@@ -8,7 +8,7 @@ import jakarta.inject.Inject;
import java.util.Map;
@ApplicationScoped
public class SemdocUtils {
public class DocumentUtils {
@Inject
ObjectMapper objectMapper;

6
modules/semrack-core/runtime/src/main/java/fr/codeanddata/semrack/core/utils/UIDGenerator.java
View File

@@ -1,6 +1,6 @@
package fr.codeanddata.semrack.core.utils;
import fr.codeanddata.semrack.core.models.SemrackDocument;
import fr.codeanddata.semrack.core.models.Document;
import jakarta.enterprise.context.ApplicationScoped;
import org.apache.commons.codec.digest.DigestUtils;
@@ -10,10 +10,10 @@ import java.util.UUID;
import java.util.function.Function;
@ApplicationScoped
public class UIDGenerator implements Function<SemrackDocument, String> {
public class UIDGenerator implements Function<Document, String> {
@Override
public String apply(SemrackDocument semrackDocument) {
public String apply(Document document) {
final String uuid = UUID.randomUUID().toString();
final LocalDateTime now = LocalDateTime.now();
final long timestamp = now.toInstant(ZoneOffset.UTC).toEpochMilli();

25
modules/semrack-core/testing/src/main/java/fr/codeanddata/semrack/core/MockProducer.java
View File

@@ -1,8 +1,9 @@
package fr.codeanddata.semrack.core;
import fr.codeanddata.semrack.core.models.IndexSearchResult;
import fr.codeanddata.semrack.core.models.SearchRequest;
import fr.codeanddata.semrack.core.models.SemrackDocument;
import fr.codeanddata.semrack.core.models.Search;
import fr.codeanddata.semrack.core.models.Document;
import fr.codeanddata.semrack.core.models.StorageGet;
import io.quarkus.arc.DefaultBean;
import io.smallrye.mutiny.Uni;
import jakarta.annotation.Priority;
@@ -16,20 +17,20 @@ public class MockProducer {
@Produces
@DefaultBean
@Priority(100)
SemdocStorage produceStorage() {
return new SemdocStorage() {
Storage produceStorage() {
return new Storage() {
@Override
public Uni<SemrackDocument> get(String uid) {
public Uni<Document> get(String uid, StorageGet request) {
return null;
}
@Override
public Uni<List<SemrackDocument>> get(List<String> uids) {
public Uni<List<Document>> get(List<String> uids, StorageGet request) {
return null;
}
@Override
public Uni<SemrackDocument> storeDocument(SemrackDocument document) {
public Uni<Document> storeDocument(Document document) {
return null;
}
};
@@ -38,15 +39,15 @@ public class MockProducer {
@Produces
@DefaultBean
@Priority(100)
SemdocIndex produceIndex() {
return new SemdocIndex() {
Index produceIndex() {
return new Index() {
@Override
public Uni<Long> count(SearchRequest request) {
public Uni<Long> count(Search request) {
return null;
}
@Override
public Uni<Boolean> exist(SearchRequest query) {
public Uni<Boolean> exist(Search query) {
return null;
}
@@ -56,7 +57,7 @@ public class MockProducer {
}
@Override
public Uni<IndexSearchResult> search(SearchRequest searchRequest) {
public Uni<IndexSearchResult> search(Search search) {
return null;
}

7
modules/semrack-index-postgres/integration-tests/src/test/java/fr/codeanddata/semrack/index/postgres/it/SemrackIndexPostgresResourceIT.java
View File

@@ -1,7 +0,0 @@
package fr.codeanddata.semrack.index.postgres.it;
import io.quarkus.test.junit.QuarkusIntegrationTest;
@QuarkusIntegrationTest
public class SemrackIndexPostgresResourceIT extends SemrackIndexPostgresResourceTest {
}

21
modules/semrack-index-postgres/integration-tests/src/test/java/fr/codeanddata/semrack/index/postgres/it/SemrackIndexPostgresResourceTest.java
View File

@@ -1,21 +0,0 @@
package fr.codeanddata.semrack.index.postgres.it;
import static io.restassured.RestAssured.given;
import static org.hamcrest.Matchers.is;
import org.junit.jupiter.api.Test;
import io.quarkus.test.junit.QuarkusTest;
@QuarkusTest
public class SemrackIndexPostgresResourceTest {
@Test
public void testHelloEndpoint() {
given()
.when().get("/semrack-index-postgres")
.then()
.statusCode(200)
.body(is("Hello semrack-index-postgres"));
}
}

28
modules/semrack-index-postgres/runtime/src/main/java/fr/codeanddata/semrack/index/postgres/SemdocJpaIndex.java → modules/semrack-index-postgres/runtime/src/main/java/fr/codeanddata/semrack/index/postgres/JpaIndex.java
View File

@@ -1,9 +1,9 @@
package fr.codeanddata.semrack.index.postgres;
import fr.codeanddata.semrack.core.SemdocIndex;
import fr.codeanddata.semrack.core.SemdocStorage;
import fr.codeanddata.semrack.core.Index;
import fr.codeanddata.semrack.core.Storage;
import fr.codeanddata.semrack.core.models.*;
import fr.codeanddata.semrack.core.services.SemrackLookupService;
import fr.codeanddata.semrack.core.services.LookupService;
import fr.codeanddata.semrack.core.utils.Traverser;
import fr.codeanddata.semrack.index.postgres.dtos.IndexEntry;
import fr.codeanddata.semrack.index.postgres.entities.SemrackIndexEntity;
@@ -19,23 +19,23 @@ import org.hibernate.query.Page;
import java.util.*;
@ApplicationScoped
public class SemdocJpaIndex implements SemdocIndex, PanacheRepository<SemrackIndexEntity> {
public class JpaIndex implements Index, PanacheRepository<SemrackIndexEntity> {
@Inject
Traverser traverser;
@Inject
SemrackLookupService lookupService;
LookupService lookupService;
@Inject
SemdocStorage storage;
Storage storage;
@Inject
SemrackIndexRepository indexRepository;
@Override
@WithSession
public Uni<Long> count(SearchRequest request) {
public Uni<Long> count(Search request) {
final StringBuilder query = new StringBuilder("SELECT DISTINCT count(uid) FROM semrack_index");
final String lookup = lookupService.lookup(request.getFilter());
@@ -47,14 +47,16 @@ public class SemdocJpaIndex implements SemdocIndex, PanacheRepository<SemrackInd
}
@Override
public Uni<Boolean> exist(SearchRequest query) {
public Uni<Boolean> exist(Search query) {
return count(query).map(count -> count > 0);
}
@Override
@WithTransaction
public Uni<Void> index(String documentId) {
return storage.get(documentId)
return storage.get(documentId, StorageGet.builder()
.metadataSource(true)
.build())
.call(document -> clear(documentId))
.call(document -> {
final Map<String, IndexEntry> indexes = new HashMap<>();
@@ -78,7 +80,7 @@ public class SemdocJpaIndex implements SemdocIndex, PanacheRepository<SemrackInd
@Override
@WithSession
public Uni<IndexSearchResult> search(SearchRequest request) {
public Uni<IndexSearchResult> search(Search request) {
final IndexSearchResult result = new IndexSearchResult();
final StringBuilder query = new StringBuilder("SELECT DISTINCT uid FROM semrack_index");
@@ -105,11 +107,11 @@ public class SemdocJpaIndex implements SemdocIndex, PanacheRepository<SemrackInd
.replaceWithVoid();
}
Uni<PaginationInfo> queryPaginationInfo(SearchRequest query) {
Uni<PaginationInfo> queryPaginationInfo(Search query) {
return count(query)
.map(count -> PaginationInfo.builder()
.page(Optional.ofNullable(query.getPaginate()).map(SemrackPagination::getPage).orElse(0))
.size(Optional.ofNullable(query.getPaginate()).map(SemrackPagination::getSize).orElse(10))
.page(Optional.ofNullable(query.getPaginate()).map(Pagination::getPage).orElse(0))
.size(Optional.ofNullable(query.getPaginate()).map(Pagination::getSize).orElse(10))
.total(count)
.build());
}

6
modules/semrack-index-postgres/runtime/src/main/java/fr/codeanddata/semrack/index/postgres/operators/AndLookup.java
View File

@@ -1,6 +1,6 @@
package fr.codeanddata.semrack.index.postgres.operators;
import fr.codeanddata.semrack.core.services.SemrackLookupService;
import fr.codeanddata.semrack.core.services.LookupService;
import io.smallrye.common.annotation.Identifier;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
@@ -10,10 +10,10 @@ import java.util.Map;
@Identifier("and")
@ApplicationScoped
public class AndLookup implements SemrackJpaLookupExpression<List<Map<String, Object>>> {
public class AndLookup implements JpaLookupExpression<List<Map<String, Object>>> {
@Inject
SemrackLookupService lookupService;
LookupService lookupService;
@Override
public String apply(Object expressions) {

2
modules/semrack-index-postgres/runtime/src/main/java/fr/codeanddata/semrack/index/postgres/operators/EqualLookup.java
View File

@@ -10,7 +10,7 @@ import jakarta.inject.Inject;
@Identifier("equal")
@ApplicationScoped
public class EqualLookup implements SemrackJpaLookupExpression<EqualLookupParams> {
public class EqualLookup implements JpaLookupExpression<EqualLookupParams> {
@Inject
ObjectMapper objectMapper;

2
modules/semrack-index-postgres/runtime/src/main/java/fr/codeanddata/semrack/index/postgres/operators/InLookup.java
View File

@@ -13,7 +13,7 @@ import java.util.List;
@Identifier("in")
@ApplicationScoped
public class InLookup implements SemrackJpaLookupExpression<InLookupParams> {
public class InLookup implements JpaLookupExpression<InLookupParams> {
@Inject
ObjectMapper objectMapper;

6
modules/semrack-index-postgres/runtime/src/main/java/fr/codeanddata/semrack/index/postgres/operators/JpaLookupExpression.java Normal file
View File

@@ -0,0 +1,6 @@
package fr.codeanddata.semrack.index.postgres.operators;
import fr.codeanddata.semrack.core.LookupExpression;
public interface JpaLookupExpression<T> extends LookupExpression<T> {
}

6
modules/semrack-index-postgres/runtime/src/main/java/fr/codeanddata/semrack/index/postgres/operators/OrLookup.java
View File

@@ -1,6 +1,6 @@
package fr.codeanddata.semrack.index.postgres.operators;
import fr.codeanddata.semrack.core.services.SemrackLookupService;
import fr.codeanddata.semrack.core.services.LookupService;
import io.smallrye.common.annotation.Identifier;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
@@ -10,10 +10,10 @@ import java.util.Map;
@Identifier("or")
@ApplicationScoped
public class OrLookup implements SemrackJpaLookupExpression<List<Map<String, Object>>> {
public class OrLookup implements JpaLookupExpression<List<Map<String, Object>>> {
@Inject
SemrackLookupService lookupService;
LookupService lookupService;
@Override
public String apply(Object expressions) {

6
modules/semrack-index-postgres/runtime/src/main/java/fr/codeanddata/semrack/index/postgres/operators/SemrackJpaLookupExpression.java
View File

@@ -1,6 +0,0 @@
package fr.codeanddata.semrack.index.postgres.operators;
import fr.codeanddata.semrack.core.SemrackLookupExpression;
public interface SemrackJpaLookupExpression<T> extends SemrackLookupExpression<T> {
}

10
modules/semrack-storage-postgres/deployment/pom.xml
View File

@@ -45,6 +45,16 @@
<artifactId>quarkus-junit5-internal</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>io.quarkus</groupId>
<artifactId>quarkus-junit5</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>io.quarkus</groupId>
<artifactId>quarkus-test-vertx</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>fr.codeanddata.semrack</groupId>
<artifactId>semrack-core-testing</artifactId>

73
modules/semrack-storage-postgres/deployment/src/test/java/fr/codeanddata/semrack/storage/postgres/storage/JpaStorageTest.java Normal file
View File

@@ -0,0 +1,73 @@
package fr.codeanddata.semrack.storage.postgres.storage;
import fr.codeanddata.semrack.core.models.StorageGet;
import io.quarkus.hibernate.reactive.panache.Panache;
import io.quarkus.test.junit.QuarkusTest;
import io.quarkus.test.vertx.RunOnVertxContext;
import io.quarkus.test.vertx.UniAsserter;
import jakarta.inject.Inject;
import org.junit.jupiter.api.Test;
import java.util.Map;
@QuarkusTest
public class JpaStorageTest {
@Inject
JpaStorage jpaStorage;
@Test
@RunOnVertxContext
void testJpaStorageGetDefault(UniAsserter asserter) {
asserter.assertThat(
() -> Panache.withSession(() -> jpaStorage.get("doc-001", StorageGet.builder().build())),
doc -> {
assert doc != null;
assert "doc-001".equals(doc.getUid());
assert doc.getAnnotations() == null;
assert doc.getMetadata() == null;
assert doc.getFields() == null;
}
);
}
@Test
@RunOnVertxContext
void testJpaStorageGetWithAnnotationsAndMetadata(UniAsserter asserter) {
asserter.assertThat(
() -> Panache.withSession(() -> jpaStorage.get("doc-001", StorageGet.builder()
.annotationsSource(true)
.metadataSource(true)
.build())),
doc -> {
assert doc != null;
assert doc.getAnnotations() != null;
assert doc.getMetadata() != null;
}
);
}
@Test
@RunOnVertxContext
void testJpaStorageGetWithFields(UniAsserter asserter) {
asserter.assertThat(
() -> Panache.withSession(() -> jpaStorage.get("doc-001", StorageGet.builder()
.fields(Map.of(
"title", "title",
"content", "content",
"sku", "product.sku"
))
.build())),
doc -> {
assert doc != null;
assert doc.getFields() != null;
assert doc.getFields().containsKey("title");
assert "Test Document".equals(doc.getFields().get("title"));
assert doc.getFields().containsKey("content");
assert "Hello World".equals(doc.getFields().get("content"));
assert doc.getFields().containsKey("sku");
assert "001".equals(doc.getFields().get("sku"));
}
);
}
}

23
modules/semrack-storage-postgres/deployment/src/test/java/fr/codeanddata/semrack/storage/postgres/test/SemrackStoragePostgresDevModeTest.java
View File

@@ -1,23 +0,0 @@
package fr.codeanddata.semrack.storage.postgres.test;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.spec.JavaArchive;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;
import io.quarkus.test.QuarkusDevModeTest;
public class SemrackStoragePostgresDevModeTest {
// Start hot reload (DevMode) test with your extension loaded
@RegisterExtension
static final QuarkusDevModeTest devModeTest = new QuarkusDevModeTest()
.setArchiveProducer(() -> ShrinkWrap.create(JavaArchive.class));
@Test
public void writeYourOwnDevModeTest() {
// Write your dev mode tests here - see the testing extension guide https://quarkus.io/guides/writing-extensions#testing-hot-reload for more information
Assertions.assertTrue(true, "Add dev mode assertions to " + getClass().getName());
}
}

23
modules/semrack-storage-postgres/deployment/src/test/java/fr/codeanddata/semrack/storage/postgres/test/SemrackStoragePostgresTest.java
View File

@@ -1,23 +0,0 @@
package fr.codeanddata.semrack.storage.postgres.test;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.spec.JavaArchive;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;
import io.quarkus.test.QuarkusUnitTest;
public class SemrackStoragePostgresTest {
// Start unit test with your extension loaded
@RegisterExtension
static final QuarkusUnitTest unitTest = new QuarkusUnitTest()
.setArchiveProducer(() -> ShrinkWrap.create(JavaArchive.class));
@Test
public void writeYourOwnUnitTest() {
// Write your unit tests here - see the testing extension guide https://quarkus.io/guides/writing-extensions#testing-extensions for more information
Assertions.assertTrue(true, "Add some assertions to " + getClass().getName());
}
}

3
modules/semrack-storage-postgres/deployment/src/test/resources/application.properties Normal file
View File

@@ -0,0 +1,3 @@
quarkus.flyway.enabled=false
quarkus.hibernate-orm.schema-management.strategy=drop-and-create
quarkus.hibernate-orm.sql-load-script=import.sql

5
modules/semrack-storage-postgres/deployment/src/test/resources/import.sql Normal file
View File

@@ -0,0 +1,5 @@
INSERT INTO semrack_document (id, uid, document)
VALUES (nextval('semrack_document_SEQ'), 'doc-001',
'{"uid":"doc-001","annotations":{"category":"test"},"metadata":{"product":{"sku":"001"},"title":"Test Document","content":"Hello World"}}'),
(nextval('semrack_document_SEQ'), 'doc-002',
'{"uid":"doc-002","annotations":{"category":"test"},"metadata":{"product":{"sku":"002"},"title":"Test Document 2","content":"Hello World 2"}}');

4
modules/semrack-storage-postgres/runtime/src/main/java/fr/codeanddata/semrack/storage/postgres/entities/SemrackDocumentEntity.java
View File

@@ -1,6 +1,6 @@
package fr.codeanddata.semrack.storage.postgres.entities;
import fr.codeanddata.semrack.core.models.SemrackDocument;
import fr.codeanddata.semrack.core.models.Document;
import io.quarkus.hibernate.reactive.panache.PanacheEntity;
import jakarta.persistence.Column;
import jakarta.persistence.Entity;
@@ -23,5 +23,5 @@ public class SemrackDocumentEntity extends PanacheEntity implements Serializable
String uid;
@JdbcTypeCode(SqlTypes.JSON)
SemrackDocument document;
Document document;
}

148
modules/semrack-storage-postgres/runtime/src/main/java/fr/codeanddata/semrack/storage/postgres/storage/JpaStorage.java Normal file
View File

@@ -0,0 +1,148 @@
package fr.codeanddata.semrack.storage.postgres.storage;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import fr.codeanddata.semrack.core.Storage;
import fr.codeanddata.semrack.core.models.Document;
import fr.codeanddata.semrack.core.models.StorageGet;
import fr.codeanddata.semrack.core.utils.UIDGenerator;
import fr.codeanddata.semrack.storage.postgres.entities.SemrackDocumentEntity;
import io.quarkus.hibernate.reactive.panache.PanacheRepository;
import io.quarkus.hibernate.reactive.panache.common.WithSession;
import io.quarkus.hibernate.reactive.panache.common.WithTransaction;
import io.smallrye.mutiny.Uni;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
import java.util.*;
@ApplicationScoped
public class JpaStorage implements Storage, PanacheRepository<SemrackDocumentEntity> {
@Inject
UIDGenerator generator;
@Inject
ObjectMapper objectMapper;
@Override
@WithSession
public Uni<Document> get(String uid, StorageGet request) {
final List<String> fields = selectFields(request);
return getSession()
.chain(session -> session.createNativeQuery("select " + String.join(",", fields) + " from semrack_document where uid = ?1")
.setParameter(1, uid)
.getSingleResult())
.map(result -> mapDocument(result, request));
}
@Override
@WithSession
public Uni<List<Document>> get(List<String> uids, StorageGet request) {
final List<String> fields = selectFields(request);
return getSession()
.chain(session -> session.createNativeQuery("select " + String.join(",", fields) + " from semrack_document where uid in ?1")
.setParameter(1, uids)
.getResultList())
.map(results -> mapDocuments(results, request));
}
@WithTransaction
@Override
public Uni<Document> storeDocument(Document document) {
if (document.getUid() == null) {
return createDocument(document)
.call(this::flush);
} else {
return find("uid = ?1", document.getUid())
.count()
.chain(n -> n == 0 ? createDocument(document) : updateDocument(document))
.call(this::flush);
}
}
// TODO err : handle existing document
@WithTransaction
Uni<Document> createDocument(Document document) {
final String uid = document.getUid() == null ? generator.apply(document) : document.getUid();
document.setUid(uid);
final SemrackDocumentEntity entity = SemrackDocumentEntity.builder()
.uid(uid)
.document(document)
.build();
return persist(entity).map(SemrackDocumentEntity::getDocument);
}
// TODO err : document not exists
@WithTransaction
Uni<Document> updateDocument(Document document) {
return update("uid = ?1, document = ?2 WHERE uid = ?1", document.getUid(), document)
.map(x -> document);
}
List<String> selectFields(StorageGet request) {
final List<String> fields = new ArrayList<>(List.of(
"uid as _uid",
(Optional.ofNullable(request.getAnnotationsSource()).orElse(false) ? "document->'annotations'" : "null") + " as _annotations",
(Optional.ofNullable(request.getMetadataSource()).orElse(false) ? "document->'metadata'" : "null") + " as _metadata"
));
if (request.getFields() != null) {
int i = 0;
for (String field : request.getFields().values()) {
fields.add("jsonb_path_query(document, '$.metadata." + field + "') as field_" + i++);
}
}
return fields;
}
Document mapDocument(Object result, StorageGet request) {
final Map<String, Integer> fieldMap = buildFieldMap(request);
final List<Object> resultList = objectMapper.convertValue(result, new TypeReference<>() {
});
final Object oAnnotations = resultList.get(fieldMap.get("_annotations"));
final Object oMetadata = resultList.get(fieldMap.get("_metadata"));
final Map<String, Object> annotations = oAnnotations == null ? null : objectMapper.convertValue(oAnnotations, new TypeReference<>() {
});
final Map<String, Object> metadata = oMetadata == null ? null : objectMapper.convertValue(oMetadata, new TypeReference<>() {
});
final Map<String, Object> fields = new HashMap<>();
if (request.getFields() != null) {
for (final String field : request.getFields().keySet()) {
fields.put(field, resultList.get(fieldMap.get(field)));
}
}
return Document.builder()
.uid((String) resultList.get(fieldMap.get("_uid")))
.annotations(annotations)
.metadata(metadata)
.fields(fields.isEmpty() ? null : fields)
.build();
}
List<Document> mapDocuments(List<Object> results, StorageGet request) {
return results.stream().map(doc -> mapDocument(doc, request)).toList();
}
Map<String, Integer> buildFieldMap(StorageGet request) {
final Map<String, Integer> fieldsMap = new HashMap<>(Map.of(
"_uid", 0,
"_annotations", 1,
"_metadata", 2
));
if (request.getFields() != null) {
Integer i = 3;
for (String field : request.getFields().keySet()) {
fieldsMap.put(field, i);
i++;
}
}
return fieldsMap;
}
}

144
modules/semrack-storage-postgres/runtime/src/main/java/fr/codeanddata/semrack/storage/postgres/storage/SemdocJpaStorage.java
View File

@@ -1,144 +0,0 @@
package fr.codeanddata.semrack.storage.postgres.storage;
import fr.codeanddata.semrack.core.SemdocStorage;
import fr.codeanddata.semrack.core.models.SemrackDocument;
import fr.codeanddata.semrack.core.services.SemrackLookupService;
import fr.codeanddata.semrack.core.utils.UIDGenerator;
import fr.codeanddata.semrack.storage.postgres.entities.SemrackDocumentEntity;
import io.quarkus.hibernate.reactive.panache.PanacheRepository;
import io.quarkus.hibernate.reactive.panache.common.WithSession;
import io.quarkus.hibernate.reactive.panache.common.WithTransaction;
import io.smallrye.mutiny.Uni;
import jakarta.enterprise.context.ApplicationScoped;
import jakarta.inject.Inject;
import java.util.List;
import java.util.Optional;
@ApplicationScoped
public class SemdocJpaStorage implements SemdocStorage, PanacheRepository<SemrackDocumentEntity> {
@Inject
UIDGenerator generator;
@Inject
SemrackLookupService lookupService;
@Override
@WithSession
public Uni<SemrackDocument> get(String uid) {
return find("uid = ?1", uid)
.firstResult().map(d -> Optional.ofNullable(d).map(SemrackDocumentEntity::getDocument).orElse(null));
}
@Override
@WithSession
public Uni<List<SemrackDocument>> get(List<String> uids) {
return find("uid in ?1", uids).list()
.map(d -> d.stream().map(SemrackDocumentEntity::getDocument).toList());
}
@WithTransaction
@Override
public Uni<SemrackDocument> storeDocument(SemrackDocument document) {
if (document.getUid() == null) {
return createDocument(document);
} else {
return find("uid = ?1", document.getUid())
.count()
.chain(n -> n == 0 ? createDocument(document) : updateDocument(document));
}
}
// TODO err : handle existing document
@WithTransaction
Uni<SemrackDocument> createDocument(SemrackDocument document) {
final String uid = document.getUid() == null ? generator.apply(document) : document.getUid();
document.setUid(uid);
final SemrackDocumentEntity entity = SemrackDocumentEntity.builder()
.uid(uid)
.document(document)
.build();
return persist(entity).map(SemrackDocumentEntity::getDocument);
}
// TODO err : document not exists
@WithTransaction
Uni<SemrackDocument> updateDocument(SemrackDocument document) {
return update("uid = ?1, document = ?2 WHERE uid = ?1", document.getUid(), document)
.map(x -> document);
}
// Uni<SearchResult> search(SearchRequest request) {
// final String baseQuery = buildBaseQuery(request);
//
// final StringBuilder sorting = new StringBuilder();
// if (request.getSort() != null && !request.getSort().isEmpty()) {
// sorting.append(" ORDER BY ");
// sorting.append(String.join(", ", request.getSort().stream().map(sort -> sort.getField() + " " + Optional.ofNullable(sort.getDirection()).orElse(SemrackSortDirection.asc).name()).toList()));
// }
//
// final StringBuilder paginate = new StringBuilder();
// if (request.getPaginate() != null) {
// final SemrackPagination pagination = request.getPaginate();
// final Integer size = pagination.getSize() == null ? 200 : pagination.getSize();
// final Integer page = pagination.getPage() == null ? 0 : pagination.getPage();
//
// paginate.append(" LIMIT ").append(size).append(" OFFSET ").append(size * page);
// }
//
// return getSession()
// .chain(s -> {
// final SearchResult searchResult = SearchResult.builder().build();
// return countDocuments(request)
// .invoke(count -> searchResult.setPagination(PaginationInfo.builder()
// .page(Optional.ofNullable(request.getPaginate()).map(SemrackPagination::getPage).orElse(0))
// .size(Optional.ofNullable(request.getPaginate()).map(SemrackPagination::getSize).orElse(0))
// .total(count)
// .build()))
// .call(() -> s
// .createNativeQuery(baseQuery + sorting + paginate, Object.class).getResultList()
// .invoke(results -> searchResult.setDocuments(project(request.getFields(), results))))
// .map(count -> searchResult);
// });
// }
//
// String buildBaseQuery(SearchRequest request) {
// final String lookup = lookupService.lookup(request.getFilter());
// final String whereClause = lookup.isEmpty() ? "" : " WHERE " + lookup;
//
// if (request.getFields() == null || request.getFields().isEmpty()) {
// return "SELECT document FROM semrack_document" + whereClause;
// } else {
// return "SELECT " + toJsonbPathExtract(request.getFields()) + " FROM semrack_document" + whereClause;
// }
// }
//
// String toJsonbPathExtract(List<String> fields) {
// return String.join(",", fields.stream().map(field -> {
// final String serializedField = "'" + String.join("','", field.split("\\.")) + "'";
// return "jsonb_extract_path(document, " + serializedField + ")";
// }).toList());
// }
//
// List<SemrackDocument> project(List<String> fields, Object results) {
// final List<?> projectedResults = objectMapper.convertValue(results, List.class);
// if (fields == null || fields.isEmpty()) {
// return projectedResults.stream().map(result -> objectMapper.convertValue(result, SemrackDocument.class)).toList();
// } else {
// return projectedResults.stream().map(result -> {
// final List<Object> row = fields.size() == 1 ? List.of(result) : objectMapper.convertValue(result, new TypeReference<>() {
// });
// final List<Object> rowMut = new ArrayList<>(row);
// final List<String> fieldCp = new ArrayList<>(fields);
// final Map<String, Object> document = new HashMap<>();
// while (!fieldCp.isEmpty() && !row.isEmpty()) {
// final String field = fieldCp.removeFirst();
// final Object value = rowMut.removeFirst();
// pathMapper.map(field, document, value);
// }
// return objectMapper.convertValue(document, SemrackDocument.class);
// }).toList();
// }
// }
}