SitestreamExample.java

/**
 * Copyright 2013 Twitter, 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 com.twitter.hbc.example;

import com.google.common.collect.Lists;
import com.twitter.hbc.ClientBuilder;
import com.twitter.hbc.SitestreamController;
import com.twitter.hbc.core.Constants;
import com.twitter.hbc.core.endpoint.SitestreamEndpoint;
import com.twitter.hbc.core.processor.StringDelimitedProcessor;
import com.twitter.hbc.httpclient.BasicClient;
import com.twitter.hbc.httpclient.ControlStreamException;
import com.twitter.hbc.httpclient.auth.Authentication;
import com.twitter.hbc.httpclient.auth.OAuth1;
import com.twitter.hbc.twitter4j.Twitter4jSitestreamClient;
import twitter4j.*;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.LinkedBlockingQueue;

public class SitestreamExample {

  // A no-op SiteStreamsListener
  private SiteStreamsListener listener = new SiteStreamsListener() {
    @Override
    public void onDisconnectionNotice(String line) { }

    @Override
    public void onStatus(long forUser, Status status) { }

    @Override
    public void onDeletionNotice(long forUser, StatusDeletionNotice statusDeletionNotice) { }

    @Override
    public void onFriendList(long forUser, long[] friendIds) { }

    @Override
    public void onFavorite(long forUser, User source, User target, Status status) { }

    @Override
    public void onUnfavorite(long forUser, User source, User target, Status status) { }

    @Override
    public void onFollow(long forUser, User source, User target) { }

    @Override
    public void onUnfollow(long forUser, User source, User target) { }

    @Override
    public void onDirectMessage(long forUser, DirectMessage directMessage) { }

    @Override
    public void onDeletionNotice(long forUser, long dmId, long userId) { }

    @Override
    public void onUserListMemberAddition(long forUser, User source, User target, UserList userList) { }

    @Override
    public void onUserListMemberDeletion(long forUser, User source, User target, UserList userList) { }

    @Override
    public void onUserListSubscription(long forUser, User source, User target, UserList userList) { }

    @Override
    public void onUserListUnsubscription(long forUser, User source, User target, UserList userList) { }

    @Override
    public void onUserListCreation(long forUser, User source, UserList userList) { }

    @Override
    public void onUserListUpdate(long forUser, User source, UserList userList) { }

    @Override
    public void onUserListDeletion(long forUser, User source, UserList userList) { }

    @Override
    public void onUserProfileUpdate(long forUser, User user) { }

    @Override
    public void onBlock(long forUser, User source, User target) { }

    @Override
    public void onUnblock(long forUser, User source, User target) { }

    @Override
    public void onException(Exception e) { }
  };

  public static void main(String[] args) {
    try {
      SitestreamExample sitestreamExample = new SitestreamExample();

      sitestreamExample.run(args[0], args[1], args[2], args[3]);

    } catch (InterruptedException e) {
      System.out.println(e);
    } catch (IOException e) {
      System.out.println(e);
    } catch (ControlStreamException e) {
      System.out.println(e);
    }
  }

  public void run(String consumerKey, String consumerSecret, String token, String tokenSecret)
          throws InterruptedException, ControlStreamException, IOException {
    // Create an appropriately sized blocking queue
    BlockingQueue<String> queue = new LinkedBlockingQueue<String>(10000);

    // Define our endpoint: By default, delimited=length is set (we need this for our processor)
    // and stall warnings are on.
    List<Long> followings = new ArrayList<Long>();
    followings.add(111111111L);
    followings.add(222222222L);

    SitestreamEndpoint endpoint = new SitestreamEndpoint(followings);
    Authentication auth = new OAuth1(consumerKey, consumerSecret, token, tokenSecret);

    // Create a new BasicClient. By default gzip is enabled.
    BasicClient client = new ClientBuilder()
            .hosts(Constants.SITESTREAM_HOST)
            .endpoint(endpoint)
            .authentication(auth)
            .processor(new StringDelimitedProcessor(queue))
            .build();

    // Create an executor service which will spawn threads to do the actual work of parsing the incoming messages and
    // calling the listeners on each message
    int numProcessingThreads = 4;
    ExecutorService service = Executors.newFixedThreadPool(numProcessingThreads);

    // Wrap our BasicClient with the twitter4j client
    Twitter4jSitestreamClient t4jClient = new Twitter4jSitestreamClient(
            client, queue, Lists.newArrayList(listener), service);

    // Establish a connection
    t4jClient.connect();
    for (int threads = 0; threads < numProcessingThreads; threads++) {
      // This must be called once per processing thread
      t4jClient.process();
    }

    Thread.sleep(5000);

    // Create a sitestream controller to issue controlstream requests
    SitestreamController controller = new SitestreamController(auth);

    controller.getFriends(t4jClient.getStreamId(), 12345L);
    controller.addUser(t4jClient.getStreamId(), 987765L);

    client.stop();
  }
}