com.arangodb.velocypack.internal.VPackAttributeTranslatorImpl Maven / Gradle / Ivy
/*
* DISCLAIMER
*
* Copyright 2016 ArangoDB GmbH, Cologne, Germany
*
* 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.
*
* Copyright holder is ArangoDB GmbH, Cologne, Germany
*/
package com.arangodb.velocypack.internal;
import java.util.HashMap;
import java.util.Map;
import com.arangodb.velocypack.VPackAttributeTranslator;
import com.arangodb.velocypack.VPackBuilder;
import com.arangodb.velocypack.VPackSlice;
import com.arangodb.velocypack.ValueType;
import com.arangodb.velocypack.exception.VPackException;
/**
* @author Mark Vollmary
*
*/
public class VPackAttributeTranslatorImpl implements VPackAttributeTranslator {
private static final String KEY = "_key";
private static final String REV = "_rev";
private static final String ID = "_id";
private static final String FROM = "_from";
private static final String TO = "_to";
private static final byte KEY_ATTRIBUTE = 0x31;
private static final byte REV_ATTRIBUTE = 0x32;
private static final byte ID_ATTRIBUTE = 0x33;
private static final byte FROM_ATTRIBUTE = 0x34;
private static final byte TO_ATTRIBUTE = 0x35;
private static final byte ATTRIBUTE_BASE = 0x30;
private VPackBuilder builder;
private final Map attributeToKey;
private final Map keyToAttribute;
public VPackAttributeTranslatorImpl() {
super();
builder = null;
attributeToKey = new HashMap<>();
keyToAttribute = new HashMap<>();
try {
add(KEY, KEY_ATTRIBUTE - ATTRIBUTE_BASE);
add(REV, REV_ATTRIBUTE - ATTRIBUTE_BASE);
add(ID, ID_ATTRIBUTE - ATTRIBUTE_BASE);
add(FROM, FROM_ATTRIBUTE - ATTRIBUTE_BASE);
add(TO, TO_ATTRIBUTE - ATTRIBUTE_BASE);
seal();
} catch (final VPackException e) {
throw new RuntimeException(e);
}
}
@Override
public void add(final String attribute, final int key) throws VPackException {
if (builder == null) {
builder = new VPackBuilder();
builder.add(ValueType.OBJECT);
}
builder.add(attribute, key);
}
@Override
public void seal() throws VPackException {
if (builder == null) {
return;
}
builder.close();
final VPackSlice slice = builder.slice();
for (int i = 0; i < slice.getLength(); i++) {
final VPackSlice key = slice.keyAt(i);
final VPackSlice value = slice.valueAt(i);
attributeToKey.put(key.getAsString(), value);
keyToAttribute.put(value.getAsInt(), key);
}
}
@Override
public VPackSlice translate(final String attribute) {
return attributeToKey.get(attribute);
}
@Override
public VPackSlice translate(final int key) {
return keyToAttribute.get(key);
}
}