/*

  * 

  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

  * 

  * Copyright 2007-2008 Sun Microsystems, Inc. All rights reserved.

  * 

  * The contents of this file are subject to the terms of either the GNU

  * General Public License Version 2 only ("GPL") or the Common Development

  * and Distribution License("CDDL") (collectively, the "License").  You

  * may not use this file except in compliance with the License. You can obtain

  * a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html

  * or glassfish/bootstrap/legal/LICENSE.txt.  See the License for the specific

  * language governing permissions and limitations under the License.

  * 

  * When distributing the software, include this License Header Notice in each

  * file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.

  * Sun designates this particular file as subject to the "Classpath" exception

  * as provided by Sun in the GPL Version 2 section of the License file that

  * accompanied this code.  If applicable, add the following below the License

  * Header, with the fields enclosed by brackets [] replaced by your own

  * identifying information: "Portions Copyrighted [year]

  * [name of copyright owner]"

  * 

  * Contributor(s):

  * 

  * If you wish your version of this file to be governed by only the CDDL or

  * only the GPL Version 2, indicate your decision by adding "[Contributor]

  * elects to include this software in this distribution under the [CDDL or GPL

  * Version 2] license."  If you don't indicate a single choice of license, a

  * recipient has the option to distribute your version of this file under

  * either the CDDL, the GPL Version 2 or to extend the choice of license to

  * its licensees as provided above.  However, if you add GPL Version 2 code

  * and therefore, elected the GPL Version 2 license, then the option applies

  * only if the new code is made subject to such option by the copyright

  * holder.

  *

  */

 package com.sun.grizzly.async;

 import com.sun.grizzly.SelectorHandler;

 import java.io.EOFException;

 import java.io.IOException;

 import java.net.SocketAddress;

 import java.nio.ByteBuffer;

 import java.nio.channels.ReadableByteChannel;

 import java.nio.channels.SocketChannel;

 /**

  *

  * @author Alexey Stashok

  */

 public class TCPAsyncQueueReader extends AbstractAsyncQueueReader {

     public TCPAsyncQueueReader(SelectorHandler selectorHandler) {

         super(selectorHandler);

     }

     protected SocketAddress doRead(ReadableByteChannel channel, ByteBuffer byteBuffer, 

             AsyncQueueDataProcessor readPostProcessor) throws IOException {

         if (readPostProcessor != null) {

             ByteBuffer inputByteBuffer = null;

             int oldPosition = byteBuffer.position();

             do {

                 inputByteBuffer = readPostProcessor.getInternalByteBuffer();

                 int readBytes = doRead(channel, inputByteBuffer);

                 if (readBytes > 0) {

                     readPostProcessor.process(byteBuffer);

                 } else if (readBytes == -1) { 

                     if (byteBuffer.position() == oldPosition) {

                         throw new EOFException();

                     } else {

                         break;

                     }

                 }

             } while(byteBuffer.hasRemaining() && 

                     !inputByteBuffer.hasRemaining());

         } else {

             doRead(channel, byteBuffer);

         }

         return ((SocketChannel) channel).socket().getRemoteSocketAddress();

     }

     private int doRead(ReadableByteChannel channel, ByteBuffer byteBuffer) 

             throws IOException {

         int readBytes = 0;

         int lastReadBytes = 0;

         do {

             lastReadBytes = channel.read(byteBuffer);

             if (lastReadBytes > 0) {

                 readBytes += lastReadBytes;

             } else if (lastReadBytes == -1 && readBytes == 0) {

                 readBytes = -1;

             }

         } while(lastReadBytes > 0 && byteBuffer.hasRemaining());

         return readBytes;

     }

 }