一個簡單的rpc框架的實現
阿新 • • 發佈:2019-02-19
為了降低開發成本,提升已有系統的利用率,企業往往會構建自己的SOA體系結構,SOA構建的手段有多種,可以通過webservice,restful,rmi,socket等等,筆者通過java socket,來構建一個簡單的SOA體系結構,以下程式碼僅供參考。
1. 需要實現SOA的服務介面
Java程式碼- package com.chenkangxian.rpc.impl;
- /**
- *
- * @Author: chenkangxian
- *
- * @Annotation: 根據key取資料介面
- *
- * @Date:2012-5-13
- *
- * @Copyright: 2012 chenkangxian, All rights reserved.
- *
- */
- public interface DataService {
- String getData(String key);
- }
2. 介面的實現
Java程式碼- package com.chenkangxian.rpc.impl;
- /**
- *
- * @Author: chenkangxian
- *
- * @Annotation: 根據key取資料服務實現
- *
- * @Date:2012-5-13
- *
- * @Copyright: 2012 chenkangxian, All rights reserved.
- *
- */
- public class
- public String getData(String key) {
- return "this is the data when key = " + key ;
- }
- }
3. 執行代理
Java程式碼- /**
- *
- */
- package com.chenkangxian.rpc.impl;
- import java.io.ObjectInputStream;
- import java.io.ObjectOutputStream;
- import
- import java.lang.reflect.Method;
- import java.net.Socket;
- /**
- * @Author: chenkangxian
- *
- * @Annotation: 執行代理
- *
- * @Date:2012-5-15
- *
- * @Copyright: 2012 chenkangxian, All rights reserved.
- *
- */
- public class InvocationProxy implements InvocationHandler{
- private String host;
- private int port;
- public InvocationProxy(String host, int port){
- this.host = host;
- this.port = port;
- }
- public Object invoke(Object proxy, Method method, Object[] arguments) throws Throwable {
- Socket socket = new Socket(host, port);
- try {
- ObjectOutputStream output = new ObjectOutputStream(socket.getOutputStream());
- try {
- output.writeUTF(method.getName());
- output.writeObject(method.getParameterTypes());
- output.writeObject(arguments);
- ObjectInputStream input = new ObjectInputStream(socket.getInputStream());
- try {
- Object result = input.readObject();
- if (result instanceof Throwable) {
- throw (Throwable) result;
- }
- return result;
- } finally {
- input.close();
- }
- } finally {
- output.close();
- }
- } finally {
- socket.close();
- }
- }
- }
4. 服務消費者
Java程式碼- package com.chenkangxian.rpc.impl;
- /**
- * @Author: chenkangxian
- *
- * @Annotation: 服務消費者
- *
- * @Date:2012-5-13
- *
- * @Copyright: 2012 chenkangxian, All rights reserved.
- *
- */
- public class RpcConsumer {
- public static void main(String[] args) throws Exception {
- DataService service = RpcFramework.refer(DataService.class, "127.0.0.1", 1234);
- for (int i = 0; i < Integer.MAX_VALUE; i ++) {
- String value = service.getData("key_" + i);
- System.out.println(value);
- Thread.sleep(1000);
- }
- }
- }
5. 遠端呼叫框架
Java程式碼- package com.chenkangxian.rpc.impl;
- import java.lang.reflect.Proxy;
- import java.net.ServerSocket;
- import java.net.Socket;
- /**
- *
- * @Author: chenkangxian
- *
- * @Annotation: 簡單的遠端呼叫框架實現
- *
- * @Date:2012-5-13
- *
- * @Copyright: 2012 chenkangxian, All rights reserved.
- *
- */
- public class RpcFramework {
- /**
- * 暴露服務
- *
- * Author: chenkangxian
- *
- * Last Modification Time: 2012-5-15
- *
- * @param service 服務實現
- * @param port 服務埠
- * @throws Exception
- */
- public static void export(final Object service, int port) throws Exception {
- if (service == null)
- throw new IllegalArgumentException("service instance == null");
- if (port <= 0 || port > 65535)
- throw new IllegalArgumentException("Invalid port " + port);
- System.out.println("Export service " + service.getClass().getName() + " on port " + port);
- ServerSocket server = new ServerSocket(port);
- for(;;) {
- try {
- final Socket socket = server.accept();
- ThreadPoolHelp.getExecutorInstance().execute(new WorkThread(service, socket));
- } catch (Exception e) {
- e.printStackTrace();
- }
- }
- }
- /**
- * 引用服務
- *
- * Author: chenkangxian
- *
- * Last Modification Time: 2012-5-15
- *
- * @param <T> 介面泛型
- * @param interfaceClass 介面型別
- * @param host 伺服器主機名
- * @param port 伺服器埠
- * @return 遠端服務
- * @throws Exception
- */
- @SuppressWarnings("unchecked")
- public static <T> T refer(final Class<T> interfaceClass, final String host, final int port) throws Exception {
- if (interfaceClass == null)
- throw new IllegalArgumentException("Interface class == null");
- if (! interfaceClass.isInterface())
- throw new IllegalArgumentException("The " + interfaceClass.getName() + " must be interface class!");
- if (host == null || host.length() == 0)
- throw new IllegalArgumentException("Host == null!");
- if (port <= 0 || port > 65535)
- throw new IllegalArgumentException("Invalid port " + port);
- System.out.println("Get remote service " + interfaceClass.getName() + " from server " + host + ":" + port);
- return (T) Proxy.newProxyInstance(interfaceClass.getClassLoader(), new Class<?>[] {interfaceClass}, new InvocationProxy(host,port));
- }
- }
6. 服務提供者
Java程式碼- package com.chenkangxian.rpc.impl;
- /**
- * @Author: chenkangxian
- *
- * @Annotation: 服務提供者
- *
- * @Date:2012-5-13
- *
- * @Copyright: 2012 chenkangxian, All rights reserved.
- *
- */
- public class RpcProvider {
- public static void main(String[] args) throws Exception {
- DataService service = new DataServiceImpl();
- RpcFramework.export(service, 1234);
- }
- }
7. 執行緒池幫助類
Java程式碼- /**
- *
- */
- package com.chenkangxian.rpc.impl;
- import java.util.concurrent.ExecutorService;
- import java.util.concurrent.Executors;
- /**
- * @Author: chenkangxian
- *
- * @Annotation: 執行緒池幫助類
- *
- * @Date:2012-5-15
- *
- * @Copyright: 2012 chenkangxian, All rights reserved.
- *
- */
- public class ThreadPoolHelp {
- private static ExecutorService executor ;
- static{
- executor = Executors.newFixedThreadPool(20);
- }
- public static ExecutorService getExecutorInstance(){
- return executor;
- }
- }
8. 工作執行緒
Java程式碼- /**
- *
- */
- package com.chenkangxian.rpc.impl;
- import java.io.ObjectInputStream;
- import java.io.ObjectOutputStream;
- import java.lang.reflect.Method;
- import java.net.Socket;
- /**
- *
- * @Author: chenkangxian
- *
- * @Annotation: 工作執行緒
- *
- * @Date:2012-5-15
- *
- * @Copyright: 2012 chenkangxian, All rights reserved.
- *
- */
- public class WorkThread implements Runnable {
- private Object service;
- private Socket socket;
- public WorkThread(Object service,Socket socket){
- this.service = service;
- this.socket = socket;
- }
- @Override
- public void run() {
- try {
- try {
- ObjectInputStream input = new ObjectInputStream(socket.getInputStream());
- try {
- String methodName = input.readUTF();
- Class<?>[] parameterTypes = (Class<?>[])input.readObject();
- Object[] arguments = (Object[])input.readObject();
- ObjectOutputStream output = new ObjectOutputStream(socket.getOutputStream());
- try {
- Method method = service.getClass().getMethod(methodName, parameterTypes);
- Object result = method.invoke(service, arguments);
- output.writeObject(result);
- } catch (Throwable t) {
- output.writeObject(t);
- } finally {
- output.close();
- }
- } finally {
- input.close();
- }
- } finally {
- socket.close();
- }
- } catch (Exception e) {
- e.printStackTrace();
- }
- }
- }