/*
 * Copyright 2013 - Elian ORIOU
 *
 * 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.
 *
 * 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.vino.backend.serialization;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.vino.backend.persistence.Persistor;
import com.vino.backend.persistence.mongo.MongoPersistor;
import com.vino.backend.reference.Reference;
import restx.factory.Factory;
import restx.factory.Name;
import restx.jackson.Views;

import java.io.IOException;

/**
 * User: walien
 * Date: 1/11/14
 * Time: 6:14 PM
 */
public class ReferenceSerializer extends JsonSerializer<Reference> {

    private static final Factory FACTORY = Factory.builder().addFromServiceLoader().build();
    private static final Persistor PERSISTOR = FACTORY.getComponent(Name.of(MongoPersistor.class));

    @Override
    public void serialize(Reference reference, JsonGenerator jsonGenerator, SerializerProvider serializerProvider) throws IOException, JsonProcessingException {
        Class<?> activeView = serializerProvider.getConfig().getActiveView();

        // If the reference is serialized to mongodb
        if (activeView == Views.Private.class) {
            jsonGenerator.writeString(reference.getKey());
        }
        // If the reference is serialized to client
        else {
            jsonGenerator.writeObject(reference.get(PERSISTOR));
        }
    }
}

@@ -1945,33 +1945,48 @@
     
-jsonGenerator.writeObject
+serializerProvider.defaultSerializeValue
 (ref
@@ -2006,16 +2006,31 @@
 RSISTOR)
+, jsonGenerator
 );%0A     

Fix reference serialization issue