/*
* Copyright 2013 Basho Technologies Inc.
*
* 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.client.core.operations;
import com.basho.riak.client.core.FutureOperation;
import com.basho.riak.client.core.RiakMessage;
import com.basho.riak.client.core.query.search.YokozunaIndex;
import com.basho.riak.protobuf.RiakMessageCodes;
import com.basho.riak.protobuf.RiakYokozunaPB;
import com.google.protobuf.ByteString;
import java.util.List;
/**
*
* @author Brian Roach <roach at basho dot com>
*/
public class YzPutIndexOperation extends FutureOperation<Void, Void, YokozunaIndex>
{
private final RiakYokozunaPB.RpbYokozunaIndexPutReq.Builder reqBuilder;
private final YokozunaIndex index;
private YzPutIndexOperation(Builder builder)
{
this.reqBuilder = builder.reqBuilder;
this.index = builder.index;
}
@Override
protected Void convert(List<Void> rawResponse)
{
return null;
}
@Override
protected RiakMessage createChannelMessage()
{
RiakYokozunaPB.RpbYokozunaIndexPutReq req = reqBuilder.build();
return new RiakMessage(RiakMessageCodes.MSG_YokozunaIndexPutReq, req.toByteArray());
}
@Override
protected Void decode(RiakMessage rawMessage)
{
Operations.checkMessageType(rawMessage, RiakMessageCodes.MSG_PutResp);
return null;
}
@Override
public YokozunaIndex getQueryInfo()
{
return index;
}
public static class Builder
{
private final RiakYokozunaPB.RpbYokozunaIndexPutReq.Builder reqBuilder =
RiakYokozunaPB.RpbYokozunaIndexPutReq.newBuilder();
private final YokozunaIndex index;
public Builder(YokozunaIndex index)
{
if (null == index)
{
throw new IllegalArgumentException("Index can not be null");
}
RiakYokozunaPB.RpbYokozunaIndex.Builder indexBuilder =
RiakYokozunaPB.RpbYokozunaIndex.newBuilder();
indexBuilder.setName(ByteString.copyFromUtf8(index.getName()));
// A null schema is valid; the default will be used
if (index.getSchema() != null)
{
indexBuilder.setSchema(ByteString.copyFromUtf8(index.getSchema()));
}
reqBuilder.setIndex(indexBuilder);
this.index = index;
}
public YzPutIndexOperation build()
{
return new YzPutIndexOperation(this);
}
}
}