diff --git a/notification-core/src/main/java/org/opengroup/osdu/notification/api/PubsubEndpoint.java b/notification-core/src/main/java/org/opengroup/osdu/notification/api/PubsubEndpoint.java index 975d66958a16e46da89103d1441d97219be42fe1..935ecaa241bc9774f720570af83de6ddcca15fa4 100644 --- a/notification-core/src/main/java/org/opengroup/osdu/notification/api/PubsubEndpoint.java +++ b/notification-core/src/main/java/org/opengroup/osdu/notification/api/PubsubEndpoint.java @@ -34,7 +34,8 @@ import org.opengroup.osdu.core.common.notification.ISubscriptionFactory; import org.opengroup.osdu.core.common.notification.ISubscriptionService; import org.opengroup.osdu.core.common.notification.SubscriptionException; import org.opengroup.osdu.notification.di.SubscriptionCacheFactory; -import org.opengroup.osdu.notification.pubsub.PubsubRequestBodyExtractor; +import org.opengroup.osdu.notification.pubsub.IPubsubHandshakeHandler; +import org.opengroup.osdu.notification.pubsub.IPubsubRequestBodyExtractor; import org.opengroup.osdu.notification.utils.Config; import org.opengroup.osdu.notification.utils.IGoogleServiceAccount; import org.springframework.beans.factory.annotation.Autowired; @@ -56,7 +57,9 @@ import java.util.Map; @RequestMapping("/push-handlers") public class PubsubEndpoint { @Autowired - private PubsubRequestBodyExtractor pubsubRequestBodyExtractor; + private IPubsubRequestBodyExtractor pubsubRequestBodyExtractor; + @Autowired + private IPubsubHandshakeHandler pubsubHandshakeHandler; @Autowired private ISignatureService signatureService; @Autowired @@ -83,6 +86,10 @@ public class PubsubEndpoint { @PostMapping("/records-changed") @PreAuthorize("@authorizationFilter.hasAnyPermission('" + Config.OPS + "', '" + Config.PUBSUB + "')") public ResponseEntity recordChanged() throws Exception { + if(this.pubsubRequestBodyExtractor.isHandshakeRequest()) { + this.pubsubHandshakeHandler.getHandshakeResponse(); + } + String notificationId = this.pubsubRequestBodyExtractor.extractNotificationIdFromRequestBody(); String pubsubMessage = this.pubsubRequestBodyExtractor.extractDataFromRequestBody(); Map<String, String> headerAttributes = this.pubsubRequestBodyExtractor.extractAttributesFromRequestBody(); diff --git a/notification-core/src/main/java/org/opengroup/osdu/notification/di/CredentialHeadersProvider.java b/notification-core/src/main/java/org/opengroup/osdu/notification/di/CredentialHeadersProvider.java index f27e7815c3f1a3b575e70a070a18a017ea7f9b02..4f85dc619d2469b9de6d1754910ded8a4d9609db 100644 --- a/notification-core/src/main/java/org/opengroup/osdu/notification/di/CredentialHeadersProvider.java +++ b/notification-core/src/main/java/org/opengroup/osdu/notification/di/CredentialHeadersProvider.java @@ -15,9 +15,10 @@ */ package org.opengroup.osdu.notification.di; + import org.opengroup.osdu.core.common.model.http.DpsHeaders; import org.opengroup.osdu.core.common.util.IServiceAccountJwtClient; -import org.opengroup.osdu.notification.pubsub.PubsubRequestBodyExtractor; +import org.opengroup.osdu.notification.pubsub.IPubsubRequestBodyExtractor; import org.opengroup.osdu.notification.utils.IAppProperties; import org.springframework.beans.factory.FactoryBean; import org.springframework.beans.factory.annotation.Autowired; @@ -39,7 +40,7 @@ public class CredentialHeadersProvider implements FactoryBean<DpsHeaders> { private HttpServletRequest httpRequest; @Autowired - private PubsubRequestBodyExtractor pubsubRequestBodyExtractor; + private IPubsubRequestBodyExtractor pubsubRequestBodyExtractor; @Autowired private IServiceAccountJwtClient serviceAccountJwtClient; diff --git a/notification-core/src/main/java/org/opengroup/osdu/notification/di/RequestInfoExt.java b/notification-core/src/main/java/org/opengroup/osdu/notification/di/RequestInfoExt.java index 13adf62cb5605971ede3ef9379057e80e752ea34..b281624634337f750f58df40047317dc76c47392 100644 --- a/notification-core/src/main/java/org/opengroup/osdu/notification/di/RequestInfoExt.java +++ b/notification-core/src/main/java/org/opengroup/osdu/notification/di/RequestInfoExt.java @@ -19,7 +19,7 @@ package org.opengroup.osdu.notification.di; import com.google.api.client.util.Strings; import org.opengroup.osdu.core.common.model.http.DpsHeaders; import org.opengroup.osdu.core.common.model.http.RequestInfo; -import org.opengroup.osdu.notification.pubsub.PubsubRequestBodyExtractor; +import org.opengroup.osdu.notification.pubsub.IPubsubRequestBodyExtractor; import org.springframework.context.annotation.Primary; import org.springframework.stereotype.Component; import org.springframework.web.context.annotation.RequestScope; @@ -40,7 +40,7 @@ public class RequestInfoExt extends RequestInfo { private HttpServletRequest httpRequest; @Inject - private PubsubRequestBodyExtractor requestBodyExtractor; + private IPubsubRequestBodyExtractor requestBodyExtractor; @Inject public RequestInfoExt(HttpServletRequest request) { diff --git a/notification-core/src/main/java/org/opengroup/osdu/notification/pubsub/IPubsubHandshakeHandler.java b/notification-core/src/main/java/org/opengroup/osdu/notification/pubsub/IPubsubHandshakeHandler.java new file mode 100644 index 0000000000000000000000000000000000000000..e8f8f043c20ea7e71457ff4c143b514c3a7b8593 --- /dev/null +++ b/notification-core/src/main/java/org/opengroup/osdu/notification/pubsub/IPubsubHandshakeHandler.java @@ -0,0 +1,9 @@ +package org.opengroup.osdu.notification.pubsub; + +import org.springframework.http.ResponseEntity; +import org.springframework.stereotype.Component; + +@Component +public interface IPubsubHandshakeHandler { + ResponseEntity getHandshakeResponse(); +} diff --git a/notification-core/src/main/java/org/opengroup/osdu/notification/pubsub/IPubsubRequestBodyExtractor.java b/notification-core/src/main/java/org/opengroup/osdu/notification/pubsub/IPubsubRequestBodyExtractor.java new file mode 100644 index 0000000000000000000000000000000000000000..613effe1769af74039023433b383de2db1494931 --- /dev/null +++ b/notification-core/src/main/java/org/opengroup/osdu/notification/pubsub/IPubsubRequestBodyExtractor.java @@ -0,0 +1,17 @@ +package org.opengroup.osdu.notification.pubsub; + +import org.springframework.stereotype.Component; +import org.springframework.http.ResponseEntity; + +import java.util.Map; + +@Component +public interface IPubsubRequestBodyExtractor { + Map<String, String> extractAttributesFromRequestBody() ; + + String extractDataFromRequestBody(); + + String extractNotificationIdFromRequestBody() ; + + boolean isHandshakeRequest(); +} diff --git a/notification-core/src/test/java/org/opengroup/osdu/notification/api/PubsubEndpointTests.java b/notification-core/src/test/java/org/opengroup/osdu/notification/api/PubsubEndpointTests.java index 67b9a4a813675698e8e57f07651102c762505dc1..4f4a2fa249d8e4f04f89c51ce153b436cf8cfa54 100644 --- a/notification-core/src/test/java/org/opengroup/osdu/notification/api/PubsubEndpointTests.java +++ b/notification-core/src/test/java/org/opengroup/osdu/notification/api/PubsubEndpointTests.java @@ -36,7 +36,7 @@ import org.opengroup.osdu.core.common.notification.SubscriptionFactory; import org.opengroup.osdu.core.common.notification.SubscriptionService; import org.opengroup.osdu.notification.di.CredentialHeadersProvider; import org.opengroup.osdu.notification.di.SubscriptionCacheFactory; -import org.opengroup.osdu.notification.pubsub.PubsubRequestBodyExtractor; +import org.opengroup.osdu.notification.pubsub.IPubsubRequestBodyExtractor; import org.opengroup.osdu.notification.utils.IGoogleServiceAccount; import org.powermock.modules.junit4.PowerMockRunner; import org.springframework.http.ResponseEntity; @@ -54,7 +54,7 @@ import static org.mockito.Mockito.when; @RunWith(PowerMockRunner.class) public class PubsubEndpointTests { @Mock - private PubsubRequestBodyExtractor pubsubRequestBodyExtractor; + private IPubsubRequestBodyExtractor pubsubRequestBodyExtractor; @Mock private ISignatureService signatureService; @Mock diff --git a/notification-core/src/test/java/org/opengroup/osdu/notification/auth/AuthorizationFilterTest.java b/notification-core/src/test/java/org/opengroup/osdu/notification/auth/AuthorizationFilterTest.java index c73e87d111a18a0e8077604389a2c8e396eb9c8b..6fe3fccafd62c2e20c25093ebdabe78a5fc80ccb 100644 --- a/notification-core/src/test/java/org/opengroup/osdu/notification/auth/AuthorizationFilterTest.java +++ b/notification-core/src/test/java/org/opengroup/osdu/notification/auth/AuthorizationFilterTest.java @@ -26,7 +26,7 @@ import org.opengroup.osdu.core.common.model.http.AppException; import org.opengroup.osdu.core.common.model.http.DpsHeaders; import org.opengroup.osdu.core.common.provider.interfaces.IAuthorizationService; import org.opengroup.osdu.notification.di.RequestInfoExt; -import org.opengroup.osdu.notification.pubsub.PubsubRequestBodyExtractor; +import org.opengroup.osdu.notification.pubsub.IPubsubRequestBodyExtractor; import org.opengroup.osdu.notification.utils.Config; import org.opengroup.osdu.notification.utils.ServiceAccountValidator; import org.powermock.modules.junit4.PowerMockRunner; @@ -59,7 +59,7 @@ public class AuthorizationFilterTest { @Mock private ServiceAccountValidator validator; @Mock - private PubsubRequestBodyExtractor extractor; + private IPubsubRequestBodyExtractor extractor; @InjectMocks private AuthorizationFilter sut; diff --git a/provider/notification-gcp/src/main/java/org/opengroup/osdu/notification/provider/gcp/pubsub/PubsubHandshakeHandler.java b/provider/notification-gcp/src/main/java/org/opengroup/osdu/notification/provider/gcp/pubsub/PubsubHandshakeHandler.java new file mode 100644 index 0000000000000000000000000000000000000000..33b9b156267067c7bf003c2d87d1fd859e461882 --- /dev/null +++ b/provider/notification-gcp/src/main/java/org/opengroup/osdu/notification/provider/gcp/pubsub/PubsubHandshakeHandler.java @@ -0,0 +1,12 @@ +package org.opengroup.osdu.notification.provider.gcp.pubsub; + +import org.opengroup.osdu.notification.pubsub.IPubsubHandshakeHandler; +import org.springframework.http.ResponseEntity; + +public class PubsubHandshakeHandler implements IPubsubHandshakeHandler { + + @Override + public ResponseEntity getHandshakeResponse() { + return null; + } +} diff --git a/notification-core/src/main/java/org/opengroup/osdu/notification/pubsub/PubsubRequestBodyExtractor.java b/provider/notification-gcp/src/main/java/org/opengroup/osdu/notification/provider/gcp/pubsub/PubsubRequestBodyExtractor.java similarity index 94% rename from notification-core/src/main/java/org/opengroup/osdu/notification/pubsub/PubsubRequestBodyExtractor.java rename to provider/notification-gcp/src/main/java/org/opengroup/osdu/notification/provider/gcp/pubsub/PubsubRequestBodyExtractor.java index b8126d6c1e863bcf085a889d7e217d9e876ce785..5137bb3ed1f808e6dae164759e2d8a7b9acb3dcb 100644 --- a/notification-core/src/main/java/org/opengroup/osdu/notification/pubsub/PubsubRequestBodyExtractor.java +++ b/provider/notification-gcp/src/main/java/org/opengroup/osdu/notification/provider/gcp/pubsub/PubsubRequestBodyExtractor.java @@ -14,7 +14,7 @@ * limitations under the License. */ -package org.opengroup.osdu.notification.pubsub; +package org.opengroup.osdu.notification.provider.gcp.pubsub; import com.google.common.base.Strings; import org.opengroup.osdu.core.common.logging.JaxRsDpsLog; @@ -25,6 +25,7 @@ import com.google.gson.Gson; import com.google.gson.JsonElement; import com.google.gson.JsonObject; import com.google.gson.JsonParser; +import org.opengroup.osdu.notification.pubsub.IPubsubRequestBodyExtractor; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.HttpStatus; import org.springframework.stereotype.Component; @@ -41,7 +42,7 @@ import java.util.stream.Stream; @Component @RequestScope -public class PubsubRequestBodyExtractor { +public class PubsubRequestBodyExtractor implements IPubsubRequestBodyExtractor { private static final String INVALID_PUBSUB_MESSAGE = "Invalid pubsub message"; private static final Gson GSON = new Gson(); private MessageContent messageContent; @@ -80,6 +81,10 @@ public class PubsubRequestBodyExtractor { return fullNotificationId[fullNotificationId.length - 1]; } + public boolean isHandshakeRequest() { + return false; + } + private MessageContent extractPubsubMessageFromRequestBody() { if (this.root == null) { this.root = this.extractRootJsonElementFromRequestBody();