//###
//# #%L
//# mosaic-java-benchmarks
//# %%
//# Copyright (C) 2010 - 2012 Seconda Università di Napoli
//#Authors: Massimiliano Rak, Giuseppe Aversano
//# %%
//# 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.
//# #L%
//###
package eu.mosaic_cloud.benchmarks;
import eu.mosaic_cloud.cloudlets.connectors.kvstore.IKvStoreConnector;
import eu.mosaic_cloud.cloudlets.connectors.kvstore.IKvStoreConnectorFactory;
import eu.mosaic_cloud.cloudlets.connectors.kvstore.KvStoreCallbackCompletionArguments;
import eu.mosaic_cloud.cloudlets.connectors.queue.amqp.AmqpQueueConsumeCallbackArguments;
import eu.mosaic_cloud.cloudlets.connectors.queue.amqp.IAmqpQueueConsumerConnector;
import eu.mosaic_cloud.cloudlets.connectors.queue.amqp.IAmqpQueueConsumerConnectorFactory;
import eu.mosaic_cloud.cloudlets.connectors.queue.amqp.IAmqpQueuePublisherConnector;
import eu.mosaic_cloud.cloudlets.connectors.queue.amqp.IAmqpQueuePublisherConnectorFactory;
import eu.mosaic_cloud.cloudlets.core.CallbackArguments;
import eu.mosaic_cloud.cloudlets.core.CloudletCallbackArguments;
import eu.mosaic_cloud.cloudlets.core.CloudletCallbackCompletionArguments;
import eu.mosaic_cloud.cloudlets.core.GenericCallbackCompletionArguments;
import eu.mosaic_cloud.cloudlets.core.ICallback;
import eu.mosaic_cloud.cloudlets.core.ICloudletController;
import eu.mosaic_cloud.cloudlets.tools.DefaultAmqpPublisherConnectorCallback;
import eu.mosaic_cloud.cloudlets.tools.DefaultAmqpQueueConsumerConnectorCallback;
import eu.mosaic_cloud.cloudlets.tools.DefaultCloudletCallback;
import eu.mosaic_cloud.cloudlets.tools.DefaultKvStoreConnectorCallback;
import eu.mosaic_cloud.platform.core.configuration.ConfigUtils;
import eu.mosaic_cloud.platform.core.configuration.ConfigurationIdentifier;
import eu.mosaic_cloud.platform.core.configuration.IConfiguration;
import eu.mosaic_cloud.platform.core.utils.NullDataEncoder;
import eu.mosaic_cloud.platform.core.utils.PlainTextDataEncoder;
import eu.mosaic_cloud.platform.core.utils.SerializedDataEncoder;
import eu.mosaic_cloud.tools.callbacks.core.CallbackCompletion;
import eu.mosaic_cloud.platform.core.utils.JsonDataEncoder;
import java.util.Vector;
import java.io.*;
import com.google.gson.Gson;
public class DebugKv
{
public static final class AmqpConsumerCallback
extends DefaultAmqpQueueConsumerConnectorCallback<DebugKvContext, String, Void>
{
@Override
public CallbackCompletion<Void> consume (final DebugKvContext context, final AmqpQueueConsumeCallbackArguments<String> arguments)
{
final String data = arguments.getMessage ();
this.logger.info ("DebugKv received message for WGCloudlet", data);
this.logger.info("\n\nMessage Ricevuto:"+data);
this.logger.info("\n\n");
context.object=context.gson.fromJson(data,JsonObjectfromString.class);
context.NumIter=Integer.parseInt(context.object.getParameters(0));
context.Oper1=Integer.parseInt(context.object.getParameters(2));
context.Oper2=Integer.parseInt(context.object.getParameters(4));
context.xml=context.object.getParameters(5);
context.NumTot=((context.Oper1+context.Oper2)*context.NumIter);
if ((context.object.getParameters(1).equals("set")) && (context.object.getParameters(3).equals("get"))) {
for(int i=0;i<context.NumIter;i++)
{
for(int j=0;j<context.Oper1;j++)
{
//context.KeySet=Integer.toString((j+(context.Oper1*i)));
context.kvStore.set(context.KeySet, context.xml);
context.KeySet=Integer.toString((Integer.parseInt(context.KeySet))+1);
this.logger.info ("Fatto set 1");
}
for(int k=0;k<context.Oper2;k++)
{
//context.KeyGet=Integer.toString((k+(context.Oper2*i)));
context.kvStore.get(context.KeyGet);
context.KeyGet=Integer.toString((Integer.parseInt(context.KeyGet))+1);
this.logger.info ("Fatto get 1");
}
}
//context.consumer.acknowledge (arguments.getToken ());
} else {
for(int i=0;i<context.NumIter;i++)
{
for(int j=0;j<context.Oper2;j++)
{
//context.KeySet=Integer.toString((j+(context.Oper2*i)));
context.kvStore.set(context.KeySet, context.xml);
context.KeySet=Integer.toString((Integer.parseInt(context.KeySet))+1);
this.logger.info ("Fatto set 2");
}
for(int k=0;k<context.Oper1;k++)
{
//context.KeyGet=Integer.toString((k+(context.Oper1*i)));
context.kvStore.get(context.KeyGet);
context.KeyGet=Integer.toString((Integer.parseInt(context.KeyGet))+1);
this.logger.info ("Fatto get 2");
}
}
// context.consumer.acknowledge (arguments.getToken ());
}
// context.publisher.publish("Numero di Set e Get sono:"+context.KeySet+" "+context.KeyGet, null);
context.consumer.acknowledge (arguments.getToken ());
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> destroySucceeded (final DebugKvContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKv destroyed successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> initializeSucceeded (final DebugKvContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKv initialized successfully.");
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> acknowledgeSucceeded (final DebugKvContext context, final GenericCallbackCompletionArguments<Void> arguments)
{
//context.cloudlet.destroy ();
return ICallback.SUCCESS;
}
}
public static final class KeyValueCallback
extends DefaultKvStoreConnectorCallback<DebugKvContext, String, Void>
{
@Override
public CallbackCompletion<Void> destroySucceeded (final DebugKvContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKv store destroyed successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> initializeSucceeded (final DebugKvContext context, final CallbackArguments arguments)
{
this.logger.info ("KeyValueResult initialized successfully");
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> setSucceeded(final DebugKvContext context, final KvStoreCallbackCompletionArguments<String,Void> arguments) {
context.CountTot++;
this.logger.info("\n\nSet nel KV Store del DebugKv Succeeded\n\n");
String k=arguments.getKey();
this.logger.info("\n\nKEY:"+k);
this.logger.info("\n\n");
String value=(String)arguments.getValue();
this.logger.info("\n\nValue:"+value);
this.logger.info("\n\n");
if (context.CountTot==context.NumTot)
{
context.CountTot=0;
context.publisher.publish("Numero di Set e Get sono:"+context.KeySet+" "+context.KeyGet, null);
}
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> deleteSucceeded (final DebugKvContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKv store delete successfully.");
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> getSucceeded(final DebugKvContext context,final KvStoreCallbackCompletionArguments<String,Void> arguments) {
context.CountTot++;
this.logger.info("\n\nGet nel KV Store del DebugKv Succeeded\n\n");
String key =(String) arguments.getKey();
this.logger.info("\n\nKEY:"+key);
this.logger.info("\n\n");
String value1=(String) arguments.getValue();
this.logger.info("\n\nValue:"+value1);
this.logger.info("\n\n");
if (context.CountTot==context.NumTot)
{
context.CountTot=0;
context.publisher.publish("Numero di Set e Get sono:"+context.KeySet+" "+context.KeyGet, null);
}
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> ungetSucceeded (final DebugKvContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKv ungetSucceeded.");
return ICallback.SUCCESS;
}
}
public static final class AmqpPublisherCallback
extends DefaultAmqpPublisherConnectorCallback<DebugKvContext, String, Void>
{
@Override
public CallbackCompletion<Void> destroySucceeded (final DebugKvContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKvPublisher destroyed successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> initializeSucceeded (final DebugKvContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKvPublisher initialized successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> publishSucceeded (final DebugKvContext context, final GenericCallbackCompletionArguments<Void> arguments)
{
//context.cloudlet.destroy ();
this.logger.info ("DebugKvPublisher publisher successfully.");
return ICallback.SUCCESS;
}
}
public static final class LifeCycleHandler
extends DefaultCloudletCallback<DebugKvContext>
{
@Override
public CallbackCompletion<Void> destroy (final DebugKvContext context, final CloudletCallbackArguments<DebugKvContext> arguments)
{
this.logger.info ("DebugKv destroying...");
return CallbackCompletion.createAndChained (context.kvStore.destroy (), context.consumer.destroy ());
}
@Override
public CallbackCompletion<Void> destroySucceeded (final DebugKvContext context, final CloudletCallbackCompletionArguments<DebugKvContext> arguments)
{
this.logger.info ("DebugKv destroyed successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> initialize (final DebugKvContext context, final CloudletCallbackArguments<DebugKvContext> arguments)
{
this.logger.info ("DebugKv initializing...");
context.cloudlet = arguments.getCloudlet ();
final IConfiguration configuration = context.cloudlet.getConfiguration ();
final IConfiguration kvConfiguration = configuration.spliceConfiguration (ConfigurationIdentifier.resolveAbsolute ("resultstore"));
context.kvStore = context.cloudlet.getConnectorFactory (IKvStoreConnectorFactory.class).create (kvConfiguration, String.class, JsonDataEncoder.create (String.class), new KeyValueCallback (), context);
final IConfiguration queueConfiguration = configuration.spliceConfiguration (ConfigurationIdentifier.resolveAbsolute ("queueresult.consumer"));
context.consumer = context.cloudlet.getConnectorFactory (IAmqpQueueConsumerConnectorFactory.class).create (queueConfiguration, String.class, PlainTextDataEncoder.DEFAULT_INSTANCE, new AmqpConsumerCallback (), context);
final IConfiguration queueConfiguration2 = configuration.spliceConfiguration (ConfigurationIdentifier.resolveAbsolute ("queuefine.publisher"));
context.publisher = context.cloudlet.getConnectorFactory (IAmqpQueuePublisherConnectorFactory.class).create (queueConfiguration2, String.class, PlainTextDataEncoder.DEFAULT_INSTANCE, new AmqpPublisherCallback (), context);
return CallbackCompletion.createAndChained (context.kvStore.initialize (), context.consumer.initialize (),context.publisher.initialize());
}
@Override
public CallbackCompletion<Void> initializeSucceeded (final DebugKvContext context, final CloudletCallbackCompletionArguments<DebugKvContext> arguments)
{
this.logger.info ("DebugKv initialized successfully.");
context.object=new JsonObjectfromString();
context.gson=new Gson();
return ICallback.SUCCESS;
}
}
public static final class DebugKvContext
{
ICloudletController<DebugKvContext> cloudlet;
IAmqpQueueConsumerConnector<String, Void> consumer;
IKvStoreConnector<String, Void> kvStore;
IAmqpQueuePublisherConnector<String, Void> publisher;
String KeySet="0";
String KeyGet="0";
String xml="";
Gson gson;
int NumIter=0;
int Oper1=0;
int Oper2=0;
int CountTot=0;
int NumTot=0;
JsonObjectfromString object;
}
}