package com.github.looly.hutool.bloomFilter;

import com.github.looly.hutool.bloomFilter.filter.DefaultFilter;
import com.github.looly.hutool.bloomFilter.filter.ELFFilter;
import com.github.looly.hutool.bloomFilter.filter.Filter;
import com.github.looly.hutool.bloomFilter.filter.JSFilter;
import com.github.looly.hutool.bloomFilter.filter.PJWFilter;
import com.github.looly.hutool.bloomFilter.filter.SDBMFilter;

/**
 * BlommFilter 实现 <br>
 * 1.构建hash算法 <br>
 * 2.散列hash映射到数组的bit位置 <br>
 * 3.验证<br>
 * 此实现方式可以指定Hash算法
 * 
 * @author Ansj
 */
public class BloomFilter {

	private Filter[] filters;

	/**
	 * 使用默认的5个过滤器
	 * @param m
	 */
	public BloomFilter(int m) {
		float mNum = m / 5;
		long size = (long) (1L * mNum * 1024 * 1024 * 8);
		
		filters = new Filter[]{
			new DefaultFilter(size),
			new ELFFilter(size),
			new JSFilter(size),
			new PJWFilter(size),
			new SDBMFilter(size)
		};
	}

	/**
	 * 使用自定的多个过滤器建立BloomFilter
	 * 
	 * @param m M值决定BitMap的大小
	 * @param filters Bloom过滤器列表
	 */
	public BloomFilter(int m, Filter... filters) {
		this.filters = filters;
	}

	/**
	 * 增加字符串到Filter映射中
	 * @param str 字符串
	 */
	public void add(String str) {
		for (Filter filter : filters) {
			filter.add(str);
		}
	}

	/**
	 * 是否可能包含此字符串，此处存在误判
	 * @param str 字符串
	 * @return 是否存在
	 */
	public boolean contains(String str) {
		for (Filter filter : filters) {
			if (filter.contains(str) == false) {
				return false;
			}
		}
		return true;
	}

	/**
	 * 判定字符串是否存在于各Filter中，如果不存在加入此Filter中
	 * @param str 字符串
	 * @return 如果全部包含
	 */
	public boolean containsAndAdd(String str) {
		boolean flag = true;
		for (Filter filter : filters) {
			flag &= filter.containsAndAdd(str);
		}
		return flag;
	}
}