nifi-issues mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From "ASF GitHub Bot (JIRA)" <j...@apache.org>
Subject [jira] [Commented] (NIFI-4345) Add a controller service and a lookup service for MongoDB
Date Mon, 11 Sep 2017 14:17:00 GMT

    [ https://issues.apache.org/jira/browse/NIFI-4345?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16161323#comment-16161323
] 

ASF GitHub Bot commented on NIFI-4345:
--------------------------------------

Github user markap14 commented on a diff in the pull request:

    https://github.com/apache/nifi/pull/2123#discussion_r138080737
  
    --- Diff: nifi-nar-bundles/nifi-standard-services/nifi-mongodb-services-bundle/nifi-mongodb-services/src/main/java/org/apache/nifi/mongodb/MongoDBLookupService.java
---
    @@ -0,0 +1,116 @@
    +/*
    + * Licensed to the Apache Software Foundation (ASF) under one or more
    + * contributor license agreements.  See the NOTICE file distributed with
    + * this work for additional information regarding copyright ownership.
    + * The ASF licenses this file to You 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.
    + */
    +
    +package org.apache.nifi.mongodb;
    +
    +import org.apache.nifi.annotation.documentation.CapabilityDescription;
    +import org.apache.nifi.annotation.documentation.Tags;
    +import org.apache.nifi.annotation.lifecycle.OnEnabled;
    +import org.apache.nifi.components.PropertyDescriptor;
    +import org.apache.nifi.controller.ConfigurationContext;
    +import org.apache.nifi.controller.ControllerServiceInitializationContext;
    +import org.apache.nifi.lookup.LookupFailureException;
    +import org.apache.nifi.lookup.LookupService;
    +import org.apache.nifi.processor.util.StandardValidators;
    +import org.apache.nifi.reporting.InitializationException;
    +import org.bson.Document;
    +import org.codehaus.jackson.map.ObjectMapper;
    +
    +import java.io.IOException;
    +import java.util.ArrayList;
    +import java.util.Collections;
    +import java.util.List;
    +import java.util.Map;
    +import java.util.Optional;
    +import java.util.Set;
    +
    +
    +@Tags({"mongo", "mongodb", "lookup", "record"})
    +@CapabilityDescription(
    +    "Provides lookup service functionality through a configured connection to MongoDB.
" +
    +    "One or more keys must defined in LookupRecord in order for it to run. Each value
" +
    +    "must correspond to a full field value in the Mongo collection, as this does not
support " +
    +    "regular expressions and other features."
    +)
    +public class MongoDBLookupService extends MongoDBControllerService implements LookupService
{
    +
    +    public static final PropertyDescriptor LOOKUP_KEY = new PropertyDescriptor.Builder()
    +            .name("mongo-lookup-key")
    +            .displayName("The key from the result to return")
    +            .addValidator(StandardValidators.NON_BLANK_VALIDATOR)
    +            .required(true)
    +            .defaultValue("somevalue")
    +            .build();
    +
    +    private ObjectMapper mapper;
    +    private String lookupKey;
    +
    +    private static final List<PropertyDescriptor> lookupDescriptors;
    +
    +    static {
    +        lookupDescriptors = new ArrayList<>();
    +        lookupDescriptors.addAll(descriptors);
    +        lookupDescriptors.add(LOOKUP_KEY);
    +    }
    +
    +    @Override
    +    public Optional lookup(Map coordinates) throws LookupFailureException {
    +        Document query = new Document(coordinates);
    +
    +        if (coordinates.size() == 0) {
    +            throw new LookupFailureException("No keys were configured. Mongo query would
return random documents.");
    +        }
    +
    +        try {
    +            Document result = this.findOne(query);
    +
    +            Object val = result.get(lookupKey);
    +
    +            return Optional.ofNullable(val);
    +        } catch (Exception ex) {
    +            getLogger().error("Error during lookup {}", new Object[]{ query.toJson()
}, ex);
    +            throw new LookupFailureException(ex);
    +        }
    +    }
    +
    +    @OnEnabled
    +    public void onEnabled(final ConfigurationContext context) throws InitializationException,
IOException, InterruptedException {
    +        this.lookupKey = context.getProperty(LOOKUP_KEY).getValue();
    +        super.onEnabled(context);
    +    }
    +
    +    @Override
    +    public Class<?> getValueType() {
    +        return null;
    --- End diff --
    
    This should return Record.class


> Add a controller service and a lookup service for MongoDB
> ---------------------------------------------------------
>
>                 Key: NIFI-4345
>                 URL: https://issues.apache.org/jira/browse/NIFI-4345
>             Project: Apache NiFi
>          Issue Type: Improvement
>            Reporter: Mike Thomsen
>
> - Create a Controller Service that wraps the functionality of the Mongo driver.
> - Create a lookup service that can return elements based on a query.



--
This message was sent by Atlassian JIRA
(v6.4.14#64029)

Mime
View raw message