/*
* Copyright 2014 TWO SIGMA OPEN SOURCE, LLC
*
* 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.
*/
package com.twosigma.beaker.jvm.module;
import com.google.inject.AbstractModule;
import com.google.inject.Injector;
import com.google.inject.Provides;
import com.google.inject.Singleton;
import com.twosigma.beaker.chart.serializer.AreaSerializer;
import com.twosigma.beaker.chart.serializer.BarsSerializer;
import com.twosigma.beaker.chart.serializer.ColorSerializer;
import com.twosigma.beaker.chart.serializer.CombinedPlotSerializer;
import com.twosigma.beaker.chart.serializer.CrosshairSerializer;
import com.twosigma.beaker.chart.serializer.LineSerializer;
import com.twosigma.beaker.chart.serializer.XYChartSerializer;
import com.twosigma.beaker.chart.serializer.PointsSerializer;
import com.twosigma.beaker.chart.serializer.StemsSerializer;
import com.twosigma.beaker.chart.serializer.YAxisSerializer;
import com.twosigma.beaker.chart.xychart.CombinedPlot;
import com.twosigma.beaker.chart.xychart.XYChart;
import com.twosigma.beaker.chart.xychart.plotitem.Area;
import com.twosigma.beaker.chart.xychart.plotitem.Bars;
import com.twosigma.beaker.chart.xychart.plotitem.Crosshair;
import com.twosigma.beaker.chart.xychart.plotitem.Line;
import com.twosigma.beaker.chart.xychart.plotitem.Points;
import com.twosigma.beaker.chart.xychart.plotitem.Stems;
import com.twosigma.beaker.chart.xychart.plotitem.YAxis;
import com.twosigma.beaker.shared.json.serializer.StringObject;
import com.twosigma.beaker.jvm.object.EvaluationResult;
import com.twosigma.beaker.jvm.object.OutputContainer;
import com.twosigma.beaker.jvm.object.SimpleEvaluationObject;
import com.twosigma.beaker.jvm.object.TableDisplay;
import com.twosigma.beaker.jvm.updater.ObservableUpdaterFactory;
import com.twosigma.beaker.jvm.updater.UpdateManager;
import com.twosigma.beaker.chart.Color;
import org.codehaus.jackson.Version;
import org.codehaus.jackson.jaxrs.JacksonJsonProvider;
import org.codehaus.jackson.map.ObjectMapper;
import org.codehaus.jackson.map.SerializationConfig;
import org.codehaus.jackson.map.module.SimpleModule;
import org.cometd.bayeux.server.BayeuxServer;
/**
* The Guice module as the registry of mapping from classes to serializers
*/
public class SerializerModule
extends AbstractModule {
@Override
protected void configure() {
bind(SimpleEvaluationObject.Serializer.class);
bind(EvaluationResult.Serializer.class);
bind(TableDisplay.Serializer.class);
bind(OutputContainer.Serializer.class);
bind(StringObject.Serializer.class);
}
@Provides
@Singleton
public UpdateManager getUpdateManager(Injector injector) {
BayeuxServer bayeuxServer = injector.getInstance(BayeuxServer.class);
UpdateManager updateManager = new UpdateManager(bayeuxServer);
updateManager.addUpdaterFactory(new ObservableUpdaterFactory());
return updateManager;
}
@Provides
@Singleton
public ObjectMapper getObjectMapper(Injector injector) {
ObjectMapper mapper = new ObjectMapper();
SimpleModule module =
new SimpleModule("MySerializer", new Version(1, 0, 0, null));
module.addSerializer(SimpleEvaluationObject.class, injector.getInstance(SimpleEvaluationObject.Serializer.class));
module.addSerializer(EvaluationResult.class, injector.getInstance(EvaluationResult.Serializer.class));
module.addSerializer(TableDisplay.class, injector.getInstance(TableDisplay.Serializer.class));
module.addSerializer(OutputContainer.class, injector.getInstance(OutputContainer.Serializer.class));
module.addSerializer(StringObject.class, injector.getInstance(StringObject.Serializer.class));
module.addSerializer(Color.class, injector.getInstance(ColorSerializer.class));
module.addSerializer(XYChart.class, injector.getInstance(XYChartSerializer.class));
module.addSerializer(CombinedPlot.class, injector.getInstance(CombinedPlotSerializer.class));
module.addSerializer(Line.class, injector.getInstance(LineSerializer.class));
module.addSerializer(Points.class, injector.getInstance(PointsSerializer.class));
module.addSerializer(Bars.class, injector.getInstance(BarsSerializer.class));
module.addSerializer(Stems.class, injector.getInstance(StemsSerializer.class));
module.addSerializer(Area.class, injector.getInstance(AreaSerializer.class));
module.addSerializer(YAxis.class, injector.getInstance(YAxisSerializer.class));
module.addSerializer(Crosshair.class, injector.getInstance(CrosshairSerializer.class));
mapper.registerModule(module);
SerializationConfig config = mapper.getSerializationConfig();
// Pretty
mapper.enable(SerializationConfig.Feature.INDENT_OUTPUT);
// Manually serialize everything, either through mixin or serializer
mapper.disable(SerializationConfig.Feature.AUTO_DETECT_GETTERS);
mapper.disable(SerializationConfig.Feature.AUTO_DETECT_IS_GETTERS);
mapper.disable(SerializationConfig.Feature.AUTO_DETECT_FIELDS);
return mapper;
}
@Provides
@Singleton
public JacksonJsonProvider getJackson(ObjectMapper mapper) {
return new JacksonJsonProvider(mapper);
}
}