Adds the ability to to disable annotation lookup by internal document id

web-services/annotations/src/main/java/datawave/webservice/annotation/AnnotationManagerBean.java

@@ -562,14 +562,20 @@ public Response updateSegment(@PathParam("idType") String idType, @PathParam("id
      *             if the id is malformed.
      */
     private List<Metadata> lookupDocumentIdentifier(String idType, String id) throws QueryException {
+        // If the idType is RECORD_ID or DOCUMENT, treat the id provided as an internal id and perform a direct lookup
+        // against the annotations table, if that's enabled.
         if (idType.equals("DOCUMENT") || idType.equals("RECORD_ID")) {
-            // If the idType is RECORD_ID or DOCUMENT treat the id provided is an internal id.
+            if (!config.isEnableInternalIdLookup()) {
+                final String message = String.format("Internal identifier lookup is disabled for '%s:%s' please use a valid document id type.", idType, id);
+                throw new QueryException(message);
+            }
+
             return parseDocumentIdentifier(id);
-        } else {
-            // Otherwise, perform a lookup to find the internal id.
-            final LookupUUIDService lookup = initializeLookupUUIDService();
-            return lookup.executeLookupUUIDQuery(idType, id);
         }
+
+        // Otherwise, use the lookup uuid service to perform a lookup to find the internal id in the shard table.
+        final LookupUUIDService lookup = initializeLookupUUIDService();
+        return lookup.executeLookupUUIDQuery(idType, id);
     }
 
     /**
@@ -671,4 +677,9 @@ private static Response jsonOk(Object responseObject) {
         // TODO: do we want to return more? (e.g., include fields like internal id, etc..
         return Response.ok(responseObject, MediaType.APPLICATION_JSON_TYPE.withCharset("utf-8")).build();
     }
+
+    @VisibleForTesting
+    protected AnnotationManagerConfig getConfig() {
+        return config;
+    }
 }

web-services/annotations/src/main/java/datawave/webservice/annotation/AnnotationManagerConfig.java

@@ -9,6 +9,7 @@ public class AnnotationManagerConfig {
     private String annotationTableName;
     private String annotationSourceTableName;
     private String connPoolName;
+    private boolean enableInternalIdLookup = false;
     private ShardQueryLogic lookupUUIDQueryLogic;
     private LookupUUIDServiceConfig lookupUUIDServiceConfig;
     private AccumuloConnectionFactory.Priority priority = AccumuloConnectionFactory.Priority.LOW;
@@ -78,4 +79,12 @@ public VisibilityTransformer getVisibilityTransformer() {
     public void setVisibilityTransformer(VisibilityTransformer visibilityTransformer) {
         this.visibilityTransformer = visibilityTransformer;
     }
+
+    public boolean isEnableInternalIdLookup() {
+        return enableInternalIdLookup;
+    }
+
+    public void setEnableInternalIdLookup(boolean enableInternalIdLookup) {
+        this.enableInternalIdLookup = enableInternalIdLookup;
+    }
 }

web-services/annotations/src/test/java/datawave/webservice/annotation/AnnotationManagerBeanFunctionalTest.java

@@ -433,6 +433,21 @@ public void testGetAnnotationInternalId() {
         assertAnnotationsEqual(expectedAnnotation, annotationList.iterator().next());
     }
 
+    @Test
+    public void testGetAnnotationInternalIdDisabled() {
+        AnnotationManagerBean bean = (AnnotationManagerBean) annotationManager;
+        AnnotationManagerConfig config = bean.getConfig();
+        config.setEnableInternalIdLookup(false);
+
+        Annotation testAnnotation = generateTestAnnotation();
+        Annotation expectedAnnotation = AnnotationUtils.injectAllHashes(testAnnotation);
+        Response response = annotationManager.getAnnotation("DOCUMENT", "20250704_249/testDataType/abcde.fghij.klmno", "23BD91EC");
+        assertResponseStatus(500, response);
+        String errorResponse = assertExpectedEntity(String.class, response);
+        assertContains("Internal identifier lookup is disabled for", errorResponse);
+        assertContains("20250704_249/testDataType/abcde.fghij.klmno", errorResponse);
+    }
+
     @Test
     public void testGetAnnotationMissingInternalId() {
         Response response = annotationManager.getAnnotation("DOCUMENT", "20250704_249/testDataType/abcde.fghij.klmno", "aaaaaaaa");
@@ -441,7 +456,6 @@ public void testGetAnnotationMissingInternalId() {
         assertContains("No annotations found for identifier", errorResponse);
         assertContains("20250704_249/testDataType/abcde.fghij.klmno", errorResponse);
         assertContains("aaaaaaaa", errorResponse);
-
     }
 
     @Test

web-services/annotations/src/test/resources/datawave/webservice/annotation/AnnotationManagerFactory.xml

@@ -14,6 +14,7 @@
         <property name="annotationTableName" value="annotation"/>
         <property name="annotationSourceTableName" value="annotationSource"/>
         <property name="connPoolName" value="WAREHOUSE" />
+        <property name="enableInternalIdLookup" value="true" />
         <property name="lookupUUIDQueryLogic">
             <bean scope="prototype" parent="BaseEventQuery">
                 <property name="auditType" value="NONE" />

web-services/deploy/configuration/src/main/resources/datawave/annotation/AnnotationManagerFactory.xml

@@ -22,6 +22,7 @@
         <property name="annotationTableName" value="${annotation.table.name}"/>
         <property name="annotationSourceTableName" value="${annotation.source.table.name}"/>
         <property name="connPoolName" value="WAREHOUSE" />
+        <property name="enableInternalIdLookup" value="true" />
         <property name="lookupUUIDQueryLogic">
             <bean scope="prototype" parent="BaseEventQuery">
                 <property name="auditType" value="NONE" />