SociopathicPixel commented on a change in pull request #230: dev/add datatypes to excel columns URL: https://github.com/apache/metamodel/pull/230#discussion_r329470830
########## File path: excel/src/main/java/org/apache/metamodel/excel/DefaultSpreadsheetReaderDelegate.java ########## @@ -167,50 +167,109 @@ private MutableTable createTable(final Workbook wb, final Sheet sheet) { break; } } - if (hasColumns) { - createColumns(table, wb, row); + createColumns(table, wb, row, columnTypes); } } - return table; } + private ColumnType[] getColumnTypes(final Sheet sheet, final Row row) { + final Iterator<Row> data = ExcelUtils.getRowIterator(sheet, _configuration, false); + final int rowLength = row.getLastCellNum(); + int eagerness = 1000; + final ColumnType[] columnTypes = new ColumnType[rowLength]; + + while (data.hasNext() && eagerness-- > 0) { + Row currentRow = data.next(); + for (int index = 0; index < rowLength; index++) { + if (currentRow.getLastCellNum() == 0) { + continue; + } + if (currentRow.getCell(index) == null) { + checkColumnType(ColumnType.STRING, columnTypes, index); + } else { + CellType cellType = currentRow.getCell(index).getCellType(); + switch (cellType) { + case NUMERIC: + if (DateUtil.isCellDateFormatted(currentRow.getCell(index))) { + checkColumnType(ColumnType.DATE, columnTypes, index); + } else { + checkColumnType((currentRow.getCell(index).getNumericCellValue() % 1 == 0) + ? ColumnType.INTEGER : ColumnType.DOUBLE, columnTypes, index); + } + break; + case BOOLEAN: + checkColumnType(ColumnType.BOOLEAN, columnTypes, index); + break; + case ERROR: + // fall through + break; + case _NONE: + // fall through + case STRING: + // fall through + case FORMULA: + // fall through + case BLANK: + checkColumnType(ColumnType.STRING, columnTypes, index); + break; + } + } + } + } + return columnTypes; + } + + private void checkColumnType(final ColumnType columnType, final ColumnType[] columnTypes, int index) { + if (columnTypes[index] != null) { + if (!columnTypes[index].equals(ColumnType.STRING) && !columnTypes[index].equals(columnType)) { + columnTypes[index] = ColumnType.STRING; + } + } else { + columnTypes[index] = columnType; + } + } + /** * Builds columns based on row/cell values. * * @param table * @param wb * @param row */ - private void createColumns(MutableTable table, Workbook wb, Row row) { + private void createColumns(MutableTable table, Workbook wb, Row row, ColumnType[] columTypes) { if (row == null) { logger.warn("Cannot create columns based on null row!"); return; } final short rowLength = row.getLastCellNum(); - final int offset = getColumnOffset(row); - // build columns based on cell values. - try (final ColumnNamingSession columnNamingSession = _configuration.getColumnNamingStrategy() + try (final ColumnNamingSession columnNamingSession = _configuration + .getColumnNamingStrategy() .startColumnNamingSession()) { for (int j = offset; j < rowLength; j++) { final Cell cell = row.getCell(j); final String intrinsicColumnName = ExcelUtils.getCellValue(wb, cell); final ColumnNamingContext columnNamingContext = new ColumnNamingContextImpl(table, intrinsicColumnName, j); final String columnName = columnNamingSession.getNextColumnName(columnNamingContext); - final Column column = new MutableColumn(columnName, ColumnType.VARCHAR, table, j, true); + Column column = null; Review comment: done with next commit ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services