/*
    *  Licensed to the Apache Software Foundation (ASF) under one
    *  or more contributor license agreements.  See the NOTICE file
    *  distributed with this work for additional information
    *  regarding copyright ownership.  The ASF licenses this file
    *  to you under the Apache License, Version 2.0 (the
    *  "License"); you may not use this file except in compliance
    *  with the License.  You may obtain a copy of the License at
    *
   *    http://www.apache.org/licenses/LICENSE-2.0
   *
   *  Unless required by applicable law or agreed to in writing,
   *  software distributed under the License is distributed on an
   *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   *  KIND, either express or implied.  See the License for the
   *  specific language governing permissions and limitations
   *  under the License.
   *
   */
  package testcase;
  
  import org.apache.mina.core.buffer.IoBuffer;
  import org.apache.mina.core.service.IoHandlerAdapter;
  import org.apache.mina.core.session.IoSession;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import static testcase.MinaRegressionTest.MSG_COUNT;
  import static testcase.MinaRegressionTest.OPEN;
  
  import java.io.IOException;
  import java.util.concurrent.atomic.AtomicInteger;
  
  /**
   * TODO: Document me !
   * @author <a href="http://mina.apache.org">Apache MINA Project</a>
   *
   */
  public class MyIoHandler extends IoHandlerAdapter {
      private static final Logger logger = LoggerFactory.getLogger(MyIoHandler.class);
  
      public static AtomicInteger received = new AtomicInteger(0);
  
      public static AtomicInteger closed = new AtomicInteger(0);
  
      private final Object LOCK;
  
      public MyIoHandler(Object lock) {
          LOCK = lock;
      }
  
      @Override
      public void exceptionCaught(IoSession session, Throwable cause) {
          if (!(cause instanceof IOException)) {
              logger.error("Exception: ", cause);
          } else {
              logger.info("I/O error: " + cause.getMessage());
          }
          session.closeNow();
      }
  
      @Override
      public void sessionOpened(IoSession session) throws Exception {
          logger.info("Session " + session.getId() + " is opened");
          session.resumeRead();
      }
  
      @Override
      public void sessionCreated(IoSession session) throws Exception {
          logger.info("Creation of session " + session.getId());
          session.setAttribute(OPEN);
          session.suspendRead();
      }
  
      @Override
      public void sessionClosed(IoSession session) throws Exception {
          session.removeAttribute(OPEN);
          logger.info("{}> Session closed", session.getId());
          final int clsd = closed.incrementAndGet();
  
          if (clsd == MSG_COUNT) {
              synchronized (LOCK) {
                  LOCK.notifyAll();
              }
          }
  
          int i = 0;
  
          try {
              int j = 2 / i;
          } catch (Exception e) {
              //e.printStackTrace();
          }
      }
  
      @Override
      public void messageReceived(IoSession session, Object message) throws Exception {
          IoBuffer msg = (IoBuffer) message;
          logger.info("MESSAGE: " + msg.remaining() + " on session " + session.getId());
          final int rec = received.incrementAndGet();
 
         if (rec == MSG_COUNT) {
             synchronized (LOCK) {
                 LOCK.notifyAll();
             }
         }
 
         session.closeNow();
     }
 }