コード例 #1
0
 protected Commutable receiveData(Connection conn, Connection nextConn) throws Exception {
   InputStream in = conn.getInputDtream();
   OutputStream outNextWorker = null;
   if (nextConn != null) {
     outNextWorker = nextConn.getOutputStream();
   }
   // Receive data
   ByteArray byteArray = receiveByteArray(in, outNextWorker);
   // Close connection
   conn.close();
   if (nextConn != null) {
     nextConn.close();
   }
   // Process byte array
   Commutable data = null;
   try {
     data = processByteArray(byteArray);
   } catch (Exception e) {
     releaseBytes(byteArray.getArray());
     releaseInts(byteArray.getMetaArray());
     throw e;
   }
   return data;
 }
コード例 #2
0
 /**
  * Receive 1. command 2. byte array size 3. meta data size 4. meta data content
  *
  * @param in
  * @param outNextWorker
  * @return
  * @throws Exception
  */
 private ByteArray receiveByteArray(InputStream in, OutputStream outNextWorker) throws Exception {
   // int size = din.readInt();
   // int metaDataSize = din.readInt();
   // Read byte array size and meta array size
   byte[] sizeBytes = this.getResourcePool().getByteArrayPool().getArray(8);
   in.read(sizeBytes);
   // Forward
   if (outNextWorker != null) {
     outNextWorker.write(this.getCommand());
     outNextWorker.write(sizeBytes);
   }
   DataDeserializer deserializer = new DataDeserializer(sizeBytes);
   int size = deserializer.readInt();
   int metaArraySize = deserializer.readInt();
   this.getResourcePool().getByteArrayPool().releaseArrayInUse(sizeBytes);
   // Read meta array
   int[] metaArray = null;
   if (metaArraySize > 0) {
     metaArray = this.getResourcePool().getIntArrayPool().getArray(metaArraySize);
     byte[] metaArrayBytes = this.getResourcePool().getByteArrayPool().getArray(4 * metaArraySize);
     in.read(metaArrayBytes);
     // Forward
     if (outNextWorker != null) {
       outNextWorker.write(metaArrayBytes);
     }
     deserializer.setData(metaArrayBytes);
     for (int i = 0; i < metaArraySize; i++) {
       metaArray[i] = deserializer.readInt();
     }
     this.getResourcePool().getByteArrayPool().releaseArrayInUse(metaArrayBytes);
   }
   if (outNextWorker != null) {
     outNextWorker.flush();
   }
   // Prepare internal bytes structure
   byte[] bytes =
       this.getResourcePool().getByteArrayPool().getArray(size + Constants.SENDRECV_BYTE_UNIT);
   try {
     receiveBytes(in, outNextWorker, bytes, size);
   } catch (Exception e) {
     releaseBytes(bytes);
     releaseInts(metaArray);
     throw e;
   }
   ByteArray byteArray = new ByteArray();
   byteArray.setArray(bytes);
   byteArray.setStart(0);
   byteArray.setSize(size);
   byteArray.setMetaArray(metaArray);
   byteArray.setMetaArraySize(metaArraySize);
   return byteArray;
 }