ConsoleProducer.java

/**
 * 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 com.leansoft.luxun.console;

import java.io.IOException;
import java.util.Properties;

import com.leansoft.luxun.producer.Producer;
import com.leansoft.luxun.producer.ProducerConfig;
import com.leansoft.luxun.producer.ProducerData;
import com.leansoft.luxun.serializer.StringEncoder;
import com.leansoft.luxun.utils.Closer;

import joptsimple.ArgumentAcceptingOptionSpec;
import joptsimple.OptionParser;
import joptsimple.OptionSet;
import joptsimple.OptionSpec;

/**
 * @author bulldog
 * 
 */
public class ConsoleProducer {

    /**
     * @param args
     * @throws IOException
     */
    public static void main(String[] args) throws Exception {

        OptionParser parser = new OptionParser();
        final ArgumentAcceptingOptionSpec<String> topicOpt = parser.accepts("topic", "REQUIRED: The topic id to produce messages to.")//
                .withRequiredArg().describedAs("topic").ofType(String.class);
        final ArgumentAcceptingOptionSpec<String> brokerListOpt = parser.accepts("broker-list", "REQUIRED: broker list, form: 0:127.0.0.1:9092")//
                .withRequiredArg().describedAs("broker_list").ofType(String.class);
        final ArgumentAcceptingOptionSpec<String> messageReaderOpt = parser.accepts("message-encoder", "The class name of the message encoder")//
                .withRequiredArg().describedAs("encoder_class").ofType(String.class).defaultsTo(LineMessageReader.class.getName());

        //
        OptionSet options = parser.parse(args);
        if (!options.has(brokerListOpt)) {
            System.err.println("Missing required argument broker-list");
            parser.printHelpOn(System.err);
            return;
        }
        ////////////////////////////////////////////////////////////////////
        final Properties props = new Properties();
        checkRequiredArgs(parser, options, topicOpt, brokerListOpt);
        props.put("broker.list", options.valueOf(brokerListOpt));

        final String topic = options.valueOf(topicOpt);
        props.put("serializer.class", StringEncoder.class.getName());

        MessageReader reader = (MessageReader) Class.forName(options.valueOf(messageReaderOpt)).newInstance();
        reader.init(System.in, null);
        final Producer<String, String> producer = new Producer<String, String>(new ProducerConfig(props));
        Runtime.getRuntime().addShutdownHook(new Thread() {

            @Override
            public void run() {
                Closer.closeQuietly(producer);
            }
        });
        //

        String message = null;
        while ((message = reader.readMessage()) != null) {
            if (message.length() == 0) {
                break;
            }
            producer.send(new ProducerData<String, String>(topic, message));
            System.out.println("Message sent : " + message);
        }
    }

    static void checkRequiredArgs(OptionParser parser, OptionSet options, OptionSpec<?>... optionSepcs) throws IOException {
        for (OptionSpec<?> arg : optionSepcs) {
            if (!options.has(arg)) {
                System.err.println("Missing required argument " + arg);
                // parser.formatHelpWith(new MyFormatter());
                parser.printHelpOn(System.err);
                System.exit(1);
            }
        }
    }

}