com.qwazr.database.store.keys.ColumnStoreKey Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of qwazr-database Show documentation
Database Engine for QWAZR
There is a newer version: 1.5.0
/*
 * Copyright 2015-2018 Emmanuel Keller / QWAZR
 * 
 * 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.qwazr.database.store.keys;

import com.qwazr.database.model.InternalColumnDefinition;
import com.qwazr.database.store.ByteConverter;
import com.qwazr.database.store.KeyStore;
import com.qwazr.database.store.ValueConsumer;
import com.qwazr.server.ServerException;
import com.qwazr.utils.ArrayUtils;

import javax.ws.rs.core.Response;
import java.io.DataOutputStream;
import java.io.IOException;
import java.util.Collection;

final public class ColumnStoreKey extends KeyAbstract {

	final InternalColumnDefinition columnDef;
	final int docId;

	protected ColumnStoreKey(InternalColumnDefinition columnDef, int docId, ByteConverter byteConverter) {
		super(KeyEnum.COLUMN_STORE, byteConverter);
		this.docId = docId;
		this.columnDef = columnDef;
	}

	@Override
	final public void buildKey(final DataOutputStream output) throws IOException {
		super.buildKey(output);
		output.writeInt(columnDef.columnId);
		output.writeInt(docId);
	}

	final public void forEach(final KeyStore store, final ValueConsumer consumer) throws IOException {
		byteConverter.forEach(getValue(store), consumer);
	}

	final public void forFirst(final KeyStore store, final ValueConsumer consumer) throws IOException {
		byteConverter.forFirst(getValue(store), consumer);
	}

	public static ColumnStoreKey newInstance(final InternalColumnDefinition colDef, final int docId) {
		switch (colDef.type) {
		case DOUBLE:
			return new ColumnStoreKey<>(colDef, docId, ByteConverter.DoubleArrayByteConverter.INSTANCE);
		case INTEGER:
			return new ColumnStoreKey<>(colDef, docId, ByteConverter.IntArrayByteConverter.INSTANCE);
		case LONG:
			return new ColumnStoreKey<>(colDef, docId, ByteConverter.LongArrayByteConverter.INSTANCE);
		case STRING:
			return new ColumnStoreKey<>(colDef, docId, ByteConverter.StringArrayByteConverter.INSTANCE);
		}
		throw new ServerException(Response.Status.NOT_ACCEPTABLE, "unknown type: " + colDef.type);
	}

	final public void setObjectValue(KeyStore store, Object value) throws IOException {
		if (value == null)
			return;
		if (value instanceof Collection)
			setValue(store, (V) collectionToArray((Collection) value));
		else if (value.getClass().isArray())
			setValue(store, (V) value);
		else
			setValue(store, (V) objectToArray(value));
	}

	private Object collectionToArray(Collection collection) {
		switch (columnDef.type) {
		case DOUBLE:
			return ArrayUtils.toPrimitiveDouble((Collection) collection);
		case INTEGER:
			return ArrayUtils.toPrimitiveInt((Collection) collection);
		case LONG:
			return ArrayUtils.toPrimitiveLong((Collection) collection);
		case STRING:
			return collection.toArray(new String[collection.size()]);
		}
		throw new ServerException(Response.Status.NOT_ACCEPTABLE, "unknown type: " + columnDef.type);
	}

	private Object objectToArray(Object object) {
		switch (columnDef.type) {
		case DOUBLE:
			return new double[] { ((Number) object).doubleValue() };
		case INTEGER:
			return new int[] { ((Number) object).intValue() };
		case LONG:
			return new long[] { ((Number) object).longValue() };
		case STRING:
			return new String[] { object.toString() };
		}
		throw new ServerException(Response.Status.NOT_ACCEPTABLE, "unknown type: " + columnDef.type);
	}
}