framework/rest/src/org/apache/cloudstack/framework/ws/jackson/UriSerializer.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.cloudstack.framework.ws.jackson;

import java.io.IOException;

import javax.ws.rs.core.UriBuilder;

import org.apache.cxf.jaxrs.impl.tl.ThreadLocalUriInfo;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;

public class UriSerializer extends JsonSerializer<String> {

    Url _annotation;

    public UriSerializer(Url annotation) {
        _annotation = annotation;
    }

    protected UriSerializer() {
    }

    @Override
    public void serialize(String id, JsonGenerator jgen, SerializerProvider provider) throws IOException, JsonProcessingException {
        jgen.writeStartObject();
        jgen.writeStringField("id", id);
        jgen.writeFieldName("uri");
        jgen.writeString(buildUri(_annotation.clazz(), _annotation.method(), id));
        jgen.writeEndObject();
    }

    protected String buildUri(Class<?> clazz, String method, String id) {
        ThreadLocalUriInfo uriInfo = new ThreadLocalUriInfo();
        UriBuilder ub = uriInfo.getAbsolutePathBuilder().path(clazz, method);
        ub.build(id);
        return ub.toString();
    }
}