usercontext.java

来自「WebChatSample」· Java 代码 · 共 65 行

package com.signorinolong.test.chat;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import com.signorinolong.test.chat.model.SampleUser;
import com.signorinolong.test.chat.model.User;

public class UserContext {
	
	private Map<String,User> users;

	public UserContext() {
		// TODO Auto-generated constructor stub
		users = new HashMap<String,User>();
	}

	public Map<String, User> getUsers() {
		return users;
	}

	public void setUsers(Map<String, User> users) {
		this.users = users;
	}
	
	public void addUser(String username) {
		if (!users.containsKey(username)) {
			users.put(username, new SampleUser(username));
		}
	}
	
	public void removeUser(String username) {
		if (users.containsKey(username)) {
			users.remove(username);
		}
	}
	
	public User getUser(String username) {
		if (users.containsKey(username)) {
			return users.get(username);
		}
		return null;
	}
	
	public boolean containUser(String username) {
		if (users.containsKey(username)) {
			return true;
		}
		return false;
	}
	
	public List<String> getUserList() {
		Set<String> kset = users.keySet();
		ArrayList<String> userlist = new ArrayList<String>();
		for (int i=0,n=kset.size();i<n;i++) {
			userlist.add((String)(kset.toArray()[i]));
		}
		return userlist;
	}
	
}