mongodbutils.Filehandler.java Source code

Introduction

Here is the source code for mongodbutils.Filehandler.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package mongodbutils;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.Date;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.DateUtil;
import org.apache.poi.ss.usermodel.Row;
import org.json.simple.JSONObject;
import org.json.simple.parser.JSONParser;
import org.json.simple.parser.ParseException;

/**
 *
 * @author NBosua
 */
public class Filehandler {

    MongodbConnection mc = null;

    public boolean processFile(String filePath, MongodbConnection mc, String strdbName, String strCollName)
            throws IOException {
        this.mc = mc;

        FileInputStream fileIn = null;
        try {
            fileIn = new FileInputStream(filePath);
            POIFSFileSystem fs = new POIFSFileSystem(fileIn);
            HSSFWorkbook wb = new HSSFWorkbook(fs);
            HSSFSheet sheet = wb.getSheetAt(0);

            Object objReturn = null;

            //Read in first row as field names
            Row rowH = sheet.getRow(sheet.getFirstRowNum());
            String fields[] = new String[sheet.getRow(0).getLastCellNum()];
            for (Cell cell : rowH) {
                objReturn = null;
                objReturn = getCellValue(cell);
                fields[cell.getColumnIndex()] = objReturn.toString();
            }

            //loop thru all cells with values
            int rowcount = 0;
            for (Row row : sheet) {
                if (row.getRowNum() == 0) {
                    continue; //skip first row
                }
                JSONObject obj = new JSONObject();

                for (Cell cell : row) {
                    if (fields.length < cell.getColumnIndex()) {
                        continue; //only export column if we have header set
                    }
                    objReturn = null;
                    objReturn = getCellValue(cell);
                    if (!objReturn.toString().equals("")) {
                        if (objReturn instanceof Double) {
                            obj.put(fields[cell.getColumnIndex()], objReturn);

                        } else if (objReturn instanceof String) {
                            if (objReturn.toString().contains("$date")) {
                                JSONParser parser = new JSONParser();
                                try {
                                    obj.put(fields[cell.getColumnIndex()], parser.parse(objReturn.toString()));
                                } catch (ParseException ex) {
                                    Logger.getLogger(Filehandler.class.getName()).log(Level.SEVERE, null, ex);
                                }
                            } else {
                                obj.put(fields[cell.getColumnIndex()], objReturn);
                            }
                        }
                    }
                }
                rowcount += 1;
                mc.insertJSON(strdbName, strCollName, obj.toJSONString());
            }

            return true;
        } catch (FileNotFoundException ex) {
            Logger.getLogger(Filehandler.class.getName()).log(Level.SEVERE, null, ex);
        } catch (IOException ex) {
            Logger.getLogger(Filehandler.class.getName()).log(Level.SEVERE, null, ex);
        } catch (Exception e) {
            Logger.getLogger(Filehandler.class.getName()).log(Level.SEVERE, null, e);
        } finally {
            if (fileIn != null) {
                fileIn.close();
            }
        }
        return false;
    }

    private Object getCellValue(Cell cell) {

        switch (cell.getCellType()) {
        case Cell.CELL_TYPE_STRING:
            return cell.getRichStringCellValue().getString();
        case Cell.CELL_TYPE_NUMERIC:
            if (DateUtil.isCellDateFormatted(cell)) {
                Date dt = cell.getDateCellValue();

                JSONObject obj = new JSONObject();
                obj.put("$date", dt.getTime());
                return obj.toString();

                //return "" + cell.getDateCellValue();
            } else {
                return cell.getNumericCellValue();
            }
        case Cell.CELL_TYPE_BOOLEAN:
            return "" + cell.getBooleanCellValue();
        case Cell.CELL_TYPE_FORMULA:
            return cell.getCellFormula();
        }
        return "";
    }

}