getAttachments();
44 |
45 | /**
46 | * set framework param
47 | *
48 | * @return
49 | */
50 | void setAttachment(String name, String value);
51 |
52 | /**
53 | * request id
54 | *
55 | * @return
56 | */
57 | long getRequestId();
58 |
59 | /**
60 | * retries
61 | *
62 | * @return
63 | */
64 | int getRetries();
65 |
66 | /**
67 | * set retries
68 | */
69 | void setRetries(int retries);
70 |
71 | /**
72 | * set the serialization number.
73 | * same to the protocol version, this value only used in server end for compatible.
74 | *
75 | * @param number
76 | */
77 | void setSerializationNumber(int number);
78 |
79 | int getSerializationNumber();
80 | }
81 |
--------------------------------------------------------------------------------
/jaws-core/src/main/java/org/hongxi/jaws/rpc/Response.java:
--------------------------------------------------------------------------------
1 | package org.hongxi.jaws.rpc;
2 |
3 | import java.util.Map;
4 |
5 | /**
6 | * Created by shenhongxi on 2020/6/14.
7 | */
8 | public interface Response {
9 |
10 | /**
11 | *
12 | * 如果 request 正常处理，那么会返回 Object value，而如果 request 处理有异常，那么 getValue 会抛出异常
13 | *
14 | *
15 | * @return
16 | * @throws RuntimeException
17 | */
18 | Object getValue();
19 |
20 | /**
21 | * 如果request处理有异常，那么调用该方法return exception 如果request还没处理完或者request处理正常，那么return null
22 | *
23 | *
24 | * 该方法不会阻塞，无论该request是处理中还是处理完成
25 | *
26 | *
27 | * @return
28 | */
29 | Exception getException();
30 |
31 | /**
32 | * 与 Request 的 requestId 相对应
33 | *
34 | * @return
35 | */
36 | long getRequestId();
37 |
38 | /**
39 | * 业务处理时间
40 | *
41 | * @return
42 | */
43 | long getProcessTime();
44 |
45 | /**
46 | * 业务处理时间
47 | *
48 | * @param time
49 | */
50 | void setProcessTime(long time);
51 |
52 | int getTimeout();
53 |
54 | Map getAttachments();
55 |
56 | void setAttachment(String key, String value);
57 |
58 | /**
59 | * set the serialization number.
60 | * same to the protocol version, this value only used in server end for compatible.
61 | *
62 | * @param number
63 | */
64 | void setSerializationNumber(int number);
65 |
66 | int getSerializationNumber();
67 | }
68 |
--------------------------------------------------------------------------------
/jaws-core/src/main/java/org/hongxi/jaws/rpc/ResponseFuture.java:
--------------------------------------------------------------------------------
1 | package org.hongxi.jaws.rpc;
2 |
3 | /**
4 | * Created by shenhongxi on 2020/7/30.
5 | */
6 | public interface ResponseFuture extends Future, Response {
7 |
8 | void onSuccess(Response response);
9 |
10 | void onFailure(Response response);
11 |
12 | long getCreateTime();
13 |
14 | void setReturnType(Class clazz);
15 | }
16 |
--------------------------------------------------------------------------------
/jaws-core/src/main/java/org/hongxi/jaws/rpc/RpcContext.java:
--------------------------------------------------------------------------------
1 | package org.hongxi.jaws.rpc;
2 |
3 | import org.hongxi.jaws.common.URLParamType;
4 |
5 | import java.util.HashMap;
6 | import java.util.Map;
7 |
8 | /**
9 | * Created by shenhongxi on 2020/6/26.
10 | */
11 | public class RpcContext {
12 | private static final ThreadLocal LOCAL_CONTEXT = ThreadLocal.withInitial(() -> new RpcContext());
13 | private Map attributes = new HashMap<>();
14 | private Map attachments = new HashMap<>();
15 | private Request request;
16 | private Response response;
17 | private String clientRequestId;
18 |
19 | public static RpcContext getContext() {
20 | return LOCAL_CONTEXT.get();
21 | }
22 |
23 | public static void destroy() {
24 | LOCAL_CONTEXT.remove();
25 | }
26 |
27 | /**
28 | * init new rpcContext with request
29 | *
30 | * @param request
31 | * @return
32 | */
33 | public static RpcContext init(Request request) {
34 | RpcContext context = new RpcContext();
35 | if (request != null) {
36 | context.setRequest(request);
37 | context.setClientRequestId(
38 | request.getAttachments().get(URLParamType.requestIdFromClient.getName()));
39 | }
40 | LOCAL_CONTEXT.set(context);
41 | return context;
42 | }
43 |
44 | public String getRequestId() {
45 | if (clientRequestId != null) return clientRequestId;
46 | if (request != null) return String.valueOf(request.getRequestId());
47 | return null;
48 | }
49 |
50 | public void putAttribute(Object key, Object value) {
51 | attributes.put(key, value);
52 | }
53 |
54 | public Object getAttribute(Object key) {
55 | return attributes.get(key);
56 | }
57 |
58 | public void removeAttribute(Object key) {
59 | attributes.remove(key);
60 | }
61 |
62 | public Map getAttributes() {
63 | return attributes;
64 | }
65 |
66 | public void setRpcAttachment(String key, String value) {
67 | attachments.put(key, value);
68 | }
69 |
70 | /**
71 | * get attachments from rpccontext only. not from request or response
72 | *
73 | * @param key
74 | * @return
75 | */
76 | public String getRpcAttachment(String key) {
77 | return attachments.get(key);
78 | }
79 |
80 | public void removeRpcAttachment(String key) {
81 | attachments.remove(key);
82 | }
83 |
84 | public Map getRpcAttachments() {
85 | return attachments;
86 | }
87 |
88 | public Request getRequest() {
89 | return request;
90 | }
91 |
92 | public void setRequest(Request request) {
93 | this.request = request;
94 | }
95 |
96 | public Response getResponse() {
97 | return response;
98 | }
99 |
100 | public void setResponse(Response response) {
101 | this.response = response;
102 | }
103 |
104 | public String getClientRequestId() {
105 | return clientRequestId;
106 | }
107 |
108 | public void setClientRequestId(String clientRequestId) {
109 | this.clientRequestId = clientRequestId;
110 | }
111 | }
112 |
--------------------------------------------------------------------------------
/jaws-core/src/main/java/org/hongxi/jaws/serialize/DeserializableObject.java:
--------------------------------------------------------------------------------
1 | package org.hongxi.jaws.serialize;
2 |
3 | import org.hongxi.jaws.codec.Serialization;
4 |
5 | import java.io.IOException;
6 |
7 | /**
8 | * Created by shenhongxi on 2020/7/28.
9 | */
10 | public class DeserializableObject {
11 | private Serialization serialization;
12 | private byte[] objBytes;
13 |
14 | public DeserializableObject(Serialization serialization, byte[] objBytes) {
15 | this.serialization = serialization;
16 | this.objBytes = objBytes;
17 | }
18 |
19 | public T deserialize(Class clz) throws IOException {
20 | return serialization.deserialize(objBytes, clz);
21 | }
22 |
23 | public Object[] deserializeMulti(Class[] paramTypes) throws IOException {
24 | Object[] ret = null;
25 | if (paramTypes != null && paramTypes.length > 0) {
26 | ret = serialization.deserializeMulti(objBytes, paramTypes);
27 | }
28 | return ret;
29 | }
30 | }
31 |
--------------------------------------------------------------------------------
/jaws-core/src/main/java/org/hongxi/jaws/serialize/FastJsonSerialization.java:
--------------------------------------------------------------------------------
1 | package org.hongxi.jaws.serialize;
2 |
3 | import com.alibaba.fastjson.JSON;
4 | import com.alibaba.fastjson.serializer.JSONSerializer;
5 | import com.alibaba.fastjson.serializer.SerializeWriter;
6 | import com.alibaba.fastjson.serializer.SerializerFeature;
7 | import org.hongxi.jaws.codec.Serialization;
8 | import org.hongxi.jaws.common.extension.SpiMeta;
9 |
10 | import java.io.IOException;
11 | import java.util.List;
12 |
13 | /**
14 | * fastjson 序列化
15 | *
16 | *
17 | * 对于嵌套场景无法支持
18 | *
19 | *
20 | * Created by shenhongxi on 2020/7/28.
21 | *
22 | */
23 | @SpiMeta(name = "fastjson")
24 | public class FastJsonSerialization implements Serialization {
25 |
26 | @Override
27 | public byte[] serialize(Object data) throws IOException {
28 | SerializeWriter out = new SerializeWriter();
29 | JSONSerializer serializer = new JSONSerializer(out);
30 | serializer.config(SerializerFeature.WriteEnumUsingToString, true);
31 | serializer.config(SerializerFeature.WriteClassName, true);
32 | serializer.write(data);
33 | return out.toBytes("UTF-8");
34 | }
35 |
36 | @Override
37 | public T deserialize(byte[] data, Class clz) throws IOException {
38 | return JSON.parseObject(new String(data), clz);
39 | }
40 |
41 | @Override
42 | public byte[] serializeMulti(Object[] data) throws IOException {
43 | return serialize(data);
44 | }
45 |
46 | @Override
47 | public Object[] deserializeMulti(byte[] data, Class[] classes) throws IOException {
48 | List