import java.io.IOException;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
 
 import javax.management.MBeanServerConnection;
 import javax.management.MBeanServerInvocationHandler;
 import javax.management.ObjectName;
 import javax.management.remote.JMXConnector;
 import javax.management.remote.JMXConnectorFactory;
 import javax.management.remote.JMXServiceURL;
 
 //copied from the tim inside sessions.jar
 import com.terracotta.session.management.SessionMonitorMBean;
 
 public class TCJMXSessionHelper {
 
 	String connectorUrl = "service:jmx:jmxmp://172.16.100.2:9520";
 
 	String appPattern = "org.terracotta:appName=localhost/myapp";
 
 	private JMXConnector getJMXConnector() throws IOException {
 		return getJMXConnector(connectorUrl);
 	}
 
 	JMXConnector getJMXConnector(final String url) throws IOException {
 		return JMXConnectorFactory.connect(new JMXServiceURL(url));
 	}
 
 	List<SessionMonitorMBean> getSessionMbeans(final JMXConnector connector) throws IOException {
 
 		final List<SessionMonitorMBean> ret = new ArrayList<>();
 		final MBeanServerConnection connection = connector.getMBeanServerConnection();
 
 		final Set<ObjectName> names = connection.queryNames(null, null);
 
 		for (final ObjectName on : names) {
 			if (on.toString().startsWith(appPattern)) {
 				ret.add(MBeanServerInvocationHandler.newProxyInstance(connection, on, SessionMonitorMBean.class, false));
 			}
 		}
 
 		return ret;
 	}
 
 	public void purgeSessions(final Iterable<String> ids) throws IOException {
 
 		final JMXConnector connector = getJMXConnector();
 
 		try {
 			final List<SessionMonitorMBean> mbeans = getSessionMbeans(connector);
 
 			for (final SessionMonitorMBean bean : mbeans) {
 				try {
 					for (final String id : ids) {
 						bean.expireSession(id);
 					}
 					break;
 				} catch (final Exception ex) {
 					// ignore go to next bean
 				}
 
 			}
 
 		} finally {
 			connector.close();
 		}
 
 	}
 
 	public void purgeSessions(final String... ids) throws IOException {
 		purgeSessions(Arrays.asList(ids));
 	}
 
 	public void purgeStaleSesssions() throws IOException {
 
 		final JMXConnector connector = getJMXConnector();
 
 		try {
 			final List<SessionMonitorMBean> mbeans = getSessionMbeans(connector);
 
 			beanLoop: for (final SessionMonitorMBean bean : mbeans) {
 				final int step = 100;
 				final long total = bean.getActiveSessionCount();
 
 				try {
 					for (long j = step; j <= total + step; j += step) {
 						final Set<String> ids = bean.getSessionIds(step);
 
 						if (ids == null || ids.isEmpty()) {
 							break beanLoop;
 						}
 
 						for (final String id : ids) {
 							final Map<String, String> map = bean.getSessionAttributes(id);
 
 							if (map == null || map.isEmpty()) {
 								if (!bean.expireSession(id)) {
 									System.out.println("Why can't TC expire " + id + " ????");
 								}
 							}
 						}
 					}
 					break;
 				} catch (final Exception ex) {
 					// ignore go to next bean
 				}
 			}
 		} finally {
 			connector.close();
 		}
 	}
 
 }
 
 

  public boolean killSession(String browserSessionId) {
     SessionId id = this.idGenerator.makeInstanceFromBrowserId(browserSessionId);
     if (id == null)
     {
       id = this.idGenerator.makeInstanceFromInternalKey(browserSessionId);
     }
     return expire(id);
   }
 
 private boolean expire(SessionId id) {
     boolean result = false;
     SessionData sd = null;
     boolean locked = false;
     try {
       sd = this.store.find(id);
       if (sd != null) {
         if (!isSessionLockingEnabled()) {
           id.getWriteLock();
         }
         locked = true;
         expire(id, sd);
         result = true;
       }
     } finally {
       if ((sd != null) && (locked)) {
         id.commitWriteLock();
       }
     }
     return result;
   }
 

  public Set<String> getSessionIds(int batchSize) {
     int count = 0;
     HashSet result = new HashSet();
     for (String id : this.store.getAllKeys()) {
       result.add(id);
       if (count++ >= batchSize) {
         break;
       }
     }
     return result;
   }
 

 private boolean expire(SessionId id) {
     boolean result = false;
     SessionData sd = null;
     boolean locked = false;
     try {
       sd = this.store.find(id);
       if (sd != null) {
         if (!isSessionLockingEnabled()) {
           id.getWriteLock();
         }
         locked = true;
         expire(id, sd);
         result = true;
       }else{
            //FORCE REMOVE
           this.store.remove(id);
        }
     } finally {
       if ((sd != null) && (locked)) {
         id.commitWriteLock();
       }
     }
     return result;
   }