From 35cc15e04411008b2f8094bbd3876e7a2daed587 Mon Sep 17 00:00:00 2001
From: Guobiao Mo <guobiaomo@chinamobile.com>
Date: Wed, 10 Apr 2019 00:35:43 -0700
Subject: Support MongoDB as a data storage

Issue-ID: DCAEGEN2-1411
Change-Id: I06b69605e88d5b81500b788847e7c90ff4017a07
Signed-off-by: Guobiao Mo <guobiaomo@chinamobile.com>
---
 .../CollectibleDocumentFieldNameValidator.java     | 60 ++++++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 components/datalake-handler/feeder/src/main/java/com/mongodb/internal/validator/CollectibleDocumentFieldNameValidator.java

(limited to 'components/datalake-handler/feeder/src/main/java/com')

diff --git a/components/datalake-handler/feeder/src/main/java/com/mongodb/internal/validator/CollectibleDocumentFieldNameValidator.java b/components/datalake-handler/feeder/src/main/java/com/mongodb/internal/validator/CollectibleDocumentFieldNameValidator.java
new file mode 100644
index 00000000..2bc6faad
--- /dev/null
+++ b/components/datalake-handler/feeder/src/main/java/com/mongodb/internal/validator/CollectibleDocumentFieldNameValidator.java
@@ -0,0 +1,60 @@
+package com.mongodb.internal.validator;
+
+//copy from https://github.com/mongodb/mongo-java-driver/blob/master/driver-core/src/main/com/mongodb/internal/validator/CollectibleDocumentFieldNameValidator.java
+//allow inserting name with dot
+/*
+* ============LICENSE_START=======================================================
+* ONAP : DataLake
+* ================================================================================
+* Copyright 2018 China Mobile
+*=================================================================================
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at
+*
+*     http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+* ============LICENSE_END=========================================================
+*/
+import org.bson.FieldNameValidator;
+
+import java.util.Arrays;
+import java.util.List;
+
+/**
+ * A field name validator for document that are meant for storage in MongoDB collections.  It ensures that no fields contain a '.',
+ * or start with '$' (with the exception of "$db", "$ref", and "$id", so that DBRefs are not rejected).
+ *
+ * <p>This class should not be considered a part of the public API.</p>
+ */
+public class CollectibleDocumentFieldNameValidator implements FieldNameValidator {
+    // Have to support DBRef fields
+    private static final List<String> EXCEPTIONS = Arrays.asList("$db", "$ref", "$id");
+
+    @Override
+    public boolean validate(final String fieldName) {
+        if (fieldName == null) {
+            throw new IllegalArgumentException("Field name can not be null");
+        }
+
+        /* dl change
+        if (fieldName.contains(".")) {
+            return false;
+        }*/
+
+        if (fieldName.startsWith("$") && !EXCEPTIONS.contains(fieldName)) {
+            return false;
+        }
+        return true;
+    }
+
+    @Override
+    public FieldNameValidator getValidatorForField(final String fieldName) {
+        return this;
+    }
+}
-- 
cgit 1.2.3-korg