org.jvirtanen.parity.ticker.TAQFormat Maven / Gradle / Ivy

Go to download
package org.jvirtanen.parity.ticker;

import static java.time.format.DateTimeFormatter.ISO_LOCAL_DATE;
import static org.jvirtanen.lang.Strings.*;

import it.unimi.dsi.fastutil.longs.Long2ObjectArrayMap;
import java.nio.charset.Charset;
import java.time.LocalDate;
import org.jvirtanen.parity.file.taq.TAQ;
import org.jvirtanen.parity.file.taq.TAQWriter;
import org.jvirtanen.parity.top.Side;

class TAQFormat extends MarketDataListener {

    private Long2ObjectArrayMap instruments;

    private TAQ.Quote quote;
    private TAQ.Trade trade;

    private TAQWriter writer;

    public TAQFormat() {
        instruments = new Long2ObjectArrayMap<>();

        quote = new TAQ.Quote();
        trade = new TAQ.Trade();

        String date = ISO_LOCAL_DATE.format(LocalDate.now());

        quote.date = date;
        trade.date = date;

        writer = new TAQWriter(System.out, Charset.defaultCharset());
        writer.flush();
    }

    @Override
    public void bbo(long instrument, long bidPrice, long bidSize, long askPrice, long askSize) {
        quote.timestampMillis = timestampMillis();
        quote.instrument      = instrument(instrument);
        quote.bidPrice        = bidPrice;
        quote.bidSize         = bidSize;
        quote.askPrice        = askPrice;
        quote.askSize         = askSize;

        writer.write(quote);
        writer.flush();
    }

    @Override
    public void trade(long instrument, Side side, long price, long size) {
        trade.timestampMillis = timestampMillis();
        trade.instrument      = instrument(instrument);
        trade.price           = price;
        trade.size            = size;
        trade.side            = side(side);

        writer.write(trade);
        writer.flush();
    }

    private String instrument(long instrument) {
        String cached = instruments.get(instrument);
        if (cached == null) {
            cached = decodeLong(instrument).trim();

            instruments.put(instrument, cached);
        }

        return cached;
    }

    private char side(Side side) {
        switch (side) {
        case BUY:
            return TAQ.BUY;
        case SELL:
            return TAQ.SELL;
        }

        return 0;
    }

}