framework.retrieval.engine.index.create.impl.file.parse.ExcelFileContentParser.java Source code

Introduction

Here is the source code for framework.retrieval.engine.index.create.impl.file.parse.ExcelFileContentParser.java

Source

/**
 * Copyright 2010 
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package framework.retrieval.engine.index.create.impl.file.parse;

import java.io.FileInputStream;
import java.io.InputStream;

import org.apache.commons.logging.Log;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;

import framework.base.snoic.base.util.StringClass;
import framework.retrieval.engine.common.RetrievalUtil;
import framework.retrieval.engine.index.doc.file.internal.RFileDocument;

/**
 * Excel?
 * @author 
 *
 */
public class ExcelFileContentParser extends AbstractFileContentParser {

    private Log log = RetrievalUtil.getLog(this.getClass());

    public String getContent(RFileDocument document, String charsetName) {

        InputStream fileInputStream = null;
        StringBuffer content = new StringBuffer();

        try {
            fileInputStream = new FileInputStream(document.getFile());
            HSSFWorkbook workbook = new HSSFWorkbook(fileInputStream);// Excel

            if (workbook != null) {
                int numSheetsNumber = workbook.getNumberOfSheets();
                for (int numSheets = 0; numSheets < numSheetsNumber; numSheets++) {
                    HSSFSheet aSheet = workbook.getSheetAt(numSheets);// sheet
                    if (aSheet != null) {
                        int lastRowNum = aSheet.getLastRowNum();
                        for (int rowNumOfSheet = 0; rowNumOfSheet <= lastRowNum; rowNumOfSheet++) {
                            HSSFRow aRow = aSheet.getRow(rowNumOfSheet); // 
                            if (aRow != null) {
                                int lastCellNum = aRow.getLastCellNum();
                                for (int cellNumOfRow = 0; cellNumOfRow <= lastCellNum; cellNumOfRow++) {
                                    HSSFCell aCell = aRow.getCell(cellNumOfRow);// 
                                    if (aCell != null) {
                                        int cellType = aCell.getCellType();
                                        if (cellType == HSSFCell.CELL_TYPE_STRING) {
                                            String value = StringClass.getString(aCell.getStringCellValue());
                                            content.append(value);
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        } catch (Exception e) {
            RetrievalUtil.errorLog(log, document.getFile().getAbsolutePath(), e);
        } finally {
            if (fileInputStream != null) {
                try {
                    fileInputStream.close();
                } catch (Exception e) {

                }
            }
        }

        return content.toString();

    }

}