com.ksa.myanmarlottery.service.parser.CSVFileParser.java Source code

Introduction

Here is the source code for com.ksa.myanmarlottery.service.parser.CSVFileParser.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 com.ksa.myanmarlottery.service.parser;

import com.ksa.myanmarlottery.model.Prize;
import com.ksa.myanmarlottery.model.Result;
import com.ksa.myanmarlottery.util.CharacterMap;
import com.ksa.myanmarlottery.util.ConstantUtil;
import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.List;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVRecord;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import sun.security.pkcs.ParsingException;

/**
 *
 * @author Kyawswa
 */
@Component
public class CSVFileParser implements FileParser {

    private final Log log = LogFactory.getLog(CSVFileParser.class);

    private final String OLD_LOTTERY_TYPE = String.valueOf(ConstantUtil.OLD_LOTTERY_TYPE); // 200 lottery for 1
    private final String NEW_LOTTERY_TYPE = String.valueOf(ConstantUtil.NEW_LOTTERY_TYPE); // 500 lottery for 2

    @Autowired
    private CharacterMap charMap;

    public List<Result> getResult(InputStream in) throws FileNotFoundException, IOException, ParseException {
        Iterable<CSVRecord> records = CSVFormat.RFC4180
                .parse(new BufferedReader(new InputStreamReader(in, "UTF-8")));
        int i = 0;
        List<Result> resultList = new ArrayList<>();
        List<Prize> prizes = null;
        for (CSVRecord record : records) {
            if (OLD_LOTTERY_TYPE.equals(record.get(0))) { // for lottery type result
                Result result = new Result();
                result.setType(ConstantUtil.OLD_LOTTERY_TYPE);
                result.setNumberOfTimes(Integer.valueOf(record.get(1)));
                result.setResultFor(new SimpleDateFormat("dd/MM/yyyy").parse(record.get(2)));
                result.setDataProvider(record.get(3));
                result.setCompanyName(record.get(4));
                prizes = new ArrayList<Prize>();
                result.setPrizes(prizes);
                resultList.add(result);

            } else if (NEW_LOTTERY_TYPE.equals(record.get(0))) {
                Result result = new Result();
                result.setType(ConstantUtil.NEW_LOTTERY_TYPE);
                result.setNumberOfTimes(Integer.valueOf(record.get(1)));
                result.setResultFor(new SimpleDateFormat("dd/MM/yyyy").parse(record.get(2)));
                result.setDataProvider(record.get(3));
                result.setCompanyName(record.get(4));

                prizes = new ArrayList<Prize>();
                result.setPrizes(prizes);
                resultList.add(result);

            } else {
                // check validation for character.
                String character = record.get(0);
                String value = charMap.get(character);
                if (value == null) {
                    throw new ParseException("Character is Not valid at Row " + i + " column:" + 0, 400);
                }
                prizes.add(new Prize(record.get(0), Integer.parseInt(record.get(1)), record.get(2), record.get(4)));
            }
            i++;
        }
        log.info("Total rows after parsing. " + i);
        return resultList;
    }
}