Subversion Repositories Integrator Subversion

package br.com.kronus.ibkr.history;

import java.io.BufferedWriter;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.TimeUnit;

import com.ib.client.Bar;
import com.ib.client.Contract;
import com.ib.client.TagValue;

import br.com.kronus.ibkr.api.IbkrClient;

public class IbkrHistoricalDataService {

    private final IbkrClient ibkr;

    public IbkrHistoricalDataService(IbkrClient ibkr) {
        this.ibkr = ibkr;
    }

    /**
     * Solicita histórico para IBKR e grava tudo em CSV.
     *
     * @param contract        contrato IBKR
     * @param endDateTime     "" para agora ou "yyyyMMdd HH:mm:ss"
     * @param durationStr     "2 D", "5 D", "1 W", "1 M", etc.
     * @param barSizeSetting  "1 min", "5 mins", "15 mins", "1 day", etc.
     * @param whatToShow      "TRADES", "MIDPOINT", etc.
     */
    public void baixarHistoricoParaCsv(
            int reqId,
            Contract contract,
            String endDateTime,
            String durationStr,
            String barSizeSetting,
            String whatToShow,
            Path arquivoCsv
    ) throws Exception {

        if (!Files.exists(arquivoCsv.getParent())) {
            Files.createDirectories(arquivoCsv.getParent());
        }

        try (BufferedWriter writer = Files.newBufferedWriter(arquivoCsv,
                java.nio.file.StandardOpenOption.CREATE,
                java.nio.file.StandardOpenOption.TRUNCATE_EXISTING)) {

            writer.write("data;hora;open;high;low;close;volume;wap;count");
            writer.newLine();

            CompletableFuture<List<Bar>> fut = ibkr.prepararFutureHistorico(reqId);

            ibkr.getClient().reqHistoricalData(
                    reqId,
                    contract,
                    endDateTime,
                    durationStr,
                    barSizeSetting,
                    whatToShow,
                    1,
                    2,
                    false,
                    (List<TagValue>) null
            );

            List<Bar> bars = fut.get(60, TimeUnit.SECONDS);

            for (Bar b : bars) {
                String timeStr = b.time(); // ex: 20250125 10:05:00
                String dataStr = timeStr.substring(0, 8);
                String horaStr = timeStr.substring(9);

                String yyyy = dataStr.substring(0, 4);
                String mm   = dataStr.substring(4, 6);
                String dd   = dataStr.substring(6, 8);
                String dataFormatada = yyyy + "-" + mm + "-" + dd;

                String linha = String.join(";",
                        dataFormatada,
                        horaStr,
                        String.valueOf(b.open()),
                        String.valueOf(b.high()),
                        String.valueOf(b.low()),
                        String.valueOf(b.close()),
                        String.valueOf(b.volume()),
                        String.valueOf(b.wap()),
                        String.valueOf(b.count())
                );
                writer.write(linha);
                writer.newLine();
            }
            writer.flush();
        }
    }
}