All Downloads are FREE. Search and download functionalities are using the official Maven repository.

net.hasor.web.invoker.InvokerContext Maven / Gradle / Ivy

There is a newer version: 4.2.5
Show newest version
/*
 * Copyright 2008-2009 the original author or authors.
 *
 * 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 net.hasor.web.invoker;
import net.hasor.core.AppContext;
import net.hasor.core.spi.SpiTrigger;
import net.hasor.utils.future.BasicFuture;
import net.hasor.web.Invoker;
import net.hasor.web.InvokerFilter;
import net.hasor.web.Mapping;
import net.hasor.web.binder.FilterDef;
import net.hasor.web.binder.MappingDef;
import net.hasor.web.binder.OneConfig;
import net.hasor.web.spi.MappingDiscoverer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Comparator;
import java.util.List;
import java.util.Objects;

/**
 * 上下文。
 * @version : 2017-01-10
 * @author 赵永春 ([email protected])
 */
public class InvokerContext {
    protected static Logger             logger         = LoggerFactory.getLogger(InvokerContext.class);
    private          AppContext         appContext     = null;
    private          Mapping[]          invokeArray    = new Mapping[0];
    private          FilterDef[]        filters        = new FilterDef[0];
    private          RootInvokerCreater invokerCreator = null;

    public void initContext(final AppContext appContext, final OneConfig configMap) throws Throwable {
        this.appContext = Objects.requireNonNull(appContext);
        //
        // .MappingData
        List mappingList = appContext.findBindingBean(MappingDef.class);
        mappingList.sort(Comparator.comparingLong(MappingDef::getIndex));
        this.invokeArray = mappingList.toArray(new Mapping[0]);
        for (Mapping inMapping : this.invokeArray) {
            logger.info("mapingTo -> type '{}' mappingTo: '{}'.", inMapping.getTargetType().getBindType(), inMapping.getMappingTo());
        }
        //
        // .discover
        SpiTrigger spiTrigger = appContext.getInstance(SpiTrigger.class);
        for (Mapping mapping : invokeArray) {
            spiTrigger.notifySpiWithoutResult(MappingDiscoverer.class, listener -> {
                listener.discover(mapping);
            });
        }
        //
        // .Filters
        this.filters = appContext.findBindingBean(FilterDef.class).stream()//
                .sorted(Comparator.comparingLong(FilterDef::getIndex))     //
                .toArray(FilterDef[]::new);                                //
        // .init
        for (FilterDef filter : this.filters) {
            filter.init(configMap);
        }
        // .creator
        this.invokerCreator = new RootInvokerCreater(appContext);
    }

    public void destroyContext() {
        for (InvokerFilter filter : this.filters) {
            filter.destroy();
        }
    }

    public Invoker newInvoker(Mapping define, HttpServletRequest request, HttpServletResponse response) {
        return this.invokerCreator.createExt(new InvokerSupplier(define, this.appContext, request, response));
    }

    public ExecuteCaller genCaller(HttpServletRequest httpReq, HttpServletResponse httpRes) {
        Mapping foundDefine = null;
        for (Mapping define : this.invokeArray) {
            if (define.matchingMapping(httpReq)) {
                foundDefine = define;
                break;
            }
        }
        //
        Invoker invoker = this.newInvoker(foundDefine, httpReq, httpRes);
        ExecuteCaller executeCaller = null;
        if (foundDefine == null) {
            executeCaller = (chain) -> {
                BasicFuture future = new BasicFuture<>();
                future.completed(new InvokerChainInvocation(filters, innerInv -> {
                    if (chain != null) {
                        chain.doFilter(innerInv.getHttpRequest(), innerInv.getHttpResponse());
                    }
                    return innerInv.get(Invoker.RETURN_DATA_KEY);
                }).doNext(invoker));
                return future;
            };
        } else {
            executeCaller = new InvokerCaller(() -> invoker, this.filters);
        }
        //
        ExecuteCaller finalExecuteCaller = executeCaller;
        return chain -> HttpParameters.executeWorker(invoker, () -> {
            return finalExecuteCaller.invoke(chain);
        });
    }
}