Mercurial > jdk8-mips64-public > corba / file revision

 /*

  * Copyright (c) 2000, 2010, Oracle and/or its affiliates. All rights reserved.

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

  *

  * This code is free software; you can redistribute it and/or modify it

  * under the terms of the GNU General Public License version 2 only, as

  * published by the Free Software Foundation.  Oracle designates this

  * particular file as subject to the "Classpath" exception as provided

  * by Oracle in the LICENSE file that accompanied this code.

  *

  * This code is distributed in the hope that it will be useful, but WITHOUT

  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

  * version 2 for more details (a copy is included in the LICENSE file that

  * accompanied this code).

  *

  * You should have received a copy of the GNU General Public License version

  * 2 along with this work; if not, write to the Free Software Foundation,

  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

  *

  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

  * or visit www.oracle.com if you need additional information or have any

  * questions.

  */

 package com.sun.corba.se.impl.encoding;

 import java.nio.ByteBuffer;

 import com.sun.corba.se.impl.orbutil.ORBConstants;

 import com.sun.corba.se.impl.protocol.giopmsgheaders.Message;

 import com.sun.corba.se.impl.protocol.giopmsgheaders.MessageBase;

 import com.sun.corba.se.impl.protocol.giopmsgheaders.FragmentMessage;

 import com.sun.corba.se.impl.protocol.giopmsgheaders.ReplyMessage;

 import com.sun.corba.se.impl.encoding.BufferManagerWrite;

 import com.sun.corba.se.impl.encoding.ByteBufferWithInfo;

 import com.sun.corba.se.impl.encoding.CDROutputObject;

 import com.sun.corba.se.spi.orb.ORB;

 import com.sun.corba.se.pept.transport.Connection;

 import com.sun.corba.se.pept.encoding.OutputObject;

 import org.omg.CORBA.SystemException;

 /**

  * Streaming buffer manager.

  */

 public class BufferManagerWriteStream extends BufferManagerWrite

 {

     private int fragmentCount = 0;

     BufferManagerWriteStream( ORB orb )

     {

         super(orb) ;

     }

     public boolean sentFragment() {

         return fragmentCount > 0;

     }

     /**

      * Returns the correct buffer size for this type of

      * buffer manager as set in the ORB.

      */

     public int getBufferSize() {

         return orb.getORBData().getGIOPFragmentSize();

     }

     public void overflow (ByteBufferWithInfo bbwi)

     {

         // Set the fragment's moreFragments field to true

         MessageBase.setFlag(bbwi.byteBuffer, Message.MORE_FRAGMENTS_BIT);

         try {

            sendFragment(false);

         } catch(SystemException se){

                 orb.getPIHandler().invokeClientPIEndingPoint(

                         ReplyMessage.SYSTEM_EXCEPTION, se);

                 throw se;

         }

         // Reuse the old buffer

         // REVISIT - need to account for case when needed > available

         // even after fragmenting.  This is the large array case, so

         // the caller should retry when it runs out of space.

         bbwi.position(0);

         bbwi.buflen = bbwi.byteBuffer.limit();

         bbwi.fragmented = true;

         // Now we must marshal in the fragment header/GIOP header

         // REVISIT - we can optimize this by not creating the fragment message

         // each time.

         FragmentMessage header = ((CDROutputObject)outputObject).getMessageHeader().createFragmentMessage();

         header.write(((CDROutputObject)outputObject));

     }

     private void sendFragment(boolean isLastFragment)

     {

         Connection conn = ((OutputObject)outputObject).getMessageMediator().getConnection();

         // REVISIT: need an ORB

         //System.out.println("sendFragment: last?: " + isLastFragment);

         conn.writeLock();

         try {

             // Send the fragment

             conn.sendWithoutLock(((OutputObject)outputObject));

             fragmentCount++;

         } finally {

             conn.writeUnlock();

         }

     }

     // Sends the last fragment

     public void sendMessage ()

     {

         sendFragment(true);

         sentFullMessage = true;

     }

     /**

      * Close the BufferManagerWrite and do any outstanding cleanup.

      *

      * No work to do for a BufferManagerWriteStream

      */

     public void close(){};

 }