cc.twittertools.search.api.TrecSearchThriftClient Maven / Gradle / Ivy

Go to download
/**
 * Twitter Tools
 *
 * 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 cc.twittertools.search.api;

import java.util.List;

import javax.annotation.Nullable;

import org.apache.thrift.TException;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;

import cc.twittertools.thrift.gen.TQuery;
import cc.twittertools.thrift.gen.TResult;
import cc.twittertools.thrift.gen.TrecSearch;

import com.google.common.base.Preconditions;

public class TrecSearchThriftClient {
  private final String group;
  private final String token;
  private final String host;
  private final int port;

  public TrecSearchThriftClient(String host, int port,
      @Nullable String group, @Nullable String token) {
    Preconditions.checkNotNull(host);
    Preconditions.checkArgument(port > 0);
    this.group = group;
    this.token = token;
    this.host= host;
    this.port = port;
  }

  public List search(String query, long maxId, int numResults) throws TException {
    TTransport transport = new TSocket(host, port);
    transport.open();

    TrecSearch.Client client = new TrecSearch.Client(new TBinaryProtocol(transport));

    TQuery q = new TQuery();
    q.text = query;
    q.max_id = maxId;
    q.num_results = numResults;

    q.group = group;
    q.token = token;

    List results = client.search(q);
    transport.close();

    return results;
  }
}