geode-core/src/main/java/org/apache/geode/internal/cache/tier/sockets/ObjectPartList.java - geode - Git at Google

 /*
  * 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 org.apache.geode.internal.cache.tier.sockets;

 import java.io.DataInput;
 import java.io.DataOutput;
 import java.io.IOException;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;

 import org.apache.logging.log4j.Logger;

 import org.apache.geode.DataSerializer;
 import org.apache.geode.internal.cache.versions.VersionTag;
 import org.apache.geode.internal.offheap.OffHeapHelper;
 import org.apache.geode.internal.offheap.Releasable;
 import org.apache.geode.internal.serialization.DataSerializableFixedID;
 import org.apache.geode.internal.serialization.DeserializationContext;
 import org.apache.geode.internal.serialization.SerializationContext;
 import org.apache.geode.internal.serialization.Version;
 import org.apache.geode.logging.internal.log4j.api.LogService;

 /**
  * Encapsulates list containing objects, serialized objects, raw byte arrays, or exceptions. It can
  * optionally also hold the list of associated keys. Assumes that keys are either provided for all
  * entries or for none.
  *
  * @since GemFire 5.7
  */
 public class ObjectPartList implements DataSerializableFixedID, Releasable {
   private static final Logger logger = LogService.getLogger();

   protected static final byte BYTES = 0;

   protected static final byte OBJECT = 1;

   protected static final byte EXCEPTION = 2;

   protected static final byte KEY_NOT_AT_SERVER = 3;

   protected byte[] objectTypeArray;

   protected boolean hasKeys;

   protected List keys;

   protected List objects;

   public void addPart(Object key, Object value, byte objectType, VersionTag versionTag) {
     int size = this.objects.size();
     int maxSize = this.objectTypeArray.length;
     if (size >= maxSize) {
       throw new IndexOutOfBoundsException("Cannot add object part beyond " + maxSize + " elements");
     }
     if (this.hasKeys) {
       if (key == null) {
         throw new IllegalArgumentException("Cannot add null key");
       }
       this.keys.add(key);
     }
     this.objectTypeArray[size] = objectType;
     this.objects.add(value);
   }

   // public methods

   public ObjectPartList() {
     this.objectTypeArray = null;
     this.hasKeys = false;
     this.keys = null;
     this.objects = new ArrayList();
   }

   public ObjectPartList(int maxSize, boolean hasKeys) {
     if (maxSize <= 0) {
       throw new IllegalArgumentException(
           "Invalid size " + maxSize + " to ObjectPartList constructor");
     }
     this.objectTypeArray = new byte[maxSize];
     this.hasKeys = hasKeys;
     if (hasKeys) {
       this.keys = new ArrayList();
     } else {
       this.keys = null;
     }
     this.objects = new ArrayList();
   }

   public void addObjectPart(Object key, Object value, boolean isObject, VersionTag versionTag) {
     addPart(key, value, isObject ? OBJECT : BYTES, versionTag);
   }

   public void addExceptionPart(Object key, Exception ex) {
     addPart(key, ex, EXCEPTION, null);
   }

   public void addObjectPartForAbsentKey(Object key, Object value) {
     // ObjectPartList is for clients < version 6.5.0, which didn't support this setting
     throw new IllegalAccessError("inappropriate use of ObjectPartList");
   }


   public void addAll(ObjectPartList other) {
     if (this.hasKeys) {
       if (other.keys != null) {
         if (this.keys == null) {
           this.keys = new ArrayList(other.keys);
         } else {
           this.keys.addAll(other.keys);
         }
       }
     } else if (other.hasKeys) {
       this.hasKeys = true;
       this.keys = new ArrayList(other.keys);
     }
     this.objects.addAll(other.objects);
   }

   public List<Object> getKeys() {
     if (this.keys == null) {
       return Collections.emptyList();
     } else {
       return Collections.unmodifiableList(this.keys);
     }
   }

   /** unprotected access to the keys collection, which may be null */
   public List getKeysForTest() {
     return this.keys;
   }

   public List getObjects() {
     if (this.objects == null) {
       return Collections.emptyList();
     } else {
       return Collections.unmodifiableList(this.objects);
     }
   }

   /** unprotected access to the objects collection, which may be null */
   public List getObjectsForTest() {
     return this.objects;
   }

   public int size() {
     // some lists have only keys and some have only objects, so we need to choose
     // the correct collection to query
     if (this.hasKeys) {
       return this.keys.size();
     } else {
       return this.objects.size();
     }
   }

   public void reinit(int maxSize) {
     if (maxSize <= 0) {
       throw new IllegalArgumentException("Invalid size " + maxSize + " to ObjectPartList.reinit");
     }
     this.objectTypeArray = new byte[maxSize];
     this.objects.clear();
     this.keys.clear();
   }

   public void clear() {
     release();
     this.objects.clear();
     if (this.keys != null) {
       this.keys.clear();
     }
   }

   @Override
   public void toData(DataOutput out,
       SerializationContext context) throws IOException {
     out.writeBoolean(this.hasKeys);
     if (this.objectTypeArray != null) {
       int numObjects = this.objects.size();
       out.writeInt(numObjects);
       for (int index = 0; index < numObjects; ++index) {
         Object value = this.objects.get(index);
         byte objectType = this.objectTypeArray[index];
         if (this.hasKeys) {
           context.getSerializer().writeObject(this.keys.get(index), out);
         }
         out.writeBoolean(objectType == EXCEPTION);
         if (objectType == OBJECT && value instanceof byte[]) {
           out.write((byte[]) value);
         } else if (objectType == EXCEPTION) {
           // write exception as byte array so native clients can skip it
           DataSerializer.writeByteArray(CacheServerHelper.serialize(value), out);
           // write the exception string for native clients
           DataSerializer.writeString(value.toString(), out);
         } else {
           context.getSerializer().writeObject(value, out);
         }
       }
     } else {
       out.writeInt(0);
     }
   }

   @Override
   public void fromData(DataInput in,
       DeserializationContext context) throws IOException, ClassNotFoundException {
     this.hasKeys = in.readBoolean();
     if (this.hasKeys) {
       this.keys = new ArrayList();
     }
     int numObjects = in.readInt();
     if (numObjects > 0) {
       for (int index = 0; index < numObjects; ++index) {
         if (this.hasKeys) {
           Object key = context.getDeserializer().readObject(in);
           this.keys.add(key);
         }
         boolean isException = in.readBoolean();
         Object value;
         if (isException) {
           byte[] exBytes = DataSerializer.readByteArray(in);
           value = CacheServerHelper.deserialize(exBytes);
           // ignore the exception string meant for native clients
           DataSerializer.readString(in);
         } else {
           value = context.getDeserializer().readObject(in);
         }
         this.objects.add(value);
       }
     }
   }

   @Override
   public int getDSFID() {
     return DataSerializableFixedID.OBJECT_PART_LIST;
   }

   @Override
   public Version[] getSerializationVersions() {
     return null;
   }

   @Override
   public void release() {
     for (Object v : this.objects) {
       OffHeapHelper.release(v);
     }
   }
 }
	/*
	* 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 org.apache.geode.internal.cache.tier.sockets;

	import java.io.DataInput;
	import java.io.DataOutput;
	import java.io.IOException;
	import java.util.ArrayList;
	import java.util.Collections;
	import java.util.List;

	import org.apache.logging.log4j.Logger;

	import org.apache.geode.DataSerializer;
	import org.apache.geode.internal.cache.versions.VersionTag;
	import org.apache.geode.internal.offheap.OffHeapHelper;
	import org.apache.geode.internal.offheap.Releasable;
	import org.apache.geode.internal.serialization.DataSerializableFixedID;
	import org.apache.geode.internal.serialization.DeserializationContext;
	import org.apache.geode.internal.serialization.SerializationContext;
	import org.apache.geode.internal.serialization.Version;
	import org.apache.geode.logging.internal.log4j.api.LogService;

	/**
	* Encapsulates list containing objects, serialized objects, raw byte arrays, or exceptions. It can
	* optionally also hold the list of associated keys. Assumes that keys are either provided for all
	* entries or for none.
	*
	* @since GemFire 5.7
	*/
	public class ObjectPartList implements DataSerializableFixedID, Releasable {
	private static final Logger logger = LogService.getLogger();

	protected static final byte BYTES = 0;

	protected static final byte OBJECT = 1;

	protected static final byte EXCEPTION = 2;

	protected static final byte KEY_NOT_AT_SERVER = 3;

	protected byte[] objectTypeArray;

	protected boolean hasKeys;

	protected List keys;

	protected List objects;

	public void addPart(Object key, Object value, byte objectType, VersionTag versionTag) {
	int size = this.objects.size();
	int maxSize = this.objectTypeArray.length;
	if (size >= maxSize) {
	throw new IndexOutOfBoundsException("Cannot add object part beyond " + maxSize + " elements");
	}
	if (this.hasKeys) {
	if (key == null) {
	throw new IllegalArgumentException("Cannot add null key");
	}
	this.keys.add(key);
	}
	this.objectTypeArray[size] = objectType;
	this.objects.add(value);
	}

	// public methods

	public ObjectPartList() {
	this.objectTypeArray = null;
	this.hasKeys = false;
	this.keys = null;
	this.objects = new ArrayList();
	}

	public ObjectPartList(int maxSize, boolean hasKeys) {
	if (maxSize <= 0) {
	throw new IllegalArgumentException(
	"Invalid size " + maxSize + " to ObjectPartList constructor");
	}
	this.objectTypeArray = new byte[maxSize];
	this.hasKeys = hasKeys;
	if (hasKeys) {
	this.keys = new ArrayList();
	} else {
	this.keys = null;
	}
	this.objects = new ArrayList();
	}

	public void addObjectPart(Object key, Object value, boolean isObject, VersionTag versionTag) {
	addPart(key, value, isObject ? OBJECT : BYTES, versionTag);
	}

	public void addExceptionPart(Object key, Exception ex) {
	addPart(key, ex, EXCEPTION, null);
	}

	public void addObjectPartForAbsentKey(Object key, Object value) {
	// ObjectPartList is for clients < version 6.5.0, which didn't support this setting
	throw new IllegalAccessError("inappropriate use of ObjectPartList");
	}



	public void addAll(ObjectPartList other) {
	if (this.hasKeys) {
	if (other.keys != null) {
	if (this.keys == null) {
	this.keys = new ArrayList(other.keys);
	} else {
	this.keys.addAll(other.keys);
	}
	}
	} else if (other.hasKeys) {
	this.hasKeys = true;
	this.keys = new ArrayList(other.keys);
	}
	this.objects.addAll(other.objects);
	}

	public List<Object> getKeys() {
	if (this.keys == null) {
	return Collections.emptyList();
	} else {
	return Collections.unmodifiableList(this.keys);
	}
	}

	/** unprotected access to the keys collection, which may be null */
	public List getKeysForTest() {
	return this.keys;
	}

	public List getObjects() {
	if (this.objects == null) {
	return Collections.emptyList();
	} else {
	return Collections.unmodifiableList(this.objects);
	}
	}

	/** unprotected access to the objects collection, which may be null */
	public List getObjectsForTest() {
	return this.objects;
	}

	public int size() {
	// some lists have only keys and some have only objects, so we need to choose
	// the correct collection to query
	if (this.hasKeys) {
	return this.keys.size();
	} else {
	return this.objects.size();
	}
	}

	public void reinit(int maxSize) {
	if (maxSize <= 0) {
	throw new IllegalArgumentException("Invalid size " + maxSize + " to ObjectPartList.reinit");
	}
	this.objectTypeArray = new byte[maxSize];
	this.objects.clear();
	this.keys.clear();
	}

	public void clear() {
	release();
	this.objects.clear();
	if (this.keys != null) {
	this.keys.clear();
	}
	}

	@Override
	public void toData(DataOutput out,
	SerializationContext context) throws IOException {
	out.writeBoolean(this.hasKeys);
	if (this.objectTypeArray != null) {
	int numObjects = this.objects.size();
	out.writeInt(numObjects);
	for (int index = 0; index < numObjects; ++index) {
	Object value = this.objects.get(index);
	byte objectType = this.objectTypeArray[index];
	if (this.hasKeys) {
	context.getSerializer().writeObject(this.keys.get(index), out);
	}
	out.writeBoolean(objectType == EXCEPTION);
	if (objectType == OBJECT && value instanceof byte[]) {
	out.write((byte[]) value);
	} else if (objectType == EXCEPTION) {
	// write exception as byte array so native clients can skip it
	DataSerializer.writeByteArray(CacheServerHelper.serialize(value), out);
	// write the exception string for native clients
	DataSerializer.writeString(value.toString(), out);
	} else {
	context.getSerializer().writeObject(value, out);
	}
	}
	} else {
	out.writeInt(0);
	}
	}

	@Override
	public void fromData(DataInput in,
	DeserializationContext context) throws IOException, ClassNotFoundException {
	this.hasKeys = in.readBoolean();
	if (this.hasKeys) {
	this.keys = new ArrayList();
	}
	int numObjects = in.readInt();
	if (numObjects > 0) {
	for (int index = 0; index < numObjects; ++index) {
	if (this.hasKeys) {
	Object key = context.getDeserializer().readObject(in);
	this.keys.add(key);
	}
	boolean isException = in.readBoolean();
	Object value;
	if (isException) {
	byte[] exBytes = DataSerializer.readByteArray(in);
	value = CacheServerHelper.deserialize(exBytes);
	// ignore the exception string meant for native clients
	DataSerializer.readString(in);
	} else {
	value = context.getDeserializer().readObject(in);
	}
	this.objects.add(value);
	}
	}
	}

	@Override
	public int getDSFID() {
	return DataSerializableFixedID.OBJECT_PART_LIST;
	}

	@Override
	public Version[] getSerializationVersions() {
	return null;
	}

	@Override
	public void release() {
	for (Object v : this.objects) {
	OffHeapHelper.release(v);
	}
	}
	}