net.spy.memcached.collection.CollectionPipedUpdate Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arcus-java-client Show documentation
Java client for Arcus memcached
There is a newer version: 1.13.2
/*
 * arcus-java-client : Arcus Java client
 * Copyright 2010-2014 NAVER Corp.
 *
 * Licensed 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 net.spy.memcached.collection;

import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import net.spy.memcached.CachedData;
import net.spy.memcached.KeyUtil;
import net.spy.memcached.collection.ElementFlagFilter.BitWiseOperands;
import net.spy.memcached.transcoders.Transcoder;

public abstract class CollectionPipedUpdate extends CollectionObject {

	public static final String PIPE = "pipe";
	public static final int MAX_PIPED_ITEM_COUNT = 500;

	protected String key;
	protected Transcoder tc;
	protected int itemCount;

	public abstract ByteBuffer getAsciiCommand();

	public abstract ByteBuffer getBinaryCommand();

	public static class BTreePipedUpdate extends CollectionPipedUpdate {

		private static final String COMMAND = "bop update";
		private List> elements;

		public BTreePipedUpdate(String key, List> elements,
				Transcoder tc) {
			this.key = key;
			this.elements = elements;
			this.tc = tc;
			this.itemCount = elements.size();
		}

		public ByteBuffer getAsciiCommand() {
			int capacity = 0;

			// decode parameters
			List decodedList = new ArrayList(elements.size());
			CachedData cd = null;
			for (Element each : elements) {
				if (each.getValue() != null) {
					cd = tc.encode(each.getValue());
					decodedList.add(cd.getData());
				} else {
					decodedList.add(null);
				}
			}

			// estimate the buffer capacity
			int i = 0;
			ElementFlagUpdate eflagUpdate;
			byte[] value;
			StringBuilder b;

			for (Element each : elements) {
				eflagUpdate = each.getElementFlagUpdate();
				if (eflagUpdate != null) {
					// eflag
					capacity += KeyUtil.getKeyBytes(eflagUpdate.getElementFlagByHex()).length;
					// fwhere bitwop
					if (eflagUpdate.getElementFlagOffset() > -1) {
						capacity += 6;
					}
				}

				capacity += KeyUtil.getKeyBytes(key).length;
				capacity += KeyUtil.getKeyBytes((each.isByteArraysBkey() ? each
						.getBkeyByHex() : String.valueOf(each.getLongBkey()))).length;
				if (decodedList.get(i) != null) {
					capacity += decodedList.get(i++).length;
				}
				capacity += 64;
			}

			// allocate the buffer
			ByteBuffer bb = ByteBuffer.allocate(capacity);

			// create ascii operation string
			i = 0;

			Iterator> iterator = elements.iterator();
			while (iterator.hasNext()) {
				Element element = iterator.next();

				value = decodedList.get(i++);
				eflagUpdate = element.getElementFlagUpdate();
				b = new StringBuilder();

				// has element eflag update
				if (eflagUpdate != null) {
					// use fwhere bitop
					if (eflagUpdate.getElementFlagOffset() > -1 && eflagUpdate.getBitOp() != null) {
						b.append(eflagUpdate.getElementFlagOffset()).append(" ");
						b.append(eflagUpdate.getBitOp()).append(" ");
					}
					b.append(eflagUpdate.getElementFlagByHex());
				}

				setArguments(bb, COMMAND, key,
						(element.isByteArraysBkey() ? element.getBkeyByHex()
								: String.valueOf(element.getLongBkey())),
						b.toString(), (value == null ? -1 : value.length),
						(iterator.hasNext()) ? PIPE : "");
				if (value != null) {
					if (value.length > 0) {
						bb.put(value);
					}
					bb.put(CRLF);
				}
			}

			// flip the buffer
			bb.flip();

			return bb;
		}

		public ByteBuffer getBinaryCommand() {
			throw new RuntimeException("not supported in binary protocol yet.");
		}
	}

	public String getKey() {
		return key;
	}

	public void setKey(String key) {
		this.key = key;
	}

	public int getItemCount() {
		return this.itemCount;
	}
}