com.precioustech.fxtrading.tradingbot.social.twitter.tweethandler.AbstractFXTweetHandler.java Source code

Introduction

Here is the source code for com.precioustech.fxtrading.tradingbot.social.twitter.tweethandler.AbstractFXTweetHandler.java

Source

/*
 *  Copyright 2015 Shekhar Varshney
 *
 *  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.precioustech.fxtrading.tradingbot.social.twitter.tweethandler;

import java.util.Collection;
import java.util.List;

import org.apache.log4j.Logger;
import org.joda.time.DateTime;
import org.joda.time.format.DateTimeFormat;
import org.joda.time.format.DateTimeFormatter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.social.twitter.api.SearchResults;
import org.springframework.social.twitter.api.Tweet;
import org.springframework.social.twitter.api.Twitter;

import com.google.common.base.Predicate;
import com.google.common.collect.Lists;
import com.precioustech.fxtrading.helper.ProviderHelper;
import com.precioustech.fxtrading.instrument.InstrumentService;
import com.precioustech.fxtrading.tradingbot.social.twitter.CloseFXTradeTweet;
import com.precioustech.fxtrading.tradingbot.social.twitter.NewFXTradeTweet;
import com.precioustech.fxtrading.utils.TradingUtils;

public abstract class AbstractFXTweetHandler<T> implements FXTweetHandler<T> {

    @Autowired
    protected Twitter twitter;
    @Autowired
    protected InstrumentService<T> instrumentService;
    @Autowired
    protected ProviderHelper providerHelper;

    protected DateTime startTime = DateTime.now();
    protected String startTimeAsStr;
    protected volatile Long lastTweetId = null;
    protected final String userId;
    protected static final String CLOSED = "Closed";
    protected static final String BOUGHT = "Bought";
    protected static final String SOLD = "Sold";
    protected static final String TP = "TP";
    protected static final String SL = "SL";
    protected static final Logger LOG = Logger.getLogger(AbstractFXTweetHandler.class);
    static final String FROM_USER_SINCE_TIME_TWITTER_QRY_TMPL = "from:%s since:%s";
    static final String FROM_USER_SINCE_ID_TWITTER_QRY_TMPL = "from:%s since_id:%d";

    protected AbstractFXTweetHandler(String userId) {
        this.userId = userId;
        setStartTimeAsStr();
    }

    private void setStartTimeAsStr() {
        DateTimeFormatter formatter = DateTimeFormat.forPattern("yyyy-MM-dd");
        startTimeAsStr = formatter.print(this.startTime);
    }

    protected abstract NewFXTradeTweet<T> parseNewTrade(String tokens[]);

    protected abstract CloseFXTradeTweet<T> parseCloseTrade(String tokens[]);

    public void setStartTime(DateTime startTime) {
        this.startTime = startTime;
        setStartTimeAsStr();
    }

    @Override
    public String getUserId() {
        return userId;
    }

    private class TweetPredicate implements Predicate<Tweet> {

        @Override
        public boolean apply(Tweet input) {
            return startTime.isBefore(input.getCreatedAt().getTime());
        }

    }

    @Override
    public Collection<Tweet> findNewTweets() {
        SearchResults results = null;
        if (lastTweetId == null) {// find new tweets since the start of the app
            synchronized (this) {
                if (lastTweetId == null) {// double check locking
                    results = twitter.searchOperations().search(
                            String.format(FROM_USER_SINCE_TIME_TWITTER_QRY_TMPL, getUserId(), this.startTimeAsStr));
                    TweetPredicate predicate = new TweetPredicate();
                    List<Tweet> filteredTweets = Lists.newArrayList();
                    for (Tweet tweet : results.getTweets()) {
                        if (predicate.apply(tweet)) {
                            if (lastTweetId == null) {/*take the first one as the tweets are in reverse chronological order */
                                this.lastTweetId = tweet.getId();
                            }
                            filteredTweets.add(tweet);
                        }
                    }
                    return filteredTweets;
                }

            }

        }
        results = twitter.searchOperations()
                .search(String.format(FROM_USER_SINCE_ID_TWITTER_QRY_TMPL, getUserId(), lastTweetId));
        List<Tweet> tweets = results.getTweets();
        if (!TradingUtils.isEmpty(tweets)) {
            lastTweetId = tweets.get(0).getId();
        }
        return tweets;

    }
}