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

Go to download
/*
 * 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.util.Map;

import net.spy.memcached.util.BTreeUtil;

public class ExtendedBTreeGet extends CollectionGet {

	private static final String command = "bop get";
	
	protected String range;
	protected int offset = -1;
	protected int count = -1;
	protected Map map;

	protected byte[] subkey;
	
	protected ElementFlagFilter elementFlagFilter;
	
	public ExtendedBTreeGet(byte[] from, byte[] to, int offset,
			int count, boolean delete, boolean dropIfEmpty,
			ElementFlagFilter elementFlagFilter) {
		this.headerCount = 2;
		this.range = BTreeUtil.toHex(from) + ".." + BTreeUtil.toHex(to);
		this.offset = offset;
		this.count = count;
		this.delete = delete;
		this.dropIfEmpty = dropIfEmpty;
		this.elementFlagFilter = elementFlagFilter;
	}
	
	public ElementFlagFilter getElementFlagFilter() {
		return elementFlagFilter;
	}
	
	public String getRange() {
		return range;
	}

	public void setRange(String range) {
		this.range = range;
	}
	
	public int getCount() {
		return count;
	}

	public void setCount(int count) {
		this.count = count;
	}
	
	public Map getMap() {
		return map;
	}
	
	public byte[] getLongSubkey() {
		return subkey;
	}
	
	public String stringify() {
		if (str != null) return str;
		
		StringBuilder b = new StringBuilder();
		b.append(range);
		
		if (elementFlagFilter != null) b.append(" ").append(elementFlagFilter.toString());
		if (offset > 0) b.append(" ").append(offset);
		if (count > 0) b.append(" ").append(count);
		if (delete && dropIfEmpty) b.append(" drop");
		if (delete && !dropIfEmpty) b.append(" delete");

		str = b.toString();
		return str;
	}

	public String getCommand() {
		return command;
	}
	
	public void resetHeaderCount(int count) {
		this.headerCount = count;
	}
	
	private int headerParseStep = 1; 
	
	private boolean elementFlagExists = false;
	
	public boolean eachRecordParseCompleted() {
		if (elementFlagExists) {
			return headerParseStep == 1;
		} else {
			return true;
		}
	}
	
	@Override
	public boolean headerReady(int spaceCount) {
		return spaceCount == 2 || spaceCount == 3;
	}
	
	public void decodeItemHeader(String itemHeader) {
		String[] splited = itemHeader.split(" ");

		if (headerParseStep == 1) {
			// found element flag.
			if (splited[1].startsWith("0x")) {
				this.elementFlagExists = true;
				this.subkey = BTreeUtil.hexStringToByteArrays(splited[0].substring(2));
				this.elementFlag = BTreeUtil.hexStringToByteArrays(splited[1].substring(2));
//				this.headerCount++;
				headerParseStep = 2;
			} else {
				this.subkey = BTreeUtil.hexStringToByteArrays(splited[0].substring(2));
				this.dataLength = Integer.parseInt(splited[1]);
			}
		} else {
			this.headerParseStep = 1;
			this.dataLength = Integer.parseInt(splited[1]);
		}
	}
}