├── .gitignore
├── LICENSE
├── README.md
├── pom.xml
└── src
└── main
├── docker
└── Dockerfile
├── java
└── com
│ └── stalary
│ └── lightmq
│ ├── CtrlTimerAdvice.java
│ ├── FacadeController.java
│ ├── JsonResponse.java
│ ├── LightmqApplication.java
│ ├── OperatorService.java
│ ├── ResponseCodeEnum.java
│ ├── TomcatListener.java
│ ├── data
│ ├── Constant.java
│ ├── MessageDto.java
│ ├── MessageFactory.java
│ ├── MessageGroup.java
│ └── SaveData.java
│ └── exception
│ ├── ExceptionEnum.java
│ ├── GlobalExceptionHandler.java
│ └── MyException.java
└── resources
├── application.yml
└── logback-spring.xml
/.gitignore:
--------------------------------------------------------------------------------
1 | # Compiled class file
2 | *.class
3 | .idea/
4 | .mvn/
5 | *.iml
6 |
7 | # Log file
8 | *.log
9 |
10 |
11 | # Package Files #
12 | *.war
13 | *.ear
14 | *.zip
15 | *.tar.gz
16 | *.rar
17 | build
18 | src/target
19 | target
20 |
21 | logs
22 | mvnw
23 | mvnw.cmd
24 | save.json
25 | upload.sh
26 |
--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
1 | Apache License
2 | Version 2.0, January 2004
3 | http://www.apache.org/licenses/
4 |
5 | TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
6 |
7 | 1. Definitions.
8 |
9 | "License" shall mean the terms and conditions for use, reproduction,
10 | and distribution as defined by Sections 1 through 9 of this document.
11 |
12 | "Licensor" shall mean the copyright owner or entity authorized by
13 | the copyright owner that is granting the License.
14 |
15 | "Legal Entity" shall mean the union of the acting entity and all
16 | other entities that control, are controlled by, or are under common
17 | control with that entity. For the purposes of this definition,
18 | "control" means (i) the power, direct or indirect, to cause the
19 | direction or management of such entity, whether by contract or
20 | otherwise, or (ii) ownership of fifty percent (50%) or more of the
21 | outstanding shares, or (iii) beneficial ownership of such entity.
22 |
23 | "You" (or "Your") shall mean an individual or Legal Entity
24 | exercising permissions granted by this License.
25 |
26 | "Source" form shall mean the preferred form for making modifications,
27 | including but not limited to software source code, documentation
28 | source, and configuration files.
29 |
30 | "Object" form shall mean any form resulting from mechanical
31 | transformation or translation of a Source form, including but
32 | not limited to compiled object code, generated documentation,
33 | and conversions to other media types.
34 |
35 | "Work" shall mean the work of authorship, whether in Source or
36 | Object form, made available under the License, as indicated by a
37 | copyright notice that is included in or attached to the work
38 | (an example is provided in the Appendix below).
39 |
40 | "Derivative Works" shall mean any work, whether in Source or Object
41 | form, that is based on (or derived from) the Work and for which the
42 | editorial revisions, annotations, elaborations, or other modifications
43 | represent, as a whole, an original work of authorship. For the purposes
44 | of this License, Derivative Works shall not include works that remain
45 | separable from, or merely link (or bind by name) to the interfaces of,
46 | the Work and Derivative Works thereof.
47 |
48 | "Contribution" shall mean any work of authorship, including
49 | the original version of the Work and any modifications or additions
50 | to that Work or Derivative Works thereof, that is intentionally
51 | submitted to Licensor for inclusion in the Work by the copyright owner
52 | or by an individual or Legal Entity authorized to submit on behalf of
53 | the copyright owner. For the purposes of this definition, "submitted"
54 | means any form of electronic, verbal, or written communication sent
55 | to the Licensor or its representatives, including but not limited to
56 | communication on electronic mailing lists, source code control systems,
57 | and issue tracking systems that are managed by, or on behalf of, the
58 | Licensor for the purpose of discussing and improving the Work, but
59 | excluding communication that is conspicuously marked or otherwise
60 | designated in writing by the copyright owner as "Not a Contribution."
61 |
62 | "Contributor" shall mean Licensor and any individual or Legal Entity
63 | on behalf of whom a Contribution has been received by Licensor and
64 | subsequently incorporated within the Work.
65 |
66 | 2. Grant of Copyright License. Subject to the terms and conditions of
67 | this License, each Contributor hereby grants to You a perpetual,
68 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable
69 | copyright license to reproduce, prepare Derivative Works of,
70 | publicly display, publicly perform, sublicense, and distribute the
71 | Work and such Derivative Works in Source or Object form.
72 |
73 | 3. Grant of Patent License. Subject to the terms and conditions of
74 | this License, each Contributor hereby grants to You a perpetual,
75 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable
76 | (except as stated in this section) patent license to make, have made,
77 | use, offer to sell, sell, import, and otherwise transfer the Work,
78 | where such license applies only to those patent claims licensable
79 | by such Contributor that are necessarily infringed by their
80 | Contribution(s) alone or by combination of their Contribution(s)
81 | with the Work to which such Contribution(s) was submitted. If You
82 | institute patent litigation against any entity (including a
83 | cross-claim or counterclaim in a lawsuit) alleging that the Work
84 | or a Contribution incorporated within the Work constitutes direct
85 | or contributory patent infringement, then any patent licenses
86 | granted to You under this License for that Work shall terminate
87 | as of the date such litigation is filed.
88 |
89 | 4. Redistribution. You may reproduce and distribute copies of the
90 | Work or Derivative Works thereof in any medium, with or without
91 | modifications, and in Source or Object form, provided that You
92 | meet the following conditions:
93 |
94 | (a) You must give any other recipients of the Work or
95 | Derivative Works a copy of this License; and
96 |
97 | (b) You must cause any modified files to carry prominent notices
98 | stating that You changed the files; and
99 |
100 | (c) You must retain, in the Source form of any Derivative Works
101 | that You distribute, all copyright, patent, trademark, and
102 | attribution notices from the Source form of the Work,
103 | excluding those notices that do not pertain to any part of
104 | the Derivative Works; and
105 |
106 | (d) If the Work includes a "NOTICE" text file as part of its
107 | distribution, then any Derivative Works that You distribute must
108 | include a readable copy of the attribution notices contained
109 | within such NOTICE file, excluding those notices that do not
110 | pertain to any part of the Derivative Works, in at least one
111 | of the following places: within a NOTICE text file distributed
112 | as part of the Derivative Works; within the Source form or
113 | documentation, if provided along with the Derivative Works; or,
114 | within a display generated by the Derivative Works, if and
115 | wherever such third-party notices normally appear. The contents
116 | of the NOTICE file are for informational purposes only and
117 | do not modify the License. You may add Your own attribution
118 | notices within Derivative Works that You distribute, alongside
119 | or as an addendum to the NOTICE text from the Work, provided
120 | that such additional attribution notices cannot be construed
121 | as modifying the License.
122 |
123 | You may add Your own copyright statement to Your modifications and
124 | may provide additional or different license terms and conditions
125 | for use, reproduction, or distribution of Your modifications, or
126 | for any such Derivative Works as a whole, provided Your use,
127 | reproduction, and distribution of the Work otherwise complies with
128 | the conditions stated in this License.
129 |
130 | 5. Submission of Contributions. Unless You explicitly state otherwise,
131 | any Contribution intentionally submitted for inclusion in the Work
132 | by You to the Licensor shall be under the terms and conditions of
133 | this License, without any additional terms or conditions.
134 | Notwithstanding the above, nothing herein shall supersede or modify
135 | the terms of any separate license agreement you may have executed
136 | with Licensor regarding such Contributions.
137 |
138 | 6. Trademarks. This License does not grant permission to use the trade
139 | names, trademarks, service marks, or product names of the Licensor,
140 | except as required for reasonable and customary use in describing the
141 | origin of the Work and reproducing the content of the NOTICE file.
142 |
143 | 7. Disclaimer of Warranty. Unless required by applicable law or
144 | agreed to in writing, Licensor provides the Work (and each
145 | Contributor provides its Contributions) on an "AS IS" BASIS,
146 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
147 | implied, including, without limitation, any warranties or conditions
148 | of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
149 | PARTICULAR PURPOSE. You are solely responsible for determining the
150 | appropriateness of using or redistributing the Work and assume any
151 | risks associated with Your exercise of permissions under this License.
152 |
153 | 8. Limitation of Liability. In no event and under no legal theory,
154 | whether in tort (including negligence), contract, or otherwise,
155 | unless required by applicable law (such as deliberate and grossly
156 | negligent acts) or agreed to in writing, shall any Contributor be
157 | liable to You for damages, including any direct, indirect, special,
158 | incidental, or consequential damages of any character arising as a
159 | result of this License or out of the use or inability to use the
160 | Work (including but not limited to damages for loss of goodwill,
161 | work stoppage, computer failure or malfunction, or any and all
162 | other commercial damages or losses), even if such Contributor
163 | has been advised of the possibility of such damages.
164 |
165 | 9. Accepting Warranty or Additional Liability. While redistributing
166 | the Work or Derivative Works thereof, You may choose to offer,
167 | and charge a fee for, acceptance of support, warranty, indemnity,
168 | or other liability obligations and/or rights consistent with this
169 | License. However, in accepting such obligations, You may act only
170 | on Your own behalf and on Your sole responsibility, not on behalf
171 | of any other Contributor, and only if You agree to indemnify,
172 | defend, and hold each Contributor harmless for any liability
173 | incurred by, or claims asserted against, such Contributor by reason
174 | of your accepting any such warranty or additional liability.
175 |
176 | END OF TERMS AND CONDITIONS
177 |
178 | APPENDIX: How to apply the Apache License to your work.
179 |
180 | To apply the Apache License to your work, attach the following
181 | boilerplate notice, with the fields enclosed by brackets "[]"
182 | replaced with your own identifying information. (Don't include
183 | the brackets!) The text should be enclosed in the appropriate
184 | comment syntax for the file format. We also recommend that a
185 | file or class name and description of purpose be included on the
186 | same "printed page" as the copyright notice for easier
187 | identification within third-party archives.
188 |
189 | Copyright [yyyy] [name of copyright owner]
190 |
191 | Licensed under the Apache License, Version 2.0 (the "License");
192 | you may not use this file except in compliance with the License.
193 | You may obtain a copy of the License at
194 |
195 | http://www.apache.org/licenses/LICENSE-2.0
196 |
197 | Unless required by applicable law or agreed to in writing, software
198 | distributed under the License is distributed on an "AS IS" BASIS,
199 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
200 | See the License for the specific language governing permissions and
201 | limitations under the License.
202 |
--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
1 | # lightMQ
2 | 由java构建的轻量级消息队列
3 |
4 | ## 特点
5 | ### 构建简单，纯java编写
6 | ### 使用方便，使用客户端可以直接通过注解监听
7 |
8 | ## 使用步骤
9 | 1. 注册topic
10 | 2. 如采用发布订阅方式，需要注册group
11 | 3. 发送消息
12 | 4. 消费消息
13 |
14 | ## 消费模式
15 | 1. 非阻塞消费，无消息直接返回null
16 | 2. 阻塞消费，无消息进入阻塞
17 |
18 | ## 生产模式
19 | 1. 顺序生产，采用同步策略，速度较慢
20 | 2. 非顺序生产，采用异步策略，速度较快
21 |
22 | ## 提供接口
23 | 1. registerTopic 注册topic，传入topic
24 | 2. registerGroup 注册group，传入topic和group
25 | 3. produce 生产消息，传入topic，key，value，key可不传
26 | 4. consume 消费消息，传入group，topic，group可不传，默认消费master
27 | 5. getAllTopic 获取所有topic
28 |
29 | ## 客户端地址
30 | https://github.com/stalary/lightMQ-client
31 |
32 | ## 部署方式
33 | - 物理机启动
34 | > mvn clean install
35 |
36 | > java -jar target/lightmq-0.1.jar
37 |
38 | - docker启动
39 | > docker run --rm --name lightmq -p 8001:8001 -d stalary/lightmq
--------------------------------------------------------------------------------
/pom.xml:
--------------------------------------------------------------------------------
1 |
2 |
4 | 4.0.0
5 |
6 | com.stalary
7 | lightmq
8 | 0.1
9 | jar
10 |
11 | lightmq
12 | a light message queue
13 |
14 |
15 | org.springframework.boot
16 | spring-boot-starter-parent
17 | 2.0.3.RELEASE
18 |
19 |
20 |
21 |
22 | UTF-8
23 | UTF-8
24 | 1.8
25 | stalary
26 |
27 |
28 |
29 |
30 | org.springframework.boot
31 | spring-boot-starter-web
32 |
33 |
34 | org.springframework.boot
35 | spring-boot-starter-aop
36 |
37 |
38 |
39 | org.projectlombok
40 | lombok
41 | true
42 |
43 |
44 | org.springframework.boot
45 | spring-boot-starter-test
46 | test
47 |
48 |
49 |
50 | org.apache.commons
51 | commons-lang3
52 | 3.7
53 |
54 |
55 |
56 | com.alibaba
57 | fastjson
58 | 1.2.47
59 |
60 |
61 |
62 | commons-io
63 | commons-io
64 | 2.6
65 |
66 |
67 |
68 | com.google.guava
69 | guava
70 | 25.1-jre
71 |
72 |
73 |
74 | com.lmax
75 | disruptor
76 | 3.4.2
77 |
78 |
79 |
80 |
81 |
82 |
83 |
84 |
85 | org.springframework.boot
86 | spring-boot-maven-plugin
87 |
88 |
89 |
90 | com.spotify
91 |
92 | docker-maven-plugin
93 | 1.0.0
94 |
95 |
96 | 0.1
97 |
98 | ${docker.image.prefix}/${project.artifactId}
99 | src/main/docker
100 |
101 |
102 | /
103 | ${project.build.directory}
104 | ${project.build.finalName}.jar
105 |
106 |
107 |
108 |
109 |
110 |
111 |
112 |
113 |
114 |
115 |
--------------------------------------------------------------------------------
/src/main/docker/Dockerfile:
--------------------------------------------------------------------------------
1 | FROM openjdk:8-jdk-alpine
2 | ADD lightmq-0.1.jar app.jar
3 | # 注意时区问题
4 | ENTRYPOINT ["java","-Duser.timezone=GMT+08","-jar","/app.jar"]
5 | EXPOSE 8001
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/CtrlTimerAdvice.java:
--------------------------------------------------------------------------------
1 |
2 | package com.stalary.lightmq;
3 |
4 | import lombok.extern.slf4j.Slf4j;
5 | import org.aspectj.lang.ProceedingJoinPoint;
6 | import org.aspectj.lang.annotation.Around;
7 | import org.aspectj.lang.annotation.Aspect;
8 | import org.aspectj.lang.reflect.MethodSignature;
9 | import org.springframework.stereotype.Component;
10 | import org.springframework.web.bind.annotation.GetMapping;
11 | import org.springframework.web.bind.annotation.PostMapping;
12 | import org.springframework.web.bind.annotation.RequestMapping;
13 |
14 | import java.util.Arrays;
15 |
16 | /**
17 | * CtrlTimerAdvice
18 | * 接口调用时间返回
19 | * @author lirongqian
20 | * @since 2018/04/09
21 | */
22 |
23 | @Slf4j
24 | @Aspect
25 | @Component
26 | public class CtrlTimerAdvice {
27 |
28 | @Around("@annotation(requestMapping)")
29 | public Object requestMappingAdvice(ProceedingJoinPoint thisJoinPoint, RequestMapping requestMapping) throws Throwable {
30 | String path = Arrays.toString(requestMapping.value());
31 | return process(thisJoinPoint, path);
32 | }
33 |
34 | @Around("@annotation(getMapping)")
35 | public Object getMappingAdvice(ProceedingJoinPoint thisJoinPoint, GetMapping getMapping) throws Throwable {
36 | String path = Arrays.toString(getMapping.value());
37 | return process(thisJoinPoint, path);
38 | }
39 |
40 | @Around("@annotation(postMapping)")
41 | public Object getMappingAdvice(ProceedingJoinPoint thisJoinPoint, PostMapping postMapping) throws Throwable {
42 | String path = Arrays.toString(postMapping.value());
43 | return process(thisJoinPoint, path);
44 | }
45 |
46 | private Object process(ProceedingJoinPoint pjp, String path) throws Throwable {
47 | long startTime = System.currentTimeMillis();
48 | Object result = pjp.proceed();
49 | long endTime = System.currentTimeMillis();
50 | MethodSignature methodSignature = (MethodSignature) pjp.getSignature();
51 | String logName = methodSignature.getMethod().getName() + "(" + path + ")";
52 | log.info(logName + ".time=" + (endTime - startTime));
53 | log.info(logName + " Args: " + Arrays.toString(pjp.getArgs()));
54 | return result;
55 | }
56 | }
57 |
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/FacadeController.java:
--------------------------------------------------------------------------------
1 | /**
2 | * @(#)FacadeController.java, 2018-06-18.
3 | *
4 | * Copyright 2018 Stalary.
5 | */
6 | package com.stalary.lightmq;
7 |
8 | import com.stalary.lightmq.data.Constant;
9 | import com.stalary.lightmq.exception.ExceptionEnum;
10 | import com.stalary.lightmq.exception.MyException;
11 | import org.springframework.util.StringUtils;
12 | import org.springframework.web.bind.annotation.GetMapping;
13 | import org.springframework.web.bind.annotation.RequestParam;
14 | import org.springframework.web.bind.annotation.RestController;
15 |
16 | import javax.annotation.Resource;
17 | import java.util.List;
18 | import java.util.stream.Collectors;
19 |
20 | /**
21 | * FacadeController
22 | *
23 | * @author lirongqian
24 | * @since 2018/06/18
25 | */
26 | @RestController
27 | public class FacadeController {
28 |
29 | @Resource
30 | private OperatorService service;
31 |
32 | /**
33 | * 生产消息
34 | * @param topic 主题
35 | * @param key 键
36 | * @param value 值
37 | * @param order 是否顺序，默认无序，异步生产多组
38 | * @return 消息发送成功
39 | */
40 | @GetMapping("/produce")
41 | public JsonResponse produce(
42 | @RequestParam String topic,
43 | @RequestParam(required = false, defaultValue = "") String key,
44 | @RequestParam(required = false, defaultValue = "false") boolean order,
45 | @RequestParam(required = false, defaultValue = "true") boolean auto,
46 | @RequestParam String value) {
47 | if (StringUtils.isEmpty(value)) {
48 | return JsonResponse.exception(new MyException(ExceptionEnum.NULL_VALUE));
49 | }
50 | if (!order) {
51 | // 异步生产，不保证顺序
52 | service.produceAsync(topic, key, value, auto);
53 | } else {
54 | service.produceSync(topic, key, value, auto);
55 | }
56 | return JsonResponse.success("消息发送成功");
57 | }
58 |
59 | /**
60 | * 消费消息，提供阻塞和非阻塞模式
61 | * @param group 消费分组
62 | * @param topic 订阅主题
63 | * @return 消费到的数据
64 | */
65 | @GetMapping("/consume")
66 | public JsonResponse consume(
67 | @RequestParam(required = false, defaultValue = Constant.DEFAULT_GROUP) String group,
68 | @RequestParam(required = false, defaultValue = "true") boolean auto,
69 | @RequestParam String topic) {
70 | return JsonResponse.success(service.consume(group, topic, auto));
71 | }
72 |
73 | /**
74 | * 获得队列中所有数据
75 | * @return 所有数据
76 | */
77 | @GetMapping("/getAll")
78 | public JsonResponse getAll() {
79 | return JsonResponse.success(service.getAllMessage());
80 | }
81 |
82 | /**
83 | * 获取所有topic
84 | * @return 所有topic
85 | */
86 | @GetMapping("/getAllTopic")
87 | public JsonResponse getAllTopic() {
88 | return JsonResponse.success(service.getAllMessage().keySet());
89 | }
90 |
91 | /**
92 | * 注册group
93 | * @param group 消息组
94 | * @param topic 订阅主题
95 | * @return
96 | */
97 | @GetMapping("/registerGroup")
98 | public JsonResponse registerGroup(
99 | @RequestParam String group,
100 | @RequestParam String topic) {
101 | service.registerGroup(group, topic);
102 | return JsonResponse.success("注册group成功");
103 | }
104 |
105 | @GetMapping("/registerTopic")
106 | public JsonResponse registerTopic(
107 | @RequestParam String topic) {
108 | service.registerTopic(topic);
109 | return JsonResponse.success("注册topic成功");
110 | }
111 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/JsonResponse.java:
--------------------------------------------------------------------------------
1 | /**
2 | * @(#)JsonResponse.java, 2018-05-31.
3 | *
4 | * Copyright 2018 Stalary.
5 | */
6 | package com.stalary.lightmq;
7 |
8 | import lombok.extern.slf4j.Slf4j;
9 |
10 | import java.io.PrintWriter;
11 | import java.io.StringWriter;
12 | import java.util.HashMap;
13 | import java.util.LinkedHashMap;
14 | import java.util.Map;
15 |
16 |
17 | /**
18 | * JsonResponse
19 | * Json格式的返回数据
20 | * {
21 | * "success": true, // 是否为正常返回，前端可读取该方法来判断是否返回正常
22 | * "code": 0, // 返回的状态码，默认为0，即正常
23 | * "msg": "", // 返回的信息，可以返回错误信息等
24 | * "data": { // 返回的数据
25 | *
26 | * },
27 | * "exception": [] //错误堆栈，无错误时不返回
28 | * }
29 | *
30 | * @author lirongqian
31 | * @since 2018/05/31
32 | */
33 | @Slf4j
34 | public class JsonResponse extends LinkedHashMap {
35 |
36 | /**
37 | * 默认的序列化编码
38 | */
39 | private static final long serialVersionUID = 1L;
40 |
41 | /**
42 | * 返回成功
43 | *
44 | * @return JsonResponse
45 | */
46 | public static JsonResponse success() {
47 | return success(null, null);
48 | }
49 |
50 | public static JsonResponse success(Object data) {
51 | return success(data, null);
52 | }
53 |
54 | public static JsonResponse success(Object data, String msg) {
55 | JsonResponse response = new JsonResponse();
56 | response.put("success", true);
57 | response.put("code", ResponseCodeEnum.SUCCESS.getCode());
58 | response.put("msg", msg);
59 | response.put("data", data);
60 | return response;
61 | }
62 |
63 | /**
64 | * 返回错误
65 | *
66 | * @return JsonResponse
67 | */
68 | public static JsonResponse fail() {
69 | return fail(null);
70 | }
71 |
72 | public static JsonResponse fail(String msg) {
73 | return fail(ResponseCodeEnum.FAIL.getCode(), msg);
74 | }
75 |
76 | public static JsonResponse fail(Integer errorCode, String msg) {
77 | JsonResponse response = new JsonResponse();
78 | response.put("success", false);
79 | response.put("code", errorCode);
80 | response.put("msg", msg);
81 | response.put("data", null);
82 | return response;
83 | }
84 |
85 | /**
86 | * 返回异常
87 | *
88 | * @return JsonResponse
89 | */
90 | public static JsonResponse exception(Exception e) {
91 | return exception(ResponseCodeEnum.EXCEPTION.getCode(), e);
92 | }
93 |
94 | public static JsonResponse exception(Integer exceptionCode, Exception e) {
95 | JsonResponse response = new JsonResponse();
96 | response.put("success", false);
97 | response.put("code", exceptionCode);
98 | response.put("msg", null);
99 | response.put("data", null);
100 | StringWriter sw = new StringWriter();
101 | e.printStackTrace(new PrintWriter(sw));
102 | response.put("exception", sw.toString());
103 | return response;
104 | }
105 |
106 | /**
107 | * 获取data中的数据
108 | *
109 | * @return Object
110 | */
111 | public Object getData() {
112 | return this.get("data");
113 | }
114 |
115 | /**
116 | * 设置data中的数据
117 | *
118 | * @return JsonResponse
119 | */
120 | public JsonResponse dataPut(String key, Object value) {
121 | mapData().put(key, value);
122 | return this;
123 | }
124 |
125 | public JsonResponse dataPutAll(Map maps) {
126 | mapData().putAll(maps);
127 | return this;
128 | }
129 |
130 | @SuppressWarnings("unchecked")
131 | private Map mapData() {
132 | // 首先获取data
133 | Object raw = this.get("data");
134 | Map data;
135 | if (raw instanceof Map) {
136 | return (Map) raw;
137 | }
138 | if (raw == null) {
139 | // data为null时，则新建并写入到JsonResponse中
140 | data = new HashMap<>();
141 | this.put("data", data);
142 | return data;
143 | }
144 | throw new IllegalArgumentException("data [" + raw
145 | + "] is not instanceof map, can not put");
146 | }
147 |
148 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/LightmqApplication.java:
--------------------------------------------------------------------------------
1 | package com.stalary.lightmq;
2 |
3 | import org.springframework.boot.SpringApplication;
4 | import org.springframework.boot.autoconfigure.SpringBootApplication;
5 | import org.springframework.boot.web.servlet.ServletComponentScan;
6 | import org.springframework.scheduling.annotation.EnableScheduling;
7 |
8 | @SpringBootApplication
9 | @ServletComponentScan
10 | @EnableScheduling
11 | public class LightmqApplication {
12 |
13 | public static void main(String[] args) {
14 | SpringApplication.run(LightmqApplication.class, args);
15 | }
16 | }
17 |
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/OperatorService.java:
--------------------------------------------------------------------------------
1 | /**
2 | * @(#)OperatorService.java, 2018-06-18.
3 | *
4 | * Copyright 2018 Stalary.
5 | */
6 | package com.stalary.lightmq;
7 |
8 | import com.stalary.lightmq.data.Constant;
9 | import com.stalary.lightmq.data.MessageDto;
10 | import com.stalary.lightmq.data.MessageFactory;
11 | import com.stalary.lightmq.data.MessageGroup;
12 | import com.stalary.lightmq.exception.ExceptionEnum;
13 | import com.stalary.lightmq.exception.MyException;
14 | import lombok.extern.slf4j.Slf4j;
15 | import org.springframework.stereotype.Service;
16 |
17 | import java.util.List;
18 | import java.util.Map;
19 | import java.util.OptionalLong;
20 | import java.util.concurrent.ExecutorService;
21 | import java.util.concurrent.Executors;
22 |
23 | /**
24 | * OperatorService
25 | *
26 | * @author lirongqian
27 | * @since 2018/06/18
28 | */
29 | @Service
30 | @Slf4j
31 | public class OperatorService {
32 |
33 | private ExecutorService executor = Executors.newFixedThreadPool(10);
34 |
35 | /**
36 | * 同步生产消息
37 | *
38 | * @param topic
39 | * @param key
40 | * @param value
41 | */
42 | public void produceSync(String topic, String key, String value, boolean auto) {
43 | Map message = getOneMessage(topic, auto);
44 | // 求出最大位移
45 | OptionalLong max = message.keySet().stream().mapToLong(v -> v).max();
46 | long offset = 0;
47 | if (max.isPresent()) {
48 | offset = max.getAsLong() + 1;
49 | }
50 | message.put(offset, new MessageDto(offset, topic, key, value));
51 | }
52 |
53 | /**
54 | * 异步生产消息,不保证消息的顺序
55 | */
56 | public void produceAsync(String topic, String key, String value, boolean auto) {
57 | // 直接放入线程池
58 | executor.execute(() -> produceSync(topic, key, value, auto));
59 | }
60 |
61 | /** 消费 **/
62 | public MessageDto consume(String group, String topic, boolean auto) {
63 | Map message = getOneMessage(topic, auto);
64 | MessageGroup oneGroup = getOneGroup(topic, group, auto);
65 | Long offset = oneGroup.getOffset();
66 | MessageDto messageDto = message.get(offset);
67 | if (messageDto == null) {
68 | return null;
69 | }
70 | // 后移offset
71 | oneGroup.incOffset();
72 | return messageDto;
73 | }
74 |
75 | /**
76 | * 注册分组
77 | **/
78 | public void registerGroup(String group, String topic) {
79 | MessageFactory.addGroup(topic, group);
80 | }
81 |
82 | /**
83 | * 注册topic
84 | **/
85 | public void registerTopic(String topic) {
86 | MessageFactory.addTopic(topic);
87 | // 创建默认分组
88 | MessageFactory.addGroup(topic, Constant.DEFAULT_GROUP);
89 | }
90 |
91 | public Map getOneMessage(String topic, boolean auto) {
92 | Map> messageMap = getAllMessage();
93 | Map message = messageMap.get(topic);
94 | if (message == null) {
95 | // 是否自动创建
96 | if (auto) {
97 | MessageFactory.addTopic(topic);
98 | message = messageMap.get(topic);
99 | } else {
100 | throw new MyException(ExceptionEnum.NO_TOPIC);
101 | }
102 | }
103 | return message;
104 | }
105 |
106 | /**
107 | * 获取当前所有消息
108 | **/
109 | public Map> getAllMessage() {
110 | return MessageFactory.messageMap;
111 | }
112 |
113 | /**
114 | * 获取一个消费者组
115 | **/
116 | public MessageGroup getOneGroup(String topic, String group, boolean auto) {
117 | Map> groupMap = MessageFactory.groupMap;
118 | List messageGroups = groupMap.get(topic);
119 | for (MessageGroup messageGroup : messageGroups) {
120 | if (group.equals(messageGroup.getGroup())) {
121 | return messageGroup;
122 | }
123 | }
124 | if (auto) {
125 | MessageFactory.addGroup(topic, group);
126 | for (MessageGroup messageGroup : messageGroups) {
127 | if (group.equals(messageGroup.getGroup())) {
128 | return messageGroup;
129 | }
130 | }
131 | }
132 | throw new MyException(ExceptionEnum.NO_GROUP);
133 | }
134 |
135 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/ResponseCodeEnum.java:
--------------------------------------------------------------------------------
1 | package com.stalary.lightmq;
2 |
3 | import lombok.Getter;
4 |
5 | /**
6 | * @author Stalary
7 | * @description 定义返回的状态码
8 | * @date 2018/5/31
9 | */
10 | public enum ResponseCodeEnum {
11 |
12 | /**
13 | * 成功返回0
14 | */
15 | SUCCESS(0, "成功"),
16 |
17 | /**
18 | * 失败返回-1，其他错误自行定义
19 | */
20 | FAIL(-1, "失败"),
21 |
22 | /**
23 | * 异常返回-100，其他异常自行定义
24 | */
25 | EXCEPTION(-100, "异常");
26 |
27 | @Getter
28 | private Integer code;
29 |
30 | @Getter
31 | private String msg;
32 |
33 | ResponseCodeEnum(Integer code, String msg) {
34 | this.code = code;
35 | this.msg = msg;
36 | }
37 |
38 |
39 | }
40 |
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/TomcatListener.java:
--------------------------------------------------------------------------------
1 | /**
2 | * @(#)TomcatListener.java, 2018-06-18.
3 | *
4 | * Copyright 2018 Stalary.
5 | */
6 | package com.stalary.lightmq;
7 |
8 | import com.alibaba.fastjson.JSON;
9 | import com.alibaba.fastjson.JSONObject;
10 | import com.stalary.lightmq.data.MessageFactory;
11 | import com.stalary.lightmq.data.SaveData;
12 | import lombok.extern.slf4j.Slf4j;
13 | import org.apache.commons.io.FileUtils;
14 | import org.apache.commons.lang3.StringUtils;
15 |
16 | import javax.servlet.ServletContextEvent;
17 | import javax.servlet.ServletContextListener;
18 | import javax.servlet.annotation.WebListener;
19 | import java.io.File;
20 | import java.io.FileWriter;
21 | import java.io.IOException;
22 |
23 | /**
24 | * TomcatListener
25 | * 监控容器的开启和关闭状态
26 | * @author lirongqian
27 | * @since 2018/06/18
28 | */
29 | @WebListener
30 | @Slf4j
31 | public class TomcatListener implements ServletContextListener {
32 |
33 | /** 当tomcat关闭时，内存中的数据持久化的文件中，下次读取文件 **/
34 | private static final String FILE_NAME = System.getProperty("user.dir") + "/save.json";
35 |
36 | @Override
37 | public void contextInitialized(ServletContextEvent servletContextEvent) {
38 | try {
39 | // 启动时读取文件中的消息
40 | File file = new File(FILE_NAME);
41 | if (file.exists()) {
42 | String content = FileUtils.readFileToString(file, "UTF-8");
43 | if (StringUtils.isNotEmpty(content)) {
44 | SaveData saveData = JSONObject.parseObject(content, SaveData.class);
45 | MessageFactory.groupMap = saveData.getGroupMap();
46 | MessageFactory.messageMap = saveData.getMessageMap();
47 | }
48 | }
49 | } catch (IOException e) {
50 | log.warn("read error.", e);
51 | }
52 | }
53 |
54 | @Override
55 | public void contextDestroyed(ServletContextEvent servletContextEvent) {
56 | try (FileWriter writer = new FileWriter(FILE_NAME, false)) {
57 | // 退出时存储消息
58 | JSONObject.writeJSONString(writer, new SaveData(MessageFactory.groupMap, MessageFactory.messageMap));
59 | } catch (IOException e) {
60 | log.warn("writer error.", e);
61 | }
62 | }
63 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/data/Constant.java:
--------------------------------------------------------------------------------
1 | /**
2 | * @(#)Constant.java, 2018-11-20.
3 | *
4 | * Copyright 2018 Stalary.
5 | */
6 | package com.stalary.lightmq.data;
7 |
8 | /**
9 | * Constant
10 | *
11 | * @author lirongqian
12 | * @since 2018/11/20
13 | */
14 | public class Constant {
15 |
16 | /** 默认分组 **/
17 | public static final String DEFAULT_GROUP = "master";
18 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/data/MessageDto.java:
--------------------------------------------------------------------------------
1 | /**
2 | * @(#)MessageDto.java, 2018-06-18.
3 | *
4 | * Copyright 2018 Stalary.
5 | */
6 | package com.stalary.lightmq.data;
7 |
8 | import lombok.Data;
9 |
10 | /**
11 | * MessageDto
12 | * 存储消息
13 | * @author lirongqian
14 | * @since 2018/06/18
15 | */
16 | @Data
17 | public class MessageDto {
18 |
19 | /**
20 | * 主题
21 | */
22 | private String topic;
23 |
24 | /**
25 | * 键，可以为空
26 | */
27 | private String key;
28 |
29 | /**
30 | * 值
31 | */
32 | private String value;
33 |
34 | private Long offset;
35 |
36 | public MessageDto(Long offset, String topic, String key, String value) {
37 | this.topic = topic;
38 | this.key = key;
39 | this.value = value;
40 | this.offset = offset;
41 | }
42 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/data/MessageFactory.java:
--------------------------------------------------------------------------------
1 | /**
2 | * @(#)MessageFactory.java, 2018-11-20.
3 | *
4 | * Copyright 2018 Stalary.
5 | */
6 | package com.stalary.lightmq.data;
7 |
8 | import com.google.common.collect.Lists;
9 | import com.stalary.lightmq.exception.ExceptionEnum;
10 | import com.stalary.lightmq.exception.MyException;
11 | import org.springframework.util.StringUtils;
12 |
13 | import java.util.HashMap;
14 | import java.util.List;
15 | import java.util.Map;
16 | import java.util.concurrent.ConcurrentHashMap;
17 |
18 | /**
19 | * MessageFactory
20 | *
21 | * @author lirongqian
22 | * @since 2018/11/20
23 | */
24 | public class MessageFactory {
25 |
26 | /** topic:group的映射 **/
27 | public static Map> groupMap = new ConcurrentHashMap<>();
28 |
29 | /** topic:(offset:message)的映射 **/
30 | public static Map> messageMap = new ConcurrentHashMap<>();
31 |
32 | public static void addGroup(String topic, String group) {
33 | if (StringUtils.isEmpty(topic)) {
34 | throw new MyException(ExceptionEnum.NULL_VALUE);
35 | }
36 | if (!messageMap.containsKey(topic)) {
37 | throw new MyException(ExceptionEnum.NO_TOPIC);
38 | }
39 | if (groupMap.containsKey(topic)) {
40 | List messageGroups = groupMap.get(topic);
41 | messageGroups.forEach(message -> {
42 | if (group.equals(message.getGroup())) {
43 | throw new MyException(ExceptionEnum.REPEAT_GROUP);
44 | }
45 | });
46 | messageGroups.add(new MessageGroup());
47 | } else {
48 | groupMap.put(topic, Lists.newArrayList(new MessageGroup()));
49 | }
50 | }
51 |
52 | public static void addTopic(String topic) {
53 | if (messageMap.containsKey(topic)) {
54 | throw new MyException(ExceptionEnum.REPEAT_TOPIC);
55 | }
56 | messageMap.put(topic, new HashMap<>());
57 | addGroup(topic, Constant.DEFAULT_GROUP);
58 | }
59 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/data/MessageGroup.java:
--------------------------------------------------------------------------------
1 | /**
2 | * @(#)MessageGroup.java, 2018-06-18.
3 | *
4 | * Copyright 2018 Stalary.
5 | */
6 | package com.stalary.lightmq.data;
7 |
8 | import lombok.AllArgsConstructor;
9 | import lombok.Data;
10 | import lombok.NoArgsConstructor;
11 |
12 | import java.util.concurrent.LinkedBlockingDeque;
13 |
14 | /**
15 | * MessageGroup
16 | * 消息组
17 | * @author lirongqian
18 | * @since 2018/06/18
19 | */
20 | @Data
21 | @AllArgsConstructor
22 | @NoArgsConstructor
23 | public class MessageGroup {
24 |
25 | /** 默认分组为master **/
26 | private String group = Constant.DEFAULT_GROUP;
27 |
28 | /** 起始offset为0 **/
29 | private Long offset = 0L;
30 |
31 | public void incOffset() {
32 | offset++;
33 | }
34 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/data/SaveData.java:
--------------------------------------------------------------------------------
1 | /**
2 | * @(#)SaveData.java, 2018-11-24.
3 | *
4 | * Copyright 2018 Stalary.
5 | */
6 | package com.stalary.lightmq.data;
7 |
8 | import lombok.AllArgsConstructor;
9 | import lombok.Data;
10 |
11 | import java.util.List;
12 | import java.util.Map;
13 |
14 | /**
15 | * SaveData
16 | *
17 | * @author lirongqian
18 | * @since 2018/11/24
19 | */
20 | @Data
21 | @AllArgsConstructor
22 | public class SaveData {
23 |
24 | /** topic:group的映射 **/
25 | private Map> groupMap;
26 |
27 | /** topic:(offset:message)的映射 **/
28 | private Map> messageMap;
29 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/exception/ExceptionEnum.java:
--------------------------------------------------------------------------------
1 | package com.stalary.lightmq.exception;
2 |
3 | /**
4 | * @author Stalary
5 | * @description
6 | * @date 2018/4/28
7 | */
8 | public enum ExceptionEnum {
9 |
10 | NO_TOPIC(1001, "不存在该topic"),
11 |
12 | NO_GROUP(1002, "不存该group"),
13 |
14 | REPEAT_TOPIC(1003, "topic重复"),
15 |
16 | REPEAT_GROUP(1004, "group重复"),
17 |
18 | NULL_VALUE(1005, "null value"),
19 |
20 | NULL_TOPIC(1005, "null topic"),
21 |
22 | NULL_GROUP(1005, "null group");
23 |
24 | Integer code;
25 |
26 | String msg;
27 |
28 | ExceptionEnum(Integer code, String msg) {
29 | this.code = code;
30 | this.msg = msg;
31 | }
32 |
33 | }
34 |
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/exception/GlobalExceptionHandler.java:
--------------------------------------------------------------------------------
1 |
2 | package com.stalary.lightmq.exception;
3 |
4 | import com.stalary.lightmq.JsonResponse;
5 | import lombok.extern.slf4j.Slf4j;
6 | import org.springframework.web.bind.annotation.ControllerAdvice;
7 | import org.springframework.web.bind.annotation.ExceptionHandler;
8 | import org.springframework.web.bind.annotation.ResponseBody;
9 |
10 | /**
11 | * GlobalExceptionHandler
12 | *
13 | * @author lirongqian
14 | * @since 2018/06/18
15 | */
16 | @ControllerAdvice
17 | @Slf4j
18 | public class GlobalExceptionHandler {
19 |
20 | @ExceptionHandler(value = Exception.class)
21 | @ResponseBody
22 | public JsonResponse handle(Exception e) {
23 | if (e instanceof MyException) {
24 | MyException myException = (MyException) e;
25 | return JsonResponse.fail(myException.getCode(), myException.getMessage());
26 | } else {
27 | log.error("[系统异常] ", e);
28 | return JsonResponse.exception(e);
29 | }
30 | }
31 |
32 | }
--------------------------------------------------------------------------------
/src/main/java/com/stalary/lightmq/exception/MyException.java:
--------------------------------------------------------------------------------
1 |
2 | package com.stalary.lightmq.exception;
3 |
4 |
5 | import lombok.Data;
6 | import lombok.EqualsAndHashCode;
7 |
8 | /**
9 | * @author lirongqian
10 | * @description
11 | * @date 2018/06/18
12 | */
13 | @Data
14 | @EqualsAndHashCode(callSuper = true)
15 | public class MyException extends RuntimeException {
16 |
17 | private Integer code;
18 |
19 | private String message;
20 |
21 | public MyException(Integer code, String message) {
22 | super(message);
23 | this.code = code;
24 | this.message = message;
25 | }
26 |
27 | public MyException(ExceptionEnum exceptionEnum) {
28 | super(exceptionEnum.msg);
29 | this.code = exceptionEnum.code;
30 | this.message = exceptionEnum.msg;
31 | }
32 |
33 | }
34 |
35 |
--------------------------------------------------------------------------------
/src/main/resources/application.yml:
--------------------------------------------------------------------------------
1 | server:
2 | port: 8001
3 | max-http-header-size: 8192
--------------------------------------------------------------------------------
/src/main/resources/logback-spring.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 |
9 |
10 | logback
11 |
12 |
13 |
14 |
15 |
16 |
17 |
18 |
19 |
20 |
21 |
22 |
23 |
24 |
25 |
26 |
27 | info
28 |
29 |
30 | ${CONSOLE_LOG_PATTERN}
31 |
32 | UTF-8
33 |
34 |
35 |
36 |
37 |
38 |
39 |
40 |
41 |
42 | ${log.path}/log_debug.log
43 |
44 |
45 | %d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{50} - %msg%n
46 | UTF-8
47 |
48 |
49 |
50 |
51 | ${log.path}/debug/log-debug-%d{yyyy-MM-dd}.%i.log
52 |
53 | 100MB
54 |
55 |
56 | 15
57 |
58 |
59 |
60 | debug
61 | ACCEPT
62 | DENY
63 |
64 |
65 |
66 |
67 |
68 |
69 | ${log.path}/log_info.log
70 |
71 |
72 | %d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{50} - %msg%n
73 | UTF-8
74 |
75 |
76 |
77 |
78 | ${log.path}/info/log-info-%d{yyyy-MM-dd}.%i.log
79 |
80 | 100MB
81 |
82 |
83 | 15
84 |
85 |
86 |
87 | info
88 | ACCEPT
89 | DENY
90 |
91 |
92 |
93 |
94 |
95 |
96 | ${log.path}/log_warn.log
97 |
98 |
99 | %d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{50} - %msg%n
100 | UTF-8
101 |
102 |
103 |
104 | ${log.path}/warn/log-warn-%d{yyyy-MM-dd}.%i.log
105 |
106 | 100MB
107 |
108 |
109 | 15
110 |
111 |
112 |
113 | warn
114 | ACCEPT
115 | DENY
116 |
117 |
118 |
119 |
120 |
121 |
122 |
123 | ${log.path}/log_error.log
124 |
125 |
126 | %d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %-5level %logger{50} - %msg%n
127 | UTF-8
128 |
129 |
130 |
131 | ${log.path}/error/log-error-%d{yyyy-MM-dd}.%i.log
132 |
133 | 100MB
134 |
135 |
136 | 15
137 |
138 |
139 |
140 | ERROR
141 | ACCEPT
142 | DENY
143 |
144 |
145 |
146 |
156 |
157 |
158 |
163 |
164 |
165 |
171 |
172 |
173 |
174 |
175 |
176 |
177 |
178 |
179 |
180 |
--------------------------------------------------------------------------------