HSEARCH-3269 Define one to one identifier bridge

hibernate · Oct 15, 2020 · 7599ec8 · 7599ec8
1 parent 6f67073
commit 7599ec8
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 1 deletion.
diff --git a/...a/org/hibernate/search/integrationtest/batch/jsr352/massindexing/bridge/DateIdBridge.java b/...a/org/hibernate/search/integrationtest/batch/jsr352/massindexing/bridge/DateIdBridge.java
@@ -0,0 +1,46 @@
+/*
+ * Hibernate Search, full-text search for your domain model
+ *
+ * License: GNU Lesser General Public License (LGPL), version 2.1 or later
+ * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
+ */
+package org.hibernate.search.integrationtest.batch.jsr352.massindexing.bridge;
+
+import java.util.Locale;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import org.hibernate.search.integrationtest.batch.jsr352.massindexing.id.EmbeddableDateId;
+import org.hibernate.search.mapper.pojo.bridge.IdentifierBridge;
+import org.hibernate.search.mapper.pojo.bridge.runtime.IdentifierBridgeFromDocumentIdentifierContext;
+import org.hibernate.search.mapper.pojo.bridge.runtime.IdentifierBridgeToDocumentIdentifierContext;
+
+public class DateIdBridge implements IdentifierBridge<EmbeddableDateId> {
+
+	private static Pattern DATE_PATTERN = Pattern.compile( "^\\d{4}-\\d{2}-\\d{2}$" );
+
+	@Override
+	public String toDocumentIdentifier(EmbeddableDateId propertyValue,
+			IdentifierBridgeToDocumentIdentifierContext context) {
+
+		return String.format( Locale.ROOT, "%04d-%02d-%02d",
+				propertyValue.getYear(), propertyValue.getMonth(), propertyValue.getDay()
+		);
+	}
+
+	@Override
+	public EmbeddableDateId fromDocumentIdentifier(String documentIdentifier,
+			IdentifierBridgeFromDocumentIdentifierContext context) {
+
+		Matcher matcher = DATE_PATTERN.matcher( documentIdentifier );
+		if ( !matcher.find() ) {
+			throw new RuntimeException( "Date does not match the pattern d{4}-d{2}-d{2}: " + documentIdentifier );
+		}
+
+		EmbeddableDateId result = new EmbeddableDateId();
+		result.setYear( Integer.parseInt( matcher.group( 1 ) ) );
+		result.setMonth( Integer.parseInt( matcher.group( 2 ) ) );
+		result.setDay( Integer.parseInt( matcher.group( 3 ) ) );
+		return result;
+	}
+}
diff --git a/...bernate/search/integrationtest/batch/jsr352/massindexing/entity/EntityWithEmbeddedId.java b/...bernate/search/integrationtest/batch/jsr352/massindexing/entity/EntityWithEmbeddedId.java
@@ -12,7 +12,9 @@
 import javax.persistence.EmbeddedId;
 import javax.persistence.Entity;
 
+import org.hibernate.search.integrationtest.batch.jsr352.massindexing.bridge.DateIdBridge;
 import org.hibernate.search.integrationtest.batch.jsr352.massindexing.id.EmbeddableDateId;
+import org.hibernate.search.mapper.pojo.bridge.mapping.annotation.IdentifierBridgeRef;
 import org.hibernate.search.mapper.pojo.mapping.definition.annotation.DocumentId;
 import org.hibernate.search.mapper.pojo.mapping.definition.annotation.FullTextField;
 import org.hibernate.search.mapper.pojo.mapping.definition.annotation.Indexed;
@@ -25,7 +27,7 @@
 public class EntityWithEmbeddedId {
 
 	@EmbeddedId
-	@DocumentId
+	@DocumentId(identifierBridge = @IdentifierBridgeRef(type = DateIdBridge.class))
 	private EmbeddableDateId embeddableDateId;
 
 	@FullTextField