package at.gv.egovernment.moa.id.storage.test;

import java.io.IOException;

import javax.xml.parsers.ParserConfigurationException;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.support.FileSystemXmlApplicationContext;
import org.w3c.dom.Element;
import org.xml.sax.SAXException;

import at.gv.egiz.eaaf.core.api.storage.ITransactionStorage;
import at.gv.egiz.eaaf.core.exceptions.EAAFException;
import at.gv.egovernment.moa.id.storage.DBTransactionStorage;
import at.gv.egovernment.moa.util.Constants;
import at.gv.util.DOMUtils;

public class DBTransactionStorageTest {

	public static void main (String[] args) throws SAXException, IOException, ParserConfigurationException, EAAFException{
		DBTransactionStorageTest t = new DBTransactionStorageTest();
		t.test();
	}
	
	@Autowired
	DBTransactionStorage rts;
	
	public DBTransactionStorageTest(){

	}

	
	public void test() throws SAXException, IOException, ParserConfigurationException, EAAFException{


		ApplicationContext context = new FileSystemXmlApplicationContext("src/test/java/testBeans.xml");


		String requestString =
				"<?xml version=\"1.0\" encoding=\"UTF-8\" ?>" +
						"<samlp:Request xmlns:samlp=\"urn:oasis:names:tc:SAML:1.0:protocol\" xmlns:saml=\"urn:oasis:names:tc:SAML:1.0:assertion\" RequestID=\"123456\" MajorVersion=\"1\" MinorVersion=\"0\" IssueInstant=\"2003-02-13T13:59:00\">" +
						"<samlp:AssertionArtifact>WRONGARTIFACT</samlp:AssertionArtifact>" +
						"</samlp:Request>";
		Element request = DOMUtils.parseDocument(requestString, false, Constants.ALL_SCHEMA_LOCATIONS, null).getDocumentElement();

		ITransactionStorage rts = (ITransactionStorage) context.getBean("DBTransactionStorage");
		//GenericToStringSerializer redisStringSerializer = (GenericToStringSerializer) context.getBean("valueObjectSerializer");

		//				rts.getTemplate().setValueSerializer(new GenericToStringSerializer<Object>(Object.class));
		//				rts.getTemplate().setHashValueSerializer(new GenericToStringSerializer<Object>(Object.class));

		int cnt = 10;
		int averageCnt = 10;
		long putTime = 0, getTime = 0, changeTime = 0, removeTime = 0;
		long total = 0;
		for(int a=0;a<averageCnt;a++){
			long totalPerRound = 0;
			
			System.out.println("Starting MySql store operation.");
			long start = System.currentTimeMillis();
			for(int i=0; i<cnt;i++)
				rts.put("test"+i, request,-1);
			long end = System.currentTimeMillis();
			putTime += end-start;
			System.out.println("MySql store operation done in "+(end-start)+" ms.");
			totalPerRound+=(end-start);

			Element test;
			System.out.println("Starting MySql get operation.");
			start = System.currentTimeMillis();
			for(int i=0; i<cnt;i++)
				test = (Element)rts.get("test"+i);
			end = System.currentTimeMillis();
			getTime += end-start;
			System.out.println("MySql get operation done in "+(end-start)+" ms.");
			totalPerRound+=(end-start);
			//Element test = (Element)rts.get("test0");
			//System.out.println("Read Element from Redis Store: "+test.getTextContent());

			String requestString2 =
					"<?xml version=\"1.0\" encoding=\"UTF-8\" ?>" +
							"<samlp:Request xmlns:samlp=\"urn:oasis:names:tc:SAML:1.0:protocol\" xmlns:saml=\"urn:oasis:names:tc:SAML:1.0:assertion\" RequestID=\"test_new\" MajorVersion=\"1\" MinorVersion=\"0\" IssueInstant=\"2003-02-13T13:59:00\">" +
							"<samlp:AssertionArtifact>WRONGARTIFACT</samlp:AssertionArtifact>" +
							"</samlp:Request>";
			Element request2 = DOMUtils.parseDocument(requestString2, false, Constants.ALL_SCHEMA_LOCATIONS, null).getDocumentElement();

			System.out.println("Starting MySql change operation.");
			start = System.currentTimeMillis();
			for(int i=0; i<cnt;i++)
				rts.changeKey("test"+i, "test_new"+i, request2);
			end = System.currentTimeMillis();
			changeTime += end-start;
			System.out.println("MySql change operation done in "+(end-start)+" ms.");
			totalPerRound+=(end-start);
			
			
			
			System.out.println("Starting MySql remove operation.");
			start = System.currentTimeMillis();
			for(int i=0; i<cnt;i++)
				rts.remove("test_new"+i);
			end = System.currentTimeMillis();
			removeTime += end-start;
			System.out.println("MySql remove operation done in "+(end-start)+" ms.");
			totalPerRound+=(end-start);
			total+=totalPerRound;
			
			System.out.println("Redis Total Time in this round: "+totalPerRound+" ms.");
			System.out.println("______________________________________________________");
		}
		System.out.println("______________________________________________________");
		System.out.println("Redis average get time over " + averageCnt +" rounds: "+getTime/averageCnt+" ms.");
		System.out.println("Redis average put time over " + averageCnt +" rounds: "+putTime/averageCnt+" ms.");
		System.out.println("Redis average change time over " + averageCnt +" rounds: "+changeTime/averageCnt+" ms.");
		System.out.println("Redis average remove time over " + averageCnt +" rounds: "+removeTime/averageCnt+" ms.");
		System.out.println("Redis average total time over " + averageCnt +" rounds: "+total/averageCnt+" ms.");
		((ConfigurableApplicationContext)context).close();
	}

}