WebSVN - Integrator Subversion - Diff - Rev 767 and 779 - /projeto-trade/stocklifepro/src/main/java/br/com/sl/core/ExcelProfitDataProvider.java

 package br.com.sl.core;
 import java.io.FileInputStream;
 import java.io.IOException;
 import java.math.BigDecimal;
+import java.nio.file.Files;
 import java.nio.file.Path;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.LocalTime;
 import java.time.format.DateTimeFormatter;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
 import java.util.Locale;
 import java.util.Map;
+import java.util.stream.Stream;
 import org.apache.poi.EncryptedDocumentException;
 import org.apache.poi.ss.usermodel.Cell;
 import org.apache.poi.ss.usermodel.Row;
 import org.apache.poi.ss.usermodel.Sheet;
     private final int colAsset = 0;
     private final int colDate  = 1;
     private final int colTime  = 2;
     private final int colLast  = 3;
-    // formatos da data e hora, conforme Excel
-    private final DateTimeFormatter fmtData = DateTimeFormatter.ofPattern("dd/MM/yyyy");
-    private final DateTimeFormatter fmtHora = DateTimeFormatter.ofPattern("HH:mm:ss");
-    private static final DateTimeFormatter STRING_DATA_FORMATTER = DateTimeFormatter.ofPattern("dd/MM/yyyy HH:mm:ss");
-    private static final DateTimeFormatter STRING_DATE_TIME_FORMATTER = DateTimeFormatter.ofPattern("dd-MM-yyyy HH:mm:ss");
+    private static final DateTimeFormatter DATE_FORMAT = DateTimeFormatter.ofPattern("dd/MM/yyyy HH:mm:ss");
+    private static final DateTimeFormatter TIME_FORMAT = DateTimeFormatter.ofPattern("dd-MM-yyyy HH:mm:ss");
     public ExcelProfitDataProvider(Path excelFile, String sheetName) {
         this.excelFile = excelFile;
         this.sheetName = sheetName;
+    }
         // TODO Auto-generated method stub
         return null;
+    }
     /**
-     * Lê o arquivo Excel dentro do resources.
-     * Exemplo de uso:
-     *    lerCandles("/dados/Dados Trade 20251117.xlsx");
-     * @throws IOException
+     * Lê um ou vários arquivos:
+     * - Se excelFile for um arquivo: lê apenas esse arquivo (.xlsx, .xls ou .csv)
+     * - Se excelFile for uma pasta: lê todos os .xlsx, .xls e .csv da pasta.
      */
     public List<Candle> lerCandles() throws IOException {
         List<Candle> candles = new ArrayList<>();
-        try (FileInputStream fis = new FileInputStream(excelFile.toFile()); Workbook workbook = new XSSFWorkbook(fis)) {
-            int numberOfSheets = workbook.getNumberOfSheets();
+        if (Files.isDirectory(excelFile)) {
+            // Percorre todos os arquivos da pasta
+            try (Stream<Path> stream = Files.list(excelFile)) {
+                stream
+                    .filter(Files::isRegularFile)
+                    .filter(p -> {
+                        String name = p.getFileName().toString().toLowerCase(Locale.ROOT);
+                        return name.endsWith(".xlsm") || name.endsWith(".xlsx") || name.endsWith(".xls") || name.endsWith(".csv");
+                    })
+                    .forEach(path -> {
+                        try {
+                            lerCandlesDeArquivo(path, candles);
+                        } catch (IOException e) {
+                            // aqui você pode trocar por log
+                            e.printStackTrace();
+                        }
+                    });
+            }
+        } else {
+            // Apenas um arquivo
+            lerCandlesDeArquivo(excelFile, candles);
+        }
+        return adicionarContadores(candles);
+    }
+    /**
+     * Decide se o arquivo é Excel ou CSV e delega para o método correto.
+     */
+    private void lerCandlesDeArquivo(Path arquivo, List<Candle> candles) throws IOException {
+        String name = arquivo.getFileName().toString().toLowerCase(Locale.ROOT);
+        if (name.endsWith(".xlsm") || name.endsWith(".xlsx") || name.endsWith(".xls")) {
+            lerCandlesDeArquivoExcel(arquivo, candles);
+        } else if (name.endsWith(".csv")) {
+//            lerCandlesDeArquivoCsv(arquivo, candles);
+        } else {
+            // Tipo não suportado, ignora ou loga
+        }
+    }
+    /**
+     * Lê os candles de UM arquivo Excel e adiciona na lista passada.
+     * (É basicamente o seu código original, só movido para cá)
+     */
+    private void lerCandlesDeArquivoExcel(Path arquivoExcel, List<Candle> candles) throws IOException {
+        try (FileInputStream fis = new FileInputStream(arquivoExcel.toFile());
+             Workbook workbook = new XSSFWorkbook(fis)) {
+            int numberOfSheets = workbook.getNumberOfSheets();
+            // LER APENAS OS ATIVOS
+            List<String> ativos = new ArrayList<String>();
             for (int i = 0; i < numberOfSheets; i++) {
                 Sheet sheet = workbook.getSheetAt(i);
                 if (sheet.getSheetName().equals(sheetName)) {
                         boolean firstRow = true;
                             // 4 = Abertura
                             // 5 = Máxima
                             // 6 = Mínima
                             // 7 = Fechamento
-                            Cell contadorCell   = row.getCell(0);
-                            Cell ativoCell      = row.getCell(1);
-                            Cell dataCell       = row.getCell(2);
-                            Cell horaCell       = row.getCell(3);
-                            Cell aberturaCell   = row.getCell(4);
-                            Cell maximaCell     = row.getCell(5);
-                            Cell minimaCell     = row.getCell(6);
-                            Cell fechamentoCell = row.getCell(7);
-                            Cell finalizadoCell = row.getCell(8);
-                            if (!ExcelDataUtils.isNumeric(aberturaCell) || !ExcelDataUtils.isNumeric(maximaCell)
-                                        || !ExcelDataUtils.isNumeric(minimaCell) || !ExcelDataUtils.isNumeric(fechamentoCell)) {
-                                continue;
-                            }
-                            Integer contador = BigDecimal.valueOf(contadorCell.getNumericCellValue()).intValue();
-                            String ativoDescricao = ativoCell.getStringCellValue();
-                            LocalDate data = ExcelDataUtils.lerData(dataCell);
-                            LocalTime hora = ExcelDataUtils.lerHora(horaCell);
-                            LocalDateTime dataHora = LocalDateTime.of(data, hora);
-                            BigDecimal abertura = BigDecimal.valueOf(aberturaCell.getNumericCellValue());
-                            BigDecimal topo = BigDecimal.valueOf(maximaCell.getNumericCellValue());
-                            BigDecimal fundo = BigDecimal.valueOf(minimaCell.getNumericCellValue());
-                            BigDecimal fechamento = BigDecimal.valueOf(fechamentoCell.getNumericCellValue());
-                            String finalizado = finalizadoCell.getStringCellValue();
-                            if (finalizado.equals("S")) {
-                                Candle candle = new Candle(contador, ativoDescricao, dataHora, abertura, topo, fundo, fechamento, TipoPeriodoCandle.M1.getValor());
-                                candles.add(candle);
+                            String ativoCell      = row.getCell(1).getStringCellValue();
+                            if (!ativos.contains(ativoCell)) {
+                                ativos.add(ativoCell);
+                            }
+                        }
+                }
+            }
+            for (String ativo : ativos) {
+                    for (int i = 0; i < numberOfSheets; i++) {
+                        Sheet sheet = workbook.getSheetAt(i);
+                        if (sheet.getSheetName().equals(sheetName)) {
+                                boolean firstRow = true;
+                                for (Row row : sheet) {
+                                    if (firstRow) {
+                                        firstRow = false; // pula o cabeçalho
+                                        continue;
+                                    }
+                                    // 0 = Contador
+                                    // 1 = Ativo
+                                    // 2 = Dia
+                                    // 3 = Hora
+                                    // 4 = Abertura
+                                    // 5 = Máxima
+                                    // 6 = Mínima
+                                    // 7 = Fechamento
+                                    Cell contadorCell   = row.getCell(0);
+                                    Cell ativoCell      = row.getCell(1);
+                                    Cell dataCell       = row.getCell(2);
+                                    Cell horaCell       = row.getCell(3);
+                                    Cell aberturaCell   = row.getCell(4);
+                                    Cell maximaCell     = row.getCell(5);
+                                    Cell minimaCell     = row.getCell(6);
+                                    Cell fechamentoCell = row.getCell(7);
+                                    Cell finalizadoCell = row.getCell(8);
+                                    String ativoDescricao = ativoCell.getStringCellValue();
+                                    if (!ExcelDataUtils.isNumeric(aberturaCell) || !ExcelDataUtils.isNumeric(maximaCell)
+                                                || !ExcelDataUtils.isNumeric(minimaCell) || !ExcelDataUtils.isNumeric(fechamentoCell)
+                                                || !ativo.equals(ativoDescricao)) {
+                                        continue;
+                                    }
+                                    Integer contador = BigDecimal.valueOf(contadorCell.getNumericCellValue()).intValue();
+                                    LocalDate data = ExcelDataUtils.lerData(dataCell);
+                                    LocalTime hora = ExcelDataUtils.lerHora(horaCell);
+                                    LocalDateTime dataHora = LocalDateTime.of(data, hora);
+                                    BigDecimal abertura = BigDecimal.valueOf(aberturaCell.getNumericCellValue());
+                                    BigDecimal topo = BigDecimal.valueOf(maximaCell.getNumericCellValue());
+                                    BigDecimal fundo = BigDecimal.valueOf(minimaCell.getNumericCellValue());
+                                    BigDecimal fechamento = BigDecimal.valueOf(fechamentoCell.getNumericCellValue());
+                                    String finalizado = finalizadoCell.getStringCellValue();
+                                    if (finalizado.equals("S")) {
+                                        Candle candle = new Candle(contador, ativoDescricao, dataHora, abertura, topo, fundo, fechamento, TipoPeriodoCandle.M1.getValor());
+                                        candles.add(candle);
+                                    }
+                                }
+                        }
+                    }
+            }
         } catch (EncryptedDocumentException e) {
-                        // TODO Auto-generated catch block
-                        e.printStackTrace();
-                }
-        return candles;
+            e.printStackTrace();
+        }
+    }
+    /**
+     * Lê os candles de UM arquivo CSV e adiciona na lista passada.
+     * Espera layout:
+     * 0 = Ativo
+     * 1 = Dia  (dd/MM/yyyy)
+     * 2 = Hora (HH:mm:ss)
+     * 3 = Abertura
+     * 4 = Máxima
+     * 5 = Mínima
+     * 6 = Fechamento
+     */
+    private void lerCandlesDeArquivoCsv(Path arquivoCsv, List<Candle> candles) throws IOException {}
+    private boolean isNumericString(String value) {
+        if (value == null) return false;
+        String normalized = value.trim().replace(".", "").replace(",", ".");
+        if (normalized.isEmpty()) return false;
+        try {
+            new BigDecimal(normalized);
+            return true;
+        } catch (NumberFormatException e) {
+            return false;
+        }
+    }
     public static List<Candle> inverterLista(List<Candle> candles) {
         List<Candle> invertida = new ArrayList<>(candles);
         Collections.reverse(invertida);
         return invertida;
+    }
+    // Contador separado para cada ativo
     public static List<Candle> adicionarContadores(List<Candle> candles) {
-        Integer contador = 1;
-        List<Candle> comContadores = new ArrayList<>();
+        Map<String, Integer> contadorPorAtivo = new java.util.HashMap<>();
+        List<Candle> comContadores = new ArrayList<>();
         for (Candle candle : candles) {
-                candle.setContadorCandle(contador);
-                comContadores.add(candle);
-                contador++;
+            String ativo = candle.getNomeAtivo();
+            Integer contadorAtual = contadorPorAtivo.get(ativo);
+            if (contadorAtual == null) {
+                contadorAtual = 0;
+            }
+            contadorAtual++;
+            contadorPorAtivo.put(ativo, contadorAtual);
+            candle.setContadorCandle(contadorAtual);
+            comContadores.add(candle);
+        }
         return comContadores;
+    }
     /**

Subversion Repositories Integrator Subversion

(root)/projeto-trade/stocklifepro/src/main/java/br/com/sl/core/ExcelProfitDataProvider.java - Rev 767 → 779