/*
Copyright 2009 Janne Hietamäki
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 mungbean.protocol.message;
import static mungbean.CollectionUtil.map;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import mungbean.ListQueryCallback;
import mungbean.protocol.LittleEndianDataReader;
import mungbean.protocol.LittleEndianDataWriter;
import mungbean.protocol.bson.AbstractBSONCoders;
import mungbean.protocol.bson.BSONMap;
import mungbean.protocol.bson.MapBSONCoders;
import mungbean.query.Query;
public class CommandRequest extends MongoRequest<CommandResponse> {
public static final AbstractBSONCoders DEFAULT_CODERS = new MapBSONCoders();
private static final BSONMap RESPONSE_CODER = new BSONMap();
private final QueryRequest<Map<String, Object>> query;
private final ListQueryCallback<Map<String, Object>> v = new ListQueryCallback<Map<String, Object>>();
public CommandRequest(String dbName, Map<String, Object> params, AbstractBSONCoders coders) {
HashMap<String, Object> orders = new HashMap<String, Object>();
query = new QueryRequest<Map<String, Object>>(dbName + ".$cmd", new QueryOptionsBuilder(), params, orders, coders, RESPONSE_CODER);
}
public CommandRequest(String dbName, String command) {
this(dbName, map(command, 1D), DEFAULT_CODERS);
}
public CommandRequest(String command) {
query = new QueryRequest<Map<String, Object>>("$cmd", new QueryOptionsBuilder(), new Query().setLimit(1).field(command).is(1D), DEFAULT_CODERS, RESPONSE_CODER);
}
@Override
public int length() {
return query.length();
}
@Override
public CommandResponse readResponse(LittleEndianDataReader reader) {
QueryResponse<Map<String, Object>> response = query.readResponse(reader);
response.readResponse(v);
List<Map<String, Object>> values = v.values();
if (values.isEmpty()) {
return new CommandResponse(new HashMap<String, Object>());
}
return new CommandResponse(values.get(0));
}
@Override
public void send(LittleEndianDataWriter writer) {
query.send(writer);
}
@Override
public RequestOpCode type() {
return query.type();
}
}