List of usage examples for org.apache.cassandra.utils ByteBufferUtil getArray
public static byte[] getArray(ByteBuffer buffer)
From source file:com.dse.pig.udfs.AbstractCassandraStorage.java
License:Apache License
/** set the value to the position of the tuple */ protected void setTupleValue(Tuple pair, int position, Object value) throws ExecException { if (value instanceof BigInteger) pair.set(position, ((BigInteger) value).intValue()); else if (value instanceof ByteBuffer) pair.set(position, new DataByteArray(ByteBufferUtil.getArray((ByteBuffer) value))); else if (value instanceof Date) pair.set(position, DateType.instance.decompose((Date) value).getLong()); else if (value instanceof java.util.UUID) pair.set(position, value.toString()); else// ww w . j a v a 2 s. co m pair.set(position, value.toString()); }
From source file:com.github.adejanovski.cassandra.jdbc.CassandraRowId.java
License:Apache License
public byte[] getBytes() { return ByteBufferUtil.getArray(bytes); }
From source file:info.archinnov.achilles.logger.ThriftLoggerHelper.java
License:Apache License
public static String format(List<Component<?>> components) { String formatted = "[]"; if (components != null && components.size() > 0) { List<String> componentsText = new ArrayList<String>(); int componentNb = components.size(); for (int i = 0; i < componentNb; i++) { Component<?> component = components.get(i); String componentValue; if (component.getSerializer() == ThriftSerializerUtils.BYTE_SRZ) { componentValue = ByteBufferUtil.getArray(component.getBytes())[0] + ""; } else { componentValue = component.getValue(component.getSerializer()).toString(); }/*www .ja va 2 s. c om*/ if (i == componentNb - 1) { componentsText.add(componentValue + "(" + component.getEquality().name() + ")"); } else { componentsText.add(componentValue); } } formatted = '[' + StringUtils.join(componentsText, ':') + ']'; ; } return formatted; }
From source file:org.apache.hadoop.hive.cassandra.CassandraPushdownPredicate.java
License:Apache License
private static IndexExpression translateSearchCondition(IndexSearchCondition condition, Set<ColumnDef> columnInfos) throws IOException { IndexExpression expr = new IndexExpression(); String columnName = condition.getColumnDesc().getColumn(); expr.setColumn_name(columnName.getBytes()); expr.setOp(getIndexOperator(condition.getComparisonOp())); ExprNodeConstantEvaluator eval = new ExprNodeConstantEvaluator(condition.getConstantDesc()); byte[] value; try {//from ww w. ja v a 2 s . c om ObjectInspector objInspector = eval.initialize(null); Object writable = eval.evaluate(null); ByteStream.Output serializeStream = new ByteStream.Output(); PrimitiveObjectInspector poi = (PrimitiveObjectInspector) objInspector; AbstractType validator = getValidator(columnInfos, columnName); ByteBuffer bytes = getIndexExpressionValue(condition.getConstantDesc(), poi, writable, validator); serializeStream.write(ByteBufferUtil.getArray(bytes)); value = new byte[serializeStream.getCount()]; System.arraycopy(serializeStream.getData(), 0, value, 0, serializeStream.getCount()); } catch (HiveException e) { throw new IOException(e); } expr.setValue(value); logger.info("IndexExpression.value : {}", new String(expr.getValue())); return expr; }
From source file:org.apache.hadoop.hive.cassandra.input.CassandraHiveRecordReader.java
License:Apache License
private BytesWritable convertByteBuffer(ByteBuffer val) { return new BytesWritable(ByteBufferUtil.getArray(val)); }
From source file:org.apache.hadoop.hive.cassandra.input.cql.CqlHiveRecordReader.java
License:Apache License
private BytesWritable convertByteBuffer(ByteBuffer val) { if (val == null) return new BytesWritable(new byte[0]); return new BytesWritable(ByteBufferUtil.getArray(val)); }
From source file:org.apache.hadoop.hive.cassandra.output.CassandraPut.java
License:Apache License
@Override public void write(DataOutput out) throws IOException { out.writeInt(key.remaining());/*from w ww .ja v a 2s . co m*/ out.write(ByteBufferUtil.getArray(key)); out.writeInt(1); out.writeInt(columns.size()); for (CassandraColumn c : columns) { c.write(out); } }
From source file:org.apache.hadoop.hive.cassandra.output.CassandraSuperPut.java
License:Apache License
@Override public void write(DataOutput out) throws IOException { out.writeInt(key.remaining());// w ww. j av a 2s . c om out.write(ByteBufferUtil.getArray(key)); out.writeInt(1); out.writeInt(subColumns.size()); for (CassandraPut c : subColumns) { c.write(out); } }
From source file:org.apache.hadoop.hive.cassandra.output.cql.CqlPut.java
License:Apache License
@Override public void write(DataOutput out) throws IOException { out.writeInt(key.remaining());/* ww w .j a va 2s . c o m*/ out.write(ByteBufferUtil.getArray(key)); out.writeInt(1); out.writeInt(columns.size()); for (CqlColumn c : columns) { c.write(out); } }
From source file:org.apache.hadoop.hive.cassandra.serde.TableMapping.java
License:Apache License
protected boolean serialize(Object obj, ObjectInspector objInspector, ObjectInspector declaredObjInspector, int level) throws IOException { switch (objInspector.getCategory()) { case PRIMITIVE: { //Marshal to expected cassandra format AbstractType validator = LazyCassandraUtils .getCassandraType((PrimitiveObjectInspector) declaredObjInspector); if (validator instanceof BytesType) { BytesWritable bw = ((BinaryObjectInspector) objInspector).getPrimitiveWritableObject(obj); serializeStream.write(bw.getBytes(), 0, bw.getLength()); } else {/*from ww w. j a v a2 s. co m*/ LazyUtils.writePrimitiveUTF8(serializeStream, obj, (PrimitiveObjectInspector) objInspector, escaped, escapeChar, needsEscape); //convert from string to cassandra type if (!declaredObjInspector.getTypeName() .equals(PrimitiveObjectInspectorUtils.stringTypeEntry.typeName)) { ByteBuffer bb = validator.fromString(serializeStream.toString()); serializeStream.reset(); serializeStream.write(ByteBufferUtil.getArray(bb)); } } return true; } case LIST: { char separator = (char) separators[level]; ListObjectInspector loi = (ListObjectInspector) objInspector; List<?> list = loi.getList(obj); ObjectInspector eoi = loi.getListElementObjectInspector(); if (list == null) { return false; } else { for (int i = 0; i < list.size(); i++) { if (i > 0) { serializeStream.write(separator); } serialize(list.get(i), eoi, PrimitiveObjectInspectorFactory.javaStringObjectInspector, level + 1); } } return true; } case MAP: { char separator = (char) separators[level]; char keyValueSeparator = (char) separators[level + 1]; MapObjectInspector moi = (MapObjectInspector) objInspector; ObjectInspector koi = moi.getMapKeyObjectInspector(); ObjectInspector voi = moi.getMapValueObjectInspector(); Map<?, ?> map = moi.getMap(obj); if (map == null) { return false; } else { boolean first = true; for (Map.Entry<?, ?> entry : map.entrySet()) { if (first) { first = false; } else { serializeStream.write(separator); } serialize(entry.getKey(), koi, PrimitiveObjectInspectorFactory.javaStringObjectInspector, level + 2); serializeStream.write(keyValueSeparator); serialize(entry.getValue(), voi, PrimitiveObjectInspectorFactory.javaStringObjectInspector, level + 2); } } return true; } case STRUCT: { char separator = (char) separators[level]; StructObjectInspector soi = (StructObjectInspector) objInspector; List<? extends StructField> fields = soi.getAllStructFieldRefs(); List<Object> list = soi.getStructFieldsDataAsList(obj); if (list == null) { return false; } else { for (int i = 0; i < list.size(); i++) { if (i > 0) { serializeStream.write(separator); } serialize(list.get(i), fields.get(i).getFieldObjectInspector(), PrimitiveObjectInspectorFactory.javaStringObjectInspector, level + 1); } } return true; } } throw new RuntimeException("Unknown category type: " + objInspector.getCategory()); }