ipholder.java

非常有影响的 j道 论 坛 源码 国外很有明的专家编写的 ....对java爱好者很有参考价值

/*
 * Copyright 2007 the original author or jdon.com
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package com.jdon.jivejdon.manager.block;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;
import java.util.regex.Pattern;

import com.jdon.jivejdon.util.StringSortComparator;

public class IPHolder {
	// set of ips that are banned, use a set to ensure uniqueness
	private Set bannedIps; 

	private List<Pattern> regexBannedIps;

	public IPHolder() {
		bannedIps =  new TreeSet(new StringSortComparator());
		regexBannedIps = new ArrayList();

	}

	public Set getBannedIps() {
		return bannedIps;
	}

	public Collection getAllBanIpList() {
		return bannedIps;
	}

	public List<Pattern> getRegexBannedIps() {
		return regexBannedIps;
	}

	public void addressAdd(String ip) {
		if (bannedIps.contains(ip))
			return;
		if (ip.indexOf("*") != -1) {// if it is 202.1.*.*
			Pattern p = Pattern.compile(ip);
			regexBannedIps.add(p);
		}
		bannedIps.add(ip);// if it is 202.1.1.1
	}

	public void addressRemove(String ip) {
		this.regexBannedIps.remove(ip);
		this.bannedIps.remove(ip);// if it is 202.1.1.1
	}

	public String addressLoad() {
		StringBuffer sb = new StringBuffer();
		for (Object o : bannedIps) {
			sb.append((String) o).append("\n");
		}
		return sb.toString();
	}
	
	public void clear(){
		bannedIps.clear();
		regexBannedIps.clear();
	}

}