in.mayurshah.util.ReadCSV Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of selenium-helper Show documentation

selenium-helper is developed to easily automate test cases using Selenium WebDriver.

The newest version!

package in.mayurshah.util;

import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * This file helps read data from CSV file.
 * @author mayur
 *
 */

public class ReadCSV {
	private String fileName;
	
	/**
	 * Constructor
	 * @param fileName -Enter location of your CSV file
	 */
	public ReadCSV(String fileName) {
		this.fileName = fileName;
	}
	/**
	 * First value is your key and each row is your value.
	 * Size of Hashmap is = no of rows - 1(headers)
	 * 
	 * @return -this will return list of hashmap
	 */
	public List> run() {
		List> mapList = new ArrayList>();
		BufferedReader br = null;
		String line = "";
		String cvsSplitBy = ",";
		try {
			String[] keys;
			br = new BufferedReader(new FileReader(this.fileName));
			HashMap map =null;
			// Read first line from CSV file and use it as key
			if ((line = br.readLine()) != null) {
				keys = line.split(cvsSplitBy);
			} else
				return null;
			// Read each line after first line and use it as values
			while ((line = br.readLine()) != null) {
				map = new HashMap();
				String[] values = line.split(cvsSplitBy);
				for (int i = 0; i < keys.length; i++) {
					// Store key and values to hash map
					map.put(keys[i], values[i]);
				}
				// add hash map to list
				mapList.add(map);
			}

		} catch (FileNotFoundException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			if (br != null) {
				try {
					br.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
		return mapList;
	}
}