1 /* 2 * Licensed to the Apache Software Foundation (ASF) under one 3 * or more contributor license agreements. See the NOTICE file 4 * distributed with this work for additional information 5 * regarding copyright ownership. The ASF licenses this file 6 * to you under the Apache License, Version 2.0 (the 7 * "License"); you may not use this file except in compliance 8 * with the License. You may obtain a copy of the License at 9 * 10 * http://www.apache.org/licenses/LICENSE-2.0 11 * 12 * Unless required by applicable law or agreed to in writing, 13 * software distributed under the License is distributed on an 14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 15 * KIND, either express or implied. See the License for the 16 * specific language governing permissions and limitations 17 * under the License. 18 * 19 */ 20 package org.apache.mina.proxy.session; 21 22 import org.apache.mina.core.future.ConnectFuture; 23 import org.apache.mina.core.session.IoSession; 24 import org.apache.mina.core.session.IoSessionInitializer; 25 26 /** 27 * ProxyIoSessionInitializer.java - {@link IoSessionInitializer} wrapper class to inject the 28 * {@link ProxyIoSession} object that contains all the attributes of the target connection 29 * into the {@link IoSession}. 30 * 31 * @param <T> The Connection Future type 32 * 33 * @author <a href="http://mina.apache.org">Apache MINA Project</a> 34 * @since MINA 2.0.0-M3 35 */ 36 public class ProxyIoSessionInitializer<T extends ConnectFuture> implements IoSessionInitializer<T> { 37 private final IoSessionInitializer<T> wrappedSessionInitializer; 38 39 private final ProxyIoSession proxyIoSession; 40 41 /** 42 * Creates a new ProxyIoSessionInitializer instance 43 * 44 * @param wrappedSessionInitializer The wrapped session initializer 45 * @param proxyIoSession The ProxyIoSession instance 46 */ 47 public ProxyIoSessionInitializer(final IoSessionInitializer<T> wrappedSessionInitializer, 48 final ProxyIoSession proxyIoSession) { 49 this.wrappedSessionInitializer = wrappedSessionInitializer; 50 this.proxyIoSession = proxyIoSession; 51 } 52 53 /** 54 * @return The ProxyIoSession instance 55 */ 56 public ProxyIoSession getProxySession() { 57 return proxyIoSession; 58 } 59 60 /** 61 * {@inheritDoc} 62 */ 63 @Override 64 public void initializeSession(final IoSession session, T future) { 65 if (wrappedSessionInitializer != null) { 66 wrappedSessionInitializer.initializeSession(session, future); 67 } 68 69 if (proxyIoSession != null) { 70 proxyIoSession.setSession(session); 71 session.setAttribute(ProxyIoSession.PROXY_SESSION, proxyIoSession); 72 } 73 } 74 }