| /* |
| * 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.dubbo.rpc.filter; |
| |
| import org.apache.dubbo.common.extension.Activate; |
| import org.apache.dubbo.common.utils.StringUtils; |
| import org.apache.dubbo.rpc.Filter; |
| import org.apache.dubbo.rpc.Invocation; |
| import org.apache.dubbo.rpc.Invoker; |
| import org.apache.dubbo.rpc.Result; |
| import org.apache.dubbo.rpc.RpcContext; |
| import org.apache.dubbo.rpc.RpcException; |
| import org.apache.dubbo.rpc.RpcInvocation; |
| import org.apache.dubbo.rpc.TimeoutCountDown; |
| import org.apache.dubbo.rpc.support.RpcUtils; |
| |
| import java.util.HashMap; |
| import java.util.HashSet; |
| import java.util.Map; |
| import java.util.Set; |
| import java.util.concurrent.TimeUnit; |
| |
| import static org.apache.dubbo.common.constants.CommonConstants.DUBBO_VERSION_KEY; |
| import static org.apache.dubbo.common.constants.CommonConstants.GROUP_KEY; |
| import static org.apache.dubbo.common.constants.CommonConstants.INTERFACE_KEY; |
| import static org.apache.dubbo.common.constants.CommonConstants.PATH_KEY; |
| import static org.apache.dubbo.common.constants.CommonConstants.PROVIDER; |
| import static org.apache.dubbo.common.constants.CommonConstants.REMOTE_APPLICATION_KEY; |
| import static org.apache.dubbo.common.constants.CommonConstants.TAG_KEY; |
| import static org.apache.dubbo.common.constants.CommonConstants.TIMEOUT_ATTACHMENT_KEY; |
| import static org.apache.dubbo.common.constants.CommonConstants.TIMEOUT_KEY; |
| import static org.apache.dubbo.common.constants.CommonConstants.TIME_COUNTDOWN_KEY; |
| import static org.apache.dubbo.common.constants.CommonConstants.VERSION_KEY; |
| import static org.apache.dubbo.rpc.Constants.ASYNC_KEY; |
| import static org.apache.dubbo.rpc.Constants.FORCE_USE_TAG; |
| import static org.apache.dubbo.rpc.Constants.TOKEN_KEY; |
| |
| |
| /** |
| * ContextFilter set the provider RpcContext with invoker, invocation, local port it is using and host for |
| * current execution thread. |
| * |
| * @see RpcContext |
| */ |
| @Activate(group = PROVIDER, order = Integer.MIN_VALUE) |
| public class ContextFilter implements Filter, Filter.Listener { |
| |
| private static final Set<String> UNLOADING_KEYS; |
| |
| static { |
| UNLOADING_KEYS = new HashSet<>(16); |
| UNLOADING_KEYS.add(PATH_KEY); |
| UNLOADING_KEYS.add(INTERFACE_KEY); |
| UNLOADING_KEYS.add(GROUP_KEY); |
| UNLOADING_KEYS.add(VERSION_KEY); |
| UNLOADING_KEYS.add(DUBBO_VERSION_KEY); |
| UNLOADING_KEYS.add(TOKEN_KEY); |
| UNLOADING_KEYS.add(TIMEOUT_KEY); |
| UNLOADING_KEYS.add(TIMEOUT_ATTACHMENT_KEY); |
| |
| // Remove async property to avoid being passed to the following invoke chain. |
| UNLOADING_KEYS.add(ASYNC_KEY); |
| UNLOADING_KEYS.add(TAG_KEY); |
| UNLOADING_KEYS.add(FORCE_USE_TAG); |
| } |
| |
| @Override |
| public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException { |
| Map<String, Object> attachments = invocation.getObjectAttachments(); |
| if (attachments != null) { |
| Map<String, Object> newAttach = new HashMap<>(attachments.size()); |
| for (Map.Entry<String, Object> entry : attachments.entrySet()) { |
| String key = entry.getKey(); |
| if (!UNLOADING_KEYS.contains(key)) { |
| newAttach.put(key, entry.getValue()); |
| } |
| } |
| attachments = newAttach; |
| } |
| |
| RpcContext.getServiceContext().setInvoker(invoker) |
| .setInvocation(invocation); |
| |
| RpcContext context = RpcContext.getServerAttachment(); |
| // .setAttachments(attachments) // merged from dubbox |
| context.setLocalAddress(invoker.getUrl().getHost(), invoker.getUrl().getPort()); |
| |
| String remoteApplication = invocation.getAttachment(REMOTE_APPLICATION_KEY); |
| if (StringUtils.isNotEmpty(remoteApplication)) { |
| RpcContext.getServiceContext().setRemoteApplicationName(remoteApplication); |
| } else { |
| RpcContext.getServiceContext().setRemoteApplicationName(context.getAttachment(REMOTE_APPLICATION_KEY)); |
| } |
| |
| long timeout = RpcUtils.getTimeout(invocation, -1); |
| if (timeout != -1) { |
| // pass to next hop |
| RpcContext.getClientAttachment().setObjectAttachment(TIME_COUNTDOWN_KEY, TimeoutCountDown.newCountDown(timeout, TimeUnit.MILLISECONDS)); |
| } |
| |
| // merged from dubbox |
| // we may already add some attachments into RpcContext before this filter (e.g. in rest protocol) |
| if (attachments != null) { |
| if (context.getObjectAttachments().size() > 0) { |
| context.getObjectAttachments().putAll(attachments); |
| } else { |
| context.setObjectAttachments(attachments); |
| } |
| } |
| |
| if (invocation instanceof RpcInvocation) { |
| ((RpcInvocation) invocation).setInvoker(invoker); |
| } |
| |
| try { |
| context.clearAfterEachInvoke(false); |
| return invoker.invoke(invocation); |
| } finally { |
| context.clearAfterEachInvoke(true); |
| } |
| } |
| |
| @Override |
| public void onResponse(Result appResponse, Invoker<?> invoker, Invocation invocation) { |
| // pass attachments to result |
| appResponse.addObjectAttachments(RpcContext.getServerContext().getObjectAttachments()); |
| removeContext(); |
| } |
| |
| @Override |
| public void onError(Throwable t, Invoker<?> invoker, Invocation invocation) { |
| removeContext(); |
| } |
| |
| private void removeContext() { |
| RpcContext.removeServerAttachment(); |
| RpcContext.removeClientAttachment(); |
| RpcContext.removeServiceContext(); |
| RpcContext.removeServerContext(); |
| } |
| } |