package com.test;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Enumeration;
import java.util.List;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;

public class CustomRequest extends HttpServletRequestWrapper {

	public CustomRequest(HttpServletRequest request) {
		super(request);
	}

	public String getHeader(String name) {

		HttpServletRequest request = (HttpServletRequest) getRequest();

		if ("datasource".equals(name)) {
			return "datasource-value";
		}

		return request.getHeader(name);
	}

	public Enumeration getHeaderNames() {

		List list = new ArrayList();
		HttpServletRequest request = (HttpServletRequest) getRequest();
		Enumeration e = request.getHeaderNames();
		while (e.hasMoreElements()) {
			String n = (String) e.nextElement();
			list.add(n);
		}
		list.add("datasource");
		Enumeration en = Collections.enumeration(list);
		return en;
	}

}