jetbrick.web.mvc.ResultHandlerResolver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jetbrick-webmvc Show documentation
jetbrick-webmvc
The newest version!
/**
 * Copyright 2013-2014 Guoqiang Chen, Shanghai, China. All rights reserved.
 *
 *   Author: Guoqiang Chen
 *    Email: [email protected]
 *   WebURL: https://github.com/subchen
 *
 * 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 jetbrick.web.mvc;

import java.util.IdentityHashMap;
import java.util.Map;
import javax.xml.bind.JAXBElement;
import jetbrick.ioc.Ioc;
import jetbrick.util.Validate;
import jetbrick.web.mvc.result.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Document;

/**
 * 全局 ResultHandler 管理器
 */
public final class ResultHandlerResolver {
    private final Logger log = LoggerFactory.getLogger(ResultHandlerResolver.class);
    private final Map, ResultHandler> mapping = new IdentityHashMap, ResultHandler>();

    public void initialize() {
        register(Void.TYPE, VoidResultHandler.class);
        register(Object.class, ObjectResultHandler.class);
        register(String.class, StringResultHandler.class);
        register(HttpStatus.class, HttpStatusResultHandler.class);
        register(RawText.class, RawTextResultHandler.class);
        register(RawData.class, RawDataResultHandler.class);
        register(RawDownload.class, RawDownloadResultHandler.class);
        register(Document.class, XmlDocumentResultHandler.class);
        register(JAXBElement.class, JAXBElementResultHandler.class);
    }

    public void register(Class resultClass, Class resultHandlerClass) {
        Validate.isAssignableFrom(ResultHandler.class, resultHandlerClass);

        log.debug("register ResultHandler: {} -> {}", resultClass.getName(), resultHandlerClass.getName());

        Ioc ioc = WebConfig.getIoc();
        ResultHandler resultHandler = (ResultHandler) ioc.newInstance(resultHandlerClass);
        ioc.injectSetters(resultHandler);
        ioc.initialize(resultHandler);
        mapping.put(resultClass, resultHandler);
    }

    @SuppressWarnings("unchecked")
    public ResultHandler