001/* 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, 013 * software distributed under the License is distributed on an 014 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 015 * KIND, either express or implied. See the License for the 016 * specific language governing permissions and limitations 017 * under the License. 018 * 019 */ 020package org.apache.mina.proxy.session; 021 022import org.apache.mina.core.future.ConnectFuture; 023import org.apache.mina.core.session.IoSession; 024import org.apache.mina.core.session.IoSessionInitializer; 025 026/** 027 * ProxyIoSessionInitializer.java - {@link IoSessionInitializer} wrapper class to inject the 028 * {@link ProxyIoSession} object that contains all the attributes of the target connection 029 * into the {@link IoSession}. 030 * 031 * @author <a href="http://mina.apache.org">Apache MINA Project</a> 032 * @since MINA 2.0.0-M3 033 */ 034public class ProxyIoSessionInitializer<T extends ConnectFuture> implements IoSessionInitializer<T> { 035 private final IoSessionInitializer<T> wrappedSessionInitializer; 036 037 private final ProxyIoSession proxyIoSession; 038 039 public ProxyIoSessionInitializer(final IoSessionInitializer<T> wrappedSessionInitializer, 040 final ProxyIoSession proxyIoSession) { 041 this.wrappedSessionInitializer = wrappedSessionInitializer; 042 this.proxyIoSession = proxyIoSession; 043 } 044 045 public ProxyIoSession getProxySession() { 046 return proxyIoSession; 047 } 048 049 public void initializeSession(final IoSession session, T future) { 050 if (wrappedSessionInitializer != null) { 051 wrappedSessionInitializer.initializeSession(session, future); 052 } 053 054 if (proxyIoSession != null) { 055 proxyIoSession.setSession(session); 056 session.setAttribute(ProxyIoSession.PROXY_SESSION, proxyIoSession); 057 } 058 } 059}