package com.ttdev.table;

import java.util.Comparator;

public class PhoneBookEntryComparator implements Comparator {
	private String columnName;

	public PhoneBookEntryComparator(String columnName) {
		this.columnName = columnName;
	}

	public int compare(Object obj1, Object obj2) {
		PhoneBookEntry entry1 = (PhoneBookEntry) obj1;
		PhoneBookEntry entry2 = (PhoneBookEntry) obj2;
		if (columnName.equals("id")) {
			return entry1.getId() - entry2.getId();
		} else if (columnName.equals("firstName")) {
			return entry1.getFirstName().compareTo(entry2.getFirstName());
		} else if (columnName.equals("lastName")) {
			return entry1.getName().getLastName().compareTo(
					entry2.getName().getLastName());
		}
		return entry1.getTelNo().compareTo(entry2.getTelNo());
	}
}