//###
//# #%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;
//Esegue i Get nei Set.
public class DebugKvVariant
{
public static final class AmqpConsumerCallback
extends DefaultAmqpQueueConsumerConnectorCallback<DebugKvVariantContext, String, Void>
{
@Override
public CallbackCompletion<Void> consume (final DebugKvVariantContext context, final AmqpQueueConsumeCallbackArguments<String> arguments)
{
final String data = arguments.getMessage ();
this.logger.info ("DebugKvVariant 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);
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+(Oper1*i)));
context.kvStore.set(context.KeySet, context.xml);
context.KeySet=Integer.toString((Integer.parseInt(context.KeySet))+1);
}
}
} else {
for(int i=0;i<context.NumIter;i++)
{
for(int k=0;k<context.Oper1;k++)
{
//context.KeyGet=Integer.toString((k+(Oper1*i)));
context.kvStore.get(context.KeyGet);
context.KeyGet=Integer.toString((Integer.parseInt(context.KeyGet))+1);
}
}
}
context.consumer.acknowledge (arguments.getToken ());
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> destroySucceeded (final DebugKvVariantContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKvVariant destroyed successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> initializeSucceeded (final DebugKvVariantContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKvVariant initialized successfully.");
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> acknowledgeSucceeded (final DebugKvVariantContext context, final GenericCallbackCompletionArguments<Void> arguments)
{
//context.cloudlet.destroy ();
return ICallback.SUCCESS;
}
}
public static final class KeyValueCallback
extends DefaultKvStoreConnectorCallback<DebugKvVariantContext, String, Void>
{
@Override
public CallbackCompletion<Void> destroySucceeded (final DebugKvVariantContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKvVariant store destroyed successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> initializeSucceeded (final DebugKvVariantContext context, final CallbackArguments arguments)
{
this.logger.info ("KeyValueResult initialized successfully");
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> setSucceeded(final DebugKvVariantContext context, final KvStoreCallbackCompletionArguments<String,Void> arguments) {
this.logger.info("\n\nSet nel KV Store del DebugKvVariant 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");
context.countset++;
if (context.object.getParameters(1).equals("set"))
{
if(context.countset==context.Oper1)
{
this.logger.info("\n\nEntro nel Ciclo dei Get OR Push");
this.logger.info("\n\nCountSet:"+context.countset);
context.countset=0;
if (context.Oper2==0){
//push
context.publisher.publish("OKSet", null);
}
else{
for(int k1=0;k1<context.Oper2;k1++)
{
context.kvStore.get(context.KeyGet);
context.KeyGet=Integer.toString((Integer.parseInt(context.KeyGet))+1);
}
}
}
}else
{
if(context.countset==(context.Oper2*context.NumIter))
{
this.logger.info("\n\nEntro nel Ciclo del Push");
this.logger.info("\n\ncCountSet:"+context.countset);
context.countset=0;
context.publisher.publish("OKSet", null);
}
}
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> deleteSucceeded (final DebugKvVariantContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKvVariant store delete successfully.");
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> getSucceeded(final DebugKvVariantContext context,final KvStoreCallbackCompletionArguments<String,Void> arguments) {
this.logger.info("\n\nGet nel KV Store del DebugKvVariant 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");
context.countget++;
if (context.object.getParameters(3).equals("get")){
if(context.countget==(context.Oper2*context.NumIter))
{
this.logger.info("\n\nEntro nel Ciclo del Push");
this.logger.info("\n\nCountGet:"+context.countget);
context.countget=0;
context.publisher.publish("OKGet", null);
}
}else
{
if(context.countget==context.Oper1)
{
this.logger.info("\n\nEntro nel Ciclo dei Set OR Push");
this.logger.info("\n\nCountGet:"+context.countget);
context.countget=0;
if (context.Oper2==0){
//push
context.publisher.publish("OKGet", null);
}
else{
for(int k1=0;k1<context.Oper2;k1++)
{
context.kvStore.set(context.KeySet, context.xml);
context.KeySet=Integer.toString((Integer.parseInt(context.KeySet))+1);
}
}
}
}
return ICallback.SUCCESS;
}
public CallbackCompletion<Void> ungetSucceeded (final DebugKvVariantContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKvVariant ungetSucceeded.");
return ICallback.SUCCESS;
}
}
public static final class AmqpPublisherCallback
extends DefaultAmqpPublisherConnectorCallback<DebugKvVariantContext, String, Void>
{
@Override
public CallbackCompletion<Void> destroySucceeded (final DebugKvVariantContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKvVariantPublisher destroyed successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> initializeSucceeded (final DebugKvVariantContext context, final CallbackArguments arguments)
{
this.logger.info ("DebugKvVariantPublisher initialized successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> publishSucceeded (final DebugKvVariantContext context, final GenericCallbackCompletionArguments<Void> arguments)
{
//context.cloudlet.destroy ();
this.logger.info ("DebugKvVariantPublisher publisher successfully.");
return ICallback.SUCCESS;
}
}
public static final class LifeCycleHandler
extends DefaultCloudletCallback<DebugKvVariantContext>
{
@Override
public CallbackCompletion<Void> destroy (final DebugKvVariantContext context, final CloudletCallbackArguments<DebugKvVariantContext> arguments)
{
this.logger.info ("DebugKvVariant destroying...");
return CallbackCompletion.createAndChained (context.kvStore.destroy (), context.consumer.destroy ());
}
@Override
public CallbackCompletion<Void> destroySucceeded (final DebugKvVariantContext context, final CloudletCallbackCompletionArguments<DebugKvVariantContext> arguments)
{
this.logger.info ("DebugKvVariant destroyed successfully.");
return ICallback.SUCCESS;
}
@Override
public CallbackCompletion<Void> initialize (final DebugKvVariantContext context, final CloudletCallbackArguments<DebugKvVariantContext> arguments)
{
this.logger.info ("DebugKvVariant 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 DebugKvVariantContext context, final CloudletCallbackCompletionArguments<DebugKvVariantContext> arguments)
{
this.logger.info ("DebugKvVariant initialized successfully.");
context.object=new JsonObjectfromString();
context.gson=new Gson();
return ICallback.SUCCESS;
}
}
public static final class DebugKvVariantContext
{
ICloudletController<DebugKvVariantContext> cloudlet;
IAmqpQueueConsumerConnector<String, Void> consumer;
IKvStoreConnector<String, Void> kvStore;
IAmqpQueuePublisherConnector<String, Void> publisher;
String KeySet="0";
String KeyGet="0";
String xml="";
Gson gson;
JsonObjectfromString object;
int NumIter=0;
int Oper1=0;
int Oper2=0;
int countset=0;
int countget=0;
}
}