org.wso2.cep.uima.demo.TwitterStreamer.java Source code

Introduction

Here is the source code for org.wso2.cep.uima.demo.TwitterStreamer.java

Source

/*
 *
 * Copyright (c) 2005-2010, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 * WSO2 Inc. 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.wso2.cep.uima.demo;

import org.apache.log4j.PropertyConfigurator;
import org.wso2.cep.uima.demo.Util.TwitterConfiguration;
import org.wso2.cep.uima.demo.Util.TwitterConfigurationBuilder;
import org.xml.sax.SAXException;
import twitter4j.FilterQuery;
import twitter4j.Logger;
import twitter4j.TwitterStream;
import twitter4j.TwitterStreamFactory;
import twitter4j.conf.ConfigurationBuilder;

import javax.jms.JMSException;
import javax.xml.parsers.ParserConfigurationException;
import java.io.IOException;

public class TwitterStreamer {

    // API keys for the TwitterStreaming API
    private String consumerKey;
    private String consumerSecret;
    private String accessToken;
    private String accessTokenSecret;

    // Configuration parameters of the ActiveMQ Message Broker
    private String JMSUrl;

    private ConfigurationBuilder cb;
    private TwitterStream twitterStream;
    private FilterQuery filter;

    private final long[] usersToFilter; // userIDs of the users to filter when the streaming tweets

    private static Logger logger = Logger.getLogger(TwitterStreamer.class);

    public static void main(String[] args)
            throws JMSException, IOException, SAXException, ParserConfigurationException {

        String JMSUrl = args[0];
        String topicName = args[1];

        if (JMSUrl.equals("") || topicName.equals("")) {
            System.out.println("Usage ant -DjmsUrl=<JMS_URL> -DtopicName=<TOPIC_NAME>");
            System.exit(0);
        }

        TwitterConfiguration config = TwitterConfigurationBuilder.getTwitterConfiguration();

        // create a streamer object
        TwitterStreamer streamer = new TwitterStreamer(config.getFollowers(), JMSUrl);
        streamer.setConsumerSecret(config.getConsumerSecret());
        streamer.setConsumerKey(config.getConsumerKey());
        streamer.setAccessToken(config.getAccessToken());
        streamer.setAccessTokenSecret(config.getAccessTokenSecret());

        // create the status handler of the stream
        StatusHandler statusHandler = new StatusHandler(JMSUrl, topicName);

        // start streaming for tweets
        streamer.startStream(statusHandler);
    }

    /***
     *
     * @param users long[] userIDs of the users to filer the stream for
     * @param JMSUrl String url of the ActiveMQ Server
     */
    public TwitterStreamer(long[] users, String JMSUrl) {
        this.JMSUrl = JMSUrl;
        this.usersToFilter = users.clone();
        PropertyConfigurator.configure("src/main/resources/log4j.properties");
    }

    public void setConsumerKey(String consumerKey) {
        this.consumerKey = consumerKey;
    }

    public void setConsumerSecret(String consumerSecret) {
        this.consumerSecret = consumerSecret;
    }

    public void setAccessToken(String accessToken) {
        this.accessToken = accessToken;
    }

    public void setAccessTokenSecret(String accessTokenSecret) {
        this.accessTokenSecret = accessTokenSecret;
    }

    /***
     *  Method to set up the API keys for the configuration builder
     */
    private void buildConfiguration() {
        cb = new ConfigurationBuilder();
        logger.debug("Building Configuration");

        if (consumerKey == null || consumerSecret == null || accessToken == null || accessTokenSecret == null) {
            logger.error("Twitter API keys not set properly in twitterConfig.xml");
            throw new NullPointerException("TWitter API Keys not set");
        }

        cb.setDebugEnabled(true).setOAuthConsumerKey(consumerKey).setOAuthConsumerSecret(consumerSecret)
                .setOAuthAccessToken(accessToken).setOAuthAccessTokenSecret(accessTokenSecret);
    }

    /***
     *  Method to Set up the filter for the Streaming API
     */
    private void createFilter() {
        if (usersToFilter == null)
            throw new NullPointerException("User list to follow not set");

        filter = new FilterQuery();
        filter.follow(usersToFilter);

    }

    /***
     * Method to start the streaming of Tweets
     * @param handler
     */
    public void startStream(StatusHandler handler) {

        buildConfiguration();
        TwitterStreamFactory tf = new TwitterStreamFactory(cb.build());
        twitterStream = tf.getInstance();
        twitterStream.addListener(handler);
        createFilter();

        if (filter == null) {
            throw new NullPointerException("Filter Not Set");
        }
        twitterStream.filter(filter);
    }

}