удалить столбец из csv в Groovy / java

Aug 18 2020

Я использую запрос Select для извлечения данных из базы данных и создания CSV, который работает нормально.

В моем FDataклассе у меня есть метод ниже, где я делаю форматирование для значения ячейки в csv. У меня есть столбец ATYPEв моем запросе выбора, который я не хочу писать в csv или, возможно, после написания просто очистите полное ATYPEполе столбца из csv, чтобы его нельзя было отобразить.

Есть ли способ сделать это в методе ниже, чтобы выполнить эту логику?

private static void IFile(File output, Sql sql, String query, List<String> columns) {
        output.withWriter { writer ->            
            writer.writeLine(columns.join(CSV_SEPARATOR))           
            sql.eachRow(query) { rec ->
                try {
                    writer.writeLine(columns.collect { columnName ->
                        def cell = rec[columnName]
                        // cell data processing for CSV output...If cell is empty then replace with "" else do the other formating
                        if (cell != null) {                                
                            if (cell.properties["class"] == Timestamp.class)
                                return IData.OUT_DATE_FORMAT.format(cell)                              
                            
                        }
                        else {
                            return ""
                        }
                        return cell
                    }.join(CSV_SEPARATOR))
                }                 }
            sql.close()
        }
    }

Ответы

1 LiveandLetLive Aug 19 2020 at 08:36

В приведенном ниже коде точный синтаксис может быть неправильным (я только начинаю разбираться в Groovy), но вы поймете, как это решить:

  1. Отфильтруйте имена столбцов перед присоединением, т.е.

    writer.writeLine(columns.findAll({ it != 'ATYPE'}).join(CSV_SEPARATOR)) 
    
  2. Отфильтруйте данные перед записью, т.е.

     writer.writeLine(columns.findAll({ it != 'ATYPE'}).collect { columnName ->
         //...
     }