/**
* This file is part of riak-java-pb-client
*
* Copyright (c) 2010 by Trifork
*
* 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 com.basho.riak.pbc;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import com.basho.riak.client.http.BinIndex;
import com.basho.riak.client.http.IntIndex;
import com.basho.riak.client.http.RiakIndex;
import com.basho.riak.protobuf.RiakKvPB.RpbContent;
import com.basho.riak.protobuf.RiakPB.RpbPair;
import com.basho.riak.protobuf.RiakKvPB.RpbContent.Builder;
import com.google.protobuf.ByteString;
/**
* PBC model of the data/meta data for a bucket/key entry in Riak
*/
public class RiakObject {
private ByteString vclock;
private ByteString bucket;
private ByteString key;
private boolean deleted;
private ByteString value;
private String contentType;
private List<RiakLink> links = Collections.synchronizedList(new ArrayList<RiakLink>());
private final Object indexLock = new Object();
@SuppressWarnings("rawtypes") private List<RiakIndex> indexes = new ArrayList<RiakIndex>();
private String vtag;
private String contentEncoding;
private String charset;
private Object userMetaDataLock = new Object();
private Map<String,String> userMetaData = new LinkedHashMap<String, String>();
private Integer lastModified;
private Integer lastModifiedUsec;
RiakObject(ByteString vclock, ByteString bucket, ByteString key, RpbContent content) {
this.vclock = vclock;
this.bucket = bucket;
this.key = key;
this.deleted = content.getDeleted();
this.value = content.getValue();
this.contentType = str(content.getContentType());
this.charset = str(content.getCharset());
this.contentEncoding = str(content.getContentEncoding());
this.vtag = str(content.getVtag());
this.links = content.getLinksCount() == 0
? Collections.synchronizedList(new ArrayList<RiakLink>())
: Collections.synchronizedList(RiakLink.decode(content.getLinksList()));
if (content.hasLastMod()) {
this.lastModified = new Integer(content.getLastMod());
this.lastModifiedUsec = new Integer(content.getLastModUsecs());
}
if (content.getUsermetaCount() > 0) {
Map<String, String> tmpUserMetaData = new LinkedHashMap<String, String>();
for (int i = 0; i < content.getUsermetaCount(); i++) {
RpbPair um = content.getUsermeta(i);
tmpUserMetaData.put(um.getKey().toStringUtf8(),
str(um.getValue()));
}
synchronized (userMetaDataLock) {
userMetaData.putAll(tmpUserMetaData);
}
}
if (content.getIndexesCount() > 0) {
@SuppressWarnings("rawtypes") List<RiakIndex> indexes = new ArrayList<RiakIndex>();
for (RpbPair p : content.getIndexesList()) {
String name = p.getKey().toStringUtf8();
String value = p.getValue().toStringUtf8();
if (name.endsWith(BinIndex.SUFFIX)) {
indexes.add(new BinIndex(name, value));
} else if (name.endsWith(IntIndex.SUFFIX)) {
indexes.add(new IntIndex(name, Long.parseLong(value)));
} else {
throw new RuntimeException("unkown index type " + name);
}
}
synchronized (indexLock) {
this.indexes.addAll(indexes);
}
}
}
public RiakObject(ByteString vclock, ByteString bucket, ByteString key, ByteString content) {
this.bucket = bucket;
this.key = key;
this.value = content;
this.vclock = vclock;
}
public RiakObject(ByteString bucket, ByteString key, ByteString content) {
this.bucket = bucket;
this.key = key;
this.value = content;
}
public RiakObject(String bucket, String key, byte[] content) {
this.bucket = ByteString.copyFromUtf8(bucket);
this.key = ByteString.copyFromUtf8(key);
this.value = ByteString.copyFrom(content);
}
public RiakObject(String bucket, String key, String content) {
this.bucket = ByteString.copyFromUtf8(bucket);
this.key = ByteString.copyFromUtf8(key);
this.value = ByteString.copyFromUtf8(content);
}
private String str(ByteString str) {
if (str == null) return null;
return str.toStringUtf8();
}
public ByteString getBucketBS() {
return bucket;
}
public String getBucket() {
return bucket.toStringUtf8();
}
public ByteString getKeyBS() {
return key;
}
public String getKey() {
if (key != null) {
return key.toStringUtf8();
} else {
return null;
}
}
public boolean getDeleted() {
return this.deleted;
}
public ByteString getVclock() {
return vclock;
}
public ByteString getValue(){
return value;
}
RpbContent buildContent() {
Builder b =
RpbContent.newBuilder()
.setValue(value);
if (contentType != null) {
b.setContentType(ByteString.copyFromUtf8(contentType));
}
if (charset != null) {
b.setCharset(ByteString.copyFromUtf8(charset));
}
if (contentEncoding != null) {
b.setContentEncoding(ByteString.copyFromUtf8(contentEncoding));
}
if (vtag != null) {
b.setVtag(ByteString.copyFromUtf8(vtag));
}
if (links.size() != 0) {
RiakLink[] localLinks = null;
synchronized (links) {
localLinks = links.toArray(new RiakLink[links.size()]);
}
for (RiakLink l : localLinks) {
b.addLinks(l.build());
}
}
if (lastModified != null) {
b.setLastMod(lastModified);
}
if (lastModifiedUsec != null) {
b.setLastModUsecs(lastModifiedUsec);
}
final Map<String, String> tmpUserMetaData = new LinkedHashMap<String, String>();
synchronized (userMetaDataLock) {
tmpUserMetaData.putAll(userMetaData);
}
if (tmpUserMetaData != null && !tmpUserMetaData.isEmpty()) {
for (Map.Entry<String, String> ent : tmpUserMetaData.entrySet()) {
ByteString key = ByteString.copyFromUtf8(ent.getKey());
com.basho.riak.protobuf.RiakPB.RpbPair.Builder pb =
com.basho.riak.protobuf.RiakPB.RpbPair.newBuilder().setKey(key);
if (ent.getValue() != null) {
pb.setValue(ByteString.copyFromUtf8(ent.getValue()));
}
b.addUsermeta(pb);
}
}
synchronized (indexLock) {
for (@SuppressWarnings("rawtypes") RiakIndex i : indexes) {
b.addIndexes(RpbPair.newBuilder()
.setKey(ByteString.copyFromUtf8(i.getName()))
.setValue(ByteString.copyFromUtf8(i.getValue().toString()))
.build());
}
}
return b.build();
}
public String getVtag() {
return this.vtag;
}
public void setContentType(String contentType) {
this.contentType = contentType;
}
public void setCharset(String charset) {
this.charset = charset;
}
public void addLink(String tag, String bucket, String key) {
links.add(new RiakLink(bucket, key, tag));
}
public void addLink(ByteString tag, ByteString bucket, ByteString key) {
links.add(new RiakLink(bucket, key, tag));
}
@SuppressWarnings("unchecked") public List<RiakLink> getLinks() {
return links != null ? Collections.unmodifiableList(links) : Collections.EMPTY_LIST;
}
/**
* Return a copy of the user meta data map (does not read or write through
* to map backing RiakObject)
*
* @return a Map of the user map as it is at method call time
*/
public Map<String, String> getUsermeta() {
synchronized (userMetaDataLock) {
return new LinkedHashMap<String, String>(userMetaData);
}
}
/**
* Add an item to the user meta data for this RiakObject.
* @param key the key of the user meta data item
* @param value the user meta data item
* @return this RiakObject
*/
public RiakObject addUsermetaItem(String key, String value) {
synchronized (userMetaDataLock) {
userMetaData.put(key, value);
}
return this;
}
/**
* @return the lastModified
*/
public Date getLastModified() {
Date d = null;
if (lastModified != null && lastModifiedUsec != null) {
long t = (lastModified * 1000L ) + (lastModifiedUsec / 1000L);
d = new Date(t);
}
return d;
}
/**
* @return the content type
*/
public String getContentType() {
return this.contentType;
}
public String getCharset() {
return this.charset;
}
/**
* @return a *copy* of the list of {@link RiakIndex}es for this object
*/
@SuppressWarnings("rawtypes") public List<RiakIndex> getIndexes() {
synchronized (indexLock) {
return new ArrayList<RiakIndex>(indexes);
}
}
/**
* Add a binary index to the object
*
* @param name
* of the index
* @param value
* the value to add to the index
* @return this
*/
public RiakObject addIndex(String name, String value) {
synchronized (indexLock) {
indexes.add(new BinIndex(name, value));
}
return this;
}
/**
* Add an int index to this object
*
* @param name
* of the index
* @param value
* the value to add to the index
* @return this
*/
public RiakObject addIndex(String name, long value) {
synchronized (indexLock) {
indexes.add(new IntIndex(name, value));
}
return this;
}
}