org.apache.camel.component.kudu.KuduEndpoint Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of connector-kudu Show documentation
Parent project for Camel Kudu connector
There is a newer version: 1.13.2
/*
 * Copyright (C) 2016 Red Hat, Inc.
 *
 * 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.
 */

package org.apache.camel.component.kudu;

import org.apache.camel.Consumer;
import org.apache.camel.Processor;
import org.apache.camel.Producer;
import org.apache.camel.impl.DefaultEndpoint;
import org.apache.camel.spi.Metadata;
import org.apache.camel.spi.UriEndpoint;
import org.apache.camel.spi.UriParam;
import org.apache.camel.spi.UriPath;
import org.apache.kudu.client.KuduClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Represents a Kudu endpoint. A kudu endpoint allows you to interact with
 * Apache Kudu,  a free and open source
 * column-oriented data store of the Apache Hadoop ecosystem.
 */
@UriEndpoint(firstVersion = "2.23.0",
        scheme = "kudu",
        title = "Apache Kudu", syntax = "kudu:type",
        consumerClass = KuduConsumer.class,
        label = "database,iot")
public class KuduEndpoint extends DefaultEndpoint {

    private static final Logger LOG = LoggerFactory.getLogger(KuduEndpoint.class);
    private KuduClient kuduClient;

    @UriPath
    @Metadata(required = "true")
    private String type;

    @UriParam
    @Metadata(required = "true")
    private String host;

    @UriParam
    @Metadata(required = "true")
    private String port;

    @UriParam(defaultValue = KuduDbOperations.INSERT)
    private String operation = KuduDbOperations.INSERT;

    @UriParam
    private String tableName;

    public KuduEndpoint(String uri, KuduComponent component) {
        super(uri, component);
    }

    @Override
    protected void doStart() throws Exception {
        LOG.debug("Connection: {}, {}", host, port);
        kuduClient = new KuduClient.KuduClientBuilder(host + ":" + port).build();
        LOG.debug("Resolved the host with the name {} as {}", host, kuduClient);
        super.doStart();
    }

    @Override
    protected void doStop() throws Exception {
        try {
            kuduClient.shutdown();
        } catch (Exception e) {
            LOG.error("Unable to shutdown kudu client", e);
        }

        super.doStop();
    }

    @Override
    public Producer createProducer() throws Exception {
        return new KuduProducer(this);
    }

    @Override
    public Consumer createConsumer(Processor processor) throws Exception {
        return new KuduConsumer(this, processor);
    }

    @Override
    public boolean isSingleton() {
        return true;
    }

    /**
     * Kudu master to connect to
     */
    public void setHost(String host) {
        this.host = host;
    }

    public String getHost() {
        return host;
    }

    public KuduClient getKuduClient() {
        return kuduClient;
    }

    /**
     * Set the client to connect to a kudu resource
     *
     * @param kuduClient
     */
    public void setKuduClient(KuduClient kuduClient) {
        this.kuduClient = kuduClient;
    }

    public String getTableName() {
        return tableName;
    }

    /**
     * The name of the table where the rows are stored
     *
     * @param tableName
     */
    public void setTableName(String tableName) {
        this.tableName = tableName;
    }

    public String getOperation() {
        return operation;
    }

    /**
     * What kind of operation is to be performed in the table
     *
     * @param operation
     */
    public void setOperation(String operation) {
        this.operation = operation;
    }

    public String getPort() {
        return port;
    }

    /**
     * Port where kudu service is listening
     *
     * @param port
     */
    public void setPort(String port) {
        this.port = port;
    }

    public String getType() {
        return type;
    }

    /**
     * Kudu type
     *
     * @param type
     */
    public void setType(String type) {
        this.type = type;
    }
}