DeserializationBenchmark.java

/*
 * Copyright (c) 2015 Guillaume Hillairet.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Guillaume Hillairet - initial API and implementation
 *
 */
package org.emfjson.jackson.bench;

import com.fasterxml.jackson.databind.ObjectMapper;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.BinaryResourceImpl;
import org.eclipse.emf.ecore.resource.impl.ResourceFactoryImpl;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl;
import org.emfjson.jackson.junit.model.ModelPackage;
import org.emfjson.jackson.module.EMFModule;
import org.emfjson.jackson.resource.JsonResourceFactory;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;

public class DeserializationBenchmark {

	int times = 10;

	public static void main(String[] args) throws IOException {
		DeserializationBenchmark b = new DeserializationBenchmark();
		// first
		System.out.println("--- 1st benchmarck ---");
		b.benchmarkXmi(Benchmarks.first());
		b.benchmarkJson(Benchmarks.first());
//		b.benchmarkBinary(Benchmarks.first());
		// second
		System.out.println("--- 2nd benchmarck ---");
		b.benchmarkXmi(Benchmarks.second());
		b.benchmarkJson(Benchmarks.second());
//		b.benchmarkBinary(Benchmarks.second());
		// third
		System.out.println("--- 3rd benchmarck ---");
		b.benchmarkXmi(Benchmarks.third());
		b.benchmarkJson(Benchmarks.third());
//		b.benchmarkBinary(Benchmarks.third());
	}

	private String content(EObject content, ResourceSet rs) {
		Resource resource = rs.createResource(URI.createURI("save"));
		resource.getContents().add(content);

		ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
		try {
			resource.save(outputStream, null);
		} catch (IOException e) {
			e.printStackTrace();
		}

		return new String(outputStream.toByteArray());
	}

	private long load(Resource resource, String content) {
		long start = System.currentTimeMillis();
		try {
			resource.load(new ByteArrayInputStream(content.getBytes()), null);
		} catch (IOException e) {
			e.printStackTrace();
		}
		return System.currentTimeMillis() - start;
	}

	private ResourceSet xmi() {
		ResourceSet resourceSet = new ResourceSetImpl();
		resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put("*", new XMIResourceFactoryImpl());
		resourceSet.getPackageRegistry().put(ModelPackage.eNS_URI, ModelPackage.eINSTANCE);

		return resourceSet;
	}

	private ResourceSet json() {
		ObjectMapper mapper = new ObjectMapper();
		mapper.registerModule(new EMFModule());

		ResourceSet resourceSet = new ResourceSetImpl();
		resourceSet.getResourceFactoryRegistry()
				.getExtensionToFactoryMap()
				.put("*", new JsonResourceFactory(mapper));

		resourceSet.getPackageRegistry()
				.put(ModelPackage.eNS_URI, ModelPackage.eINSTANCE);

		return resourceSet;
	}

	public void benchmarkXmi(EObject content) throws IOException {
		long sum = 0;

		String payload = content(content, xmi());

		for (int i = 0; i < times; i++) {
			ResourceSet resourceSet = xmi();
			Resource resource = resourceSet.createResource(URI.createURI("test"));
			sum += load(resource, payload);
		}

		long average = sum / times;
		System.out.println("XMI: " + average / 1000.);
	}

	public void benchmarkBinary(EObject content) throws IOException {
		long sum = 0;

		for (int i = 0; i < times; i++) {
			ResourceSet resourceSet = new ResourceSetImpl();
			resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put("*", new ResourceFactoryImpl() {
				@Override
				public Resource createResource(URI uri) {
					return new BinaryResourceImpl(uri);
				}
			});
			resourceSet.getPackageRegistry().put(ModelPackage.eNS_URI, ModelPackage.eINSTANCE);
			Resource resource = resourceSet.createResource(URI.createURI("test"));

			sum += load(resource, content(content, resourceSet));
		}

		long average = sum / times;
		System.out.println("Binary: " + average / 1000.);
	}

	public void benchmarkJson(EObject content) throws IOException {
		long sum = 0;

		String payload = content(content, json());

		for (int i = 0; i < times; i++) {
			ResourceSet resourceSet = json();
			Resource resource = resourceSet.createResource(URI.createURI("test"));
			sum += load(resource, payload);
		}

		long average = sum / times;
		System.out.println("JSON: " + average / 1000.);
	}

}