| /*
|
| * Copyright 1999-2011 Alibaba Group.
|
| *
|
| * 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 com.alibaba.dubbo.rpc.protocol; |
| |
| import java.lang.reflect.InvocationTargetException; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.Map; |
| |
| import com.alibaba.dubbo.common.URL; |
| import com.alibaba.dubbo.common.Version; |
| import com.alibaba.dubbo.common.logger.Logger; |
| import com.alibaba.dubbo.common.logger.LoggerFactory; |
| import com.alibaba.dubbo.common.utils.NetUtils; |
| import com.alibaba.dubbo.rpc.Invoker; |
| import com.alibaba.dubbo.rpc.RpcContext; |
| import com.alibaba.dubbo.rpc.RpcException; |
| import com.alibaba.dubbo.rpc.Invocation; |
| import com.alibaba.dubbo.rpc.Result; |
| import com.alibaba.dubbo.rpc.RpcInvocation;
|
| import com.alibaba.dubbo.rpc.RpcResult;
|
| |
| /** |
| * AbstractInvoker. |
| * |
| * @author qian.lei |
| * @author william.liangf |
| */ |
| public abstract class AbstractInvoker<T> implements Invoker<T> { |
| |
| protected final Logger logger = LoggerFactory.getLogger(getClass()); |
| |
| private final Class<T> type; |
| |
| private final URL url; |
| |
| private final Map<String, String> attachment; |
| |
| private volatile boolean available = true; |
| |
| private volatile boolean destroyed = false; |
| |
| public AbstractInvoker(Class<T> type, URL url){ |
| this(type, url, (Map<String, String>) null); |
| } |
| |
| public AbstractInvoker(Class<T> type, URL url, String[] keys) { |
| this(type, url, convertAttachment(url, keys)); |
| } |
| |
| public AbstractInvoker(Class<T> type, URL url, Map<String, String> attachment) { |
| if (type == null) |
| throw new IllegalArgumentException("service type == null"); |
| if (url == null) |
| throw new IllegalArgumentException("service url == null"); |
| this.type = type; |
| this.url = url; |
| this.attachment = attachment == null ? null : Collections.unmodifiableMap(attachment); |
| } |
| |
| private static Map<String, String> convertAttachment(URL url, String[] keys) { |
| if (keys == null || keys.length == 0) { |
| return null; |
| } |
| Map<String, String> attachment = new HashMap<String, String>(); |
| for (String key : keys) { |
| String value = url.getParameter(key); |
| if (value != null && value.length() > 0) { |
| attachment.put(key, value); |
| } |
| } |
| return attachment; |
| } |
| |
| public Class<T> getInterface() { |
| return type; |
| } |
| |
| public URL getUrl() { |
| return url; |
| } |
| |
| public boolean isAvailable() { |
| return available; |
| } |
| |
| protected void setAvailable(boolean available) { |
| this.available = available; |
| } |
| |
| public void destroy() { |
| if (destroyed) {
|
| return;
|
| } |
| destroyed = true; |
| setAvailable(false); |
| } |
| |
| public String toString() { |
| return getInterface() + " -> " + getUrl()==null?" ":getUrl().toString(); |
| } |
| |
| public Result invoke(Invocation inv) throws RpcException { |
| if(destroyed) { |
| throw new RpcException("Rpc invoker for service " + this + " on consumer " + NetUtils.getLocalHost() |
| + " use dubbo version " + Version.getVersion() |
| + " is DESTROYED, can not be invoked any more!");
|
| } |
| RpcInvocation invocation = (RpcInvocation) inv; |
| Map<String, String> attachments = new HashMap<String, String>(); |
| if (attachment != null && attachment.size() > 0) { |
| attachments.putAll(attachment); |
| } |
| Map<String, String> context = RpcContext.getContext().getAttachments(); |
| if (context != null) { |
| attachments.putAll(context); |
| } |
| if (invocation.getAttachments() != null) { |
| attachments.putAll(invocation.getAttachments()); |
| } |
| invocation.setAttachments(attachments); |
| try { |
| return doInvoke(invocation); |
| } catch (InvocationTargetException e) { // biz exception |
| Throwable te = e.getTargetException(); |
| if (te == null) { |
| return new RpcResult(e); |
| } else { |
| if (te instanceof RpcException) { |
| ((RpcException) te).setCode(RpcException.BIZ_EXCEPTION); |
| } |
| return new RpcResult(te); |
| } |
| } catch (RpcException e) { |
| if (e.isBiz()) { |
| return new RpcResult(e); |
| } else { |
| throw e; |
| } |
| } catch (Throwable e) { |
| return new RpcResult(e); |
| } |
| } |
| |
| protected abstract Result doInvoke(Invocation invocation) throws Throwable; |
| |
| } |