com.qq.tars.server.core.AsyncContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tars-server Show documentation
There is a newer version: 1.7.2
/**
 * Tencent is pleased to support the open source community by making Tars available.
 *
 * Copyright (C) 2016 THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * 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.qq.tars.server.core;

import java.io.IOException;

import com.qq.tars.protocol.util.TarsHelper;
import com.qq.tars.rpc.protocol.tars.TarsServantRequest;
import com.qq.tars.rpc.protocol.tars.TarsServantResponse;
import com.qq.tars.server.apps.AppContextImpl;
import com.qq.tars.support.log.Logger;
import com.qq.tars.support.log.Logger.LogType;

public final class AsyncContext {

    private static final AppContainer container = ContainerManager.getContainer(AppContainer.class);

    public static final String PORTAL_CAP_ASYNC_CONTEXT_ATTRIBUTE = "internal.asynccontext";

    private Context context = null;
    private Logger flowLogger = Logger.getLogger("tarsserver.log", LogType.ALL);

    public static AsyncContext startAsync() throws IOException {
        Context context = ContextManager.getContext();
        AsyncContext aContext = new AsyncContext(context);
        context.response().asyncCallStart();
        context.setAttribute(PORTAL_CAP_ASYNC_CONTEXT_ATTRIBUTE, aContext);
        return aContext;
    }

    private AsyncContext(Context context) {
        this.context = context;
    }

    private ServantHomeSkeleton getCapHomeSkeleton() {
        AppContext appContext = container.getDefaultAppContext();
        return appContext.getCapHomeSkeleton(this.context.request().getServantName());
    }

    @SuppressWarnings("unchecked")
    public  T getAttribute(String name) {
        return (T) this.context.getAttribute(name);
    }

    public  T getAttribute(String name, T defaultValue) {
        return (T) this.context.getAttribute(name, defaultValue);
    }

    public  void setAttribute(String name, T value) {
        this.context.setAttribute(name, value);
    }

    public void writeException(Throwable ex) throws IOException {
        TarsServantResponse response = this.context.response();
        response.setRet(TarsHelper.SERVERUNKNOWNERR);
        response.setCause(ex);
        response.setResult(null);
        response.asyncCallEnd();

        getCapHomeSkeleton().postInvokeCapHomeSkeleton();
        Long startTime = this.context.getAttribute(Context.INTERNAL_START_TIME);
        TarsServantProcessor.printServiceFlowLog(flowLogger, this.context.request(), response.getRet(), (System.currentTimeMillis() - startTime.longValue()), ex.toString());

    }

    public void writeResult(Object result) throws IOException {
        TarsServantResponse response = this.context.response();
        response.setRet(TarsHelper.SERVERSUCCESS);
        response.setCause(null);
        response.setResult(result);
        response.asyncCallEnd();

        getCapHomeSkeleton().postInvokeCapHomeSkeleton();
        Long startTime = this.context.getAttribute(Context.INTERNAL_START_TIME);
        TarsServantProcessor.printServiceFlowLog(flowLogger, this.context.request(), response.getRet(), (System.currentTimeMillis() - startTime.longValue()), "");
    }
}