traveldept.java

旅游自助系统

package org.tshs.entity;

import java.sql.SQLException;
import java.sql.Types;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Vector;

import org.tshs.core.CacheManager;
import org.tshs.core.Constant.ObjectType;
import org.tshs.storage.rdbms.ColumnInfo;
import org.tshs.storage.rdbms.DbManager;
import org.tshs.storage.rdbms.TableRow;

public class TravelDept extends TshsObject {
	
	public static final int propertyLength = 5;

	private String name;
	
	private String password;
	
	private String email;
	
	private String address;
	
	private String description;

	private List<Long> sightIds;
    
    TravelDept(Long id){
		super(id, ObjectType.DEPT);
    }
	
	/**
	 * @param row
	 */
	public TravelDept(TableRow row) {
		super(Long.valueOf(row.getColumn("id").toString()), ObjectType.DEPT);
		setName((String)row.getColumn("name"));
		setPassword((String)row.getColumn("password"));
		setEmail((String)row.getColumn("email"));
		setAddress((String)row.getColumn("address"));
		setDescription((String)row.getColumn("description"));
		List<Long> idlist = (List<Long>) row.getColumn("sightIds");
		setSightIds(idlist);
	}

	public void setName(String name){
		this.name = name;
	}

	public String getName() {
		return name;
	}
	
	public void setPassword(String password){
		this.password = password;
	}
	
	public String getPassword(){
		return password;
	}
	
	public void setEmail(String email){
		this.email = email;
	}

	public String getEmail() {
		return email;
	}
	
	public void setAddress(String address){
		this.address = address;
	}
	
	public String getAddress(){
		return address;
	}
	
	public void setDescription(String description){
		this.description = description;
	}
	
	public String getDescription(){
		return description;
	}
    
    public void setSightIds(List<Long> sightIds){
    	this.sightIds = sightIds;
    }
	
	public List<Long> getSightIds(){
		return sightIds;
	}
	
	public boolean checkPassword(String password){
		if(this.password.equals(password)){
			return true;
		}else{
			return false;
		}
	}
    
    public List<SightSpot> getAllSights() throws Exception{
    	List<SightSpot> sightList = new ArrayList<SightSpot>();
    	Iterator<Long> i = sightIds.iterator();
    	while(i.hasNext()){
    		sightList.add((SightSpot)CacheManager.retrieve(i.next()));
    	}
    	return sightList;
    }
	
	public Vector selfList() throws Exception{
		Vector<String> sightNames = new Vector<String>();
		List<SightSpot> sights = getAllSights();
		Iterator<SightSpot> i = sights.iterator();
		while (i.hasNext()) {
			sightNames.add(i.next().getName());
		}
		
		Vector<Object> self = new Vector<Object>();
		self.add(name);
		self.add(password);
		self.add(email);
		self.add(address);
		self.add(description);
		self.add(sightNames);
		return self;
	}

	public boolean modify(Vector data) {
		if(data.size() != propertyLength){
			System.out.println("Error from TravelDept: data.size() != propertyLength");
			return false;
		}
		
		setName((String) data.get(0));
		setPassword((String)data.get(1));
		setEmail((String)data.get(2));
		setAddress((String)data.get(3));
		setDescription((String)data.get(4));
		
		try {
			DbManager.update(buildTableRow());
		} catch (Exception e) {
			e.printStackTrace();
		}
		
		return true;
	}
	
	protected TableRow buildTableRow(){
		List<String> names = new ArrayList<String>();
		List<Integer> types = new ArrayList<Integer>();
		
		names.add("id");
		types.add(Types.BIGINT);
		names.add("name");
		types.add(Types.VARCHAR);
		names.add("password");
		types.add(Types.VARCHAR);
		names.add("email");
		types.add(Types.VARCHAR);
		names.add("address");
		types.add(Types.VARCHAR);
		names.add("description");
		types.add(Types.VARCHAR);
		
		ColumnInfo info = null;
		try {
			info = new ColumnInfo(names, types);
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			return null;
		}
		
		Map<String, Object> map = new HashMap<String, Object>();
		map.put("id", id);
		map.put("name", name);
		map.put("password", password);
		map.put("email", email);
		map.put("address", address);
		map.put("description", description);
		TableRow row = new TableRow("traveldept", map, info);
		
		return row;
	}

	/**
	 * @return
	 */
	public boolean update() {
		try {
			DbManager.update(this.buildTableRow());
		} catch (SQLException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			return false;
		}
		return true;
	}
	
}