RestServiceImpl.java

/*
 * Copyright 2013-2014 High-Level Technologies
 *
 * 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 org.zodiark.service.rest;

import com.fasterxml.jackson.databind.ObjectMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.zodiark.server.Context;
import org.zodiark.server.Reply;
import org.zodiark.server.ReplyException;
import org.zodiark.service.util.ReflectionUtils;

import javax.inject.Inject;

import static org.zodiark.service.rest.RestService.METHOD.DELETE;
import static org.zodiark.service.rest.RestService.METHOD.GET;
import static org.zodiark.service.rest.RestService.METHOD.POST;
import static org.zodiark.service.rest.RestService.METHOD.PUT;

/**
 * Mock class for testing purpose only. This class does nothing.
 */
public class RestServiceImpl implements RestService {

    private final Logger logger = LoggerFactory.getLogger(RestServiceImpl.class);

    @Inject
    public Context context;

    @Inject
    public ObjectMapper mapper;

    @Inject
    public RestClient rest;

    @Override
    public void get(String uri, Reply r) {
        send(GET, uri, "", r);
    }

    @Override
    public void put(String uri, Object o, Reply r) {
        send(PUT, uri, o, r);
    }

    @Override
    public void post(String uri, Object o, Reply r) {
        send(POST, uri, o, r);
    }

    //@Override
    public void delete(String uri, Object o, Reply r) {
        send(DELETE, uri, o, r);
    }

    protected void send(METHOD method, String uri, Object o, Reply r) {
        try {
            Class<?> success = ReflectionUtils.getTypeArguments(Reply.class, r.getClass()).get(0);
            Class<?> failure = ReflectionUtils.getTypeArguments(Reply.class, r.getClass()).get(1);

            String restResponse = rest.serve(method, uri, o == null ? "" : o.toString());

            try {
                Object object = context.newInstance(success);
                r.ok(String.class.isAssignableFrom(success) ? restResponse : mapper.readerForUpdating(object).readValue(restResponse));
            } catch (Exception ex) {
                logger.error("", ex);
                Object object = context.newInstance(failure);
                ReplyException<Object> e = new ReplyException<>();
                e.error(String.class.isAssignableFrom(failure) ? restResponse : mapper.readerForUpdating(object).readValue(restResponse));
                r.fail(e);
            }
        } catch (Exception e) {
            logger.error("", e);
            r.fail(new ReplyException(e, null));
        }
    }
}