package com.tesora.dve.queryplan;
/*
* #%L
* Tesora Inc.
* Database Virtualization Engine
* %%
* Copyright (C) 2011 - 2014 Tesora Inc.
* %%
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License, version 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
* #L%
*/
import java.util.Collection;
import java.util.Iterator;
import com.tesora.dve.common.MultiMap;
import com.tesora.dve.common.catalog.StorageSite;
import com.tesora.dve.comms.client.messages.MessageType;
import com.tesora.dve.comms.client.messages.MessageVersion;
import com.tesora.dve.concurrent.CompletionHandle;
import com.tesora.dve.concurrent.PEDefaultPromise;
import com.tesora.dve.server.connectionmanager.SSContext;
import com.tesora.dve.server.messaging.SQLCommand;
import com.tesora.dve.server.messaging.WorkerExecuteRequest;
import com.tesora.dve.server.statistics.manager.LogSiteStatisticRequest;
import com.tesora.dve.server.statistics.SiteStatKey.OperationClass;
import com.tesora.dve.worker.Worker;
public class WorkerMultiInsertRequest extends WorkerExecuteRequest {
private static final long serialVersionUID = 1L;
private final MultiMap<StorageSite, SQLCommand> mappedInserts;
public WorkerMultiInsertRequest(SSContext ctx, MultiMap<StorageSite, SQLCommand> mappedInserts) {
super(ctx, SQLCommand.EMPTY);
this.mappedInserts = mappedInserts;
}
@Override
public void executeRequest(Worker w, CompletionHandle<Boolean> callersResults) {
try {
Collection<SQLCommand> cmds = mappedInserts.get(w.getWorkerSite());
if ( cmds != null ) {
Iterator<SQLCommand> commandIterator = cmds.iterator();
executeNextInsert(w, callersResults, commandIterator);
} else {
callersResults.success(true);
}
} catch (Exception e){
callersResults.failure(e);
}
}
private void executeNextInsert(final Worker w, final CompletionHandle<Boolean> callersResults, final Iterator<SQLCommand> commandIterator) {
if (!commandIterator.hasNext()){
callersResults.success(true);
return;
}
SQLCommand sqlCommand = commandIterator.next();
PEDefaultPromise<Boolean> oneInsert = new PEDefaultPromise<Boolean>(){
@Override
public void success(Boolean returnValue) {
//this insert was OK, do the next one.
executeNextInsert(w, callersResults,commandIterator);
}
@Override
public void failure(Exception t) {
callersResults.failure(t);
}
};
executeStatement(w, sqlCommand, oneInsert);
}
@Override
public LogSiteStatisticRequest getStatisticsNotice() {
return new LogSiteStatisticRequest(OperationClass.EXECUTE);
}
@Override
public MessageType getMessageType() {
return null;
}
@Override
public MessageVersion getVersion() {
return MessageVersion.VERSION1;
}
}