forked from actframework/actframework
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathJobAdmin.java
More file actions
117 lines (103 loc) · 3.98 KB
/
JobAdmin.java
File metadata and controls
117 lines (103 loc) · 3.98 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
package act.job;
/*-
* #%L
* ACT Framework
* %%
* Copyright (C) 2014 - 2017 ActFramework
* %%
* 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.
* #L%
*/
import act.cli.*;
import act.event.OnEvent;
import act.util.PropertySpec;
import act.util.SimpleProgressGauge;
import act.ws.WebSocketConnectEvent;
import act.ws.WebSocketConnectionManager;
import act.ws.WebSocketContext;
import act.ws.WsEndpoint;
import com.alibaba.fastjson.JSONObject;
import org.osgl.$;
import org.osgl.util.C;
import org.osgl.util.S;
import javax.inject.Inject;
import java.util.List;
import java.util.concurrent.ScheduledThreadPoolExecutor;
/**
* Provide admin service to act {@link AppJobManager}
*/
@SuppressWarnings("unused")
public class JobAdmin {
/**
* List all jobs in the job manager
* @return a list of {@link Job jobs}
*/
@Command(value = "act.job.list", help = "List jobs")
@PropertySpec(Job.BRIEF_VIEW)
@TableView
public List<Job> listJobs(@Optional(lead = "-q") final String q, AppJobManager jobManager) {
C.List<Job> jobs = jobManager.jobs().append(jobManager.virtualJobs());
if (S.notBlank(q)) {
jobs = jobs.filter(new $.Predicate<Job>() {
@Override
public boolean test(Job job) {
return job.toString().contains(q);
}
});
}
return jobs;
}
@Command(value = "act.job.show", help = "Show job details")
@JsonView
@PropertySpec(Job.DETAIL_VIEW)
public Job getJob(@Required("specify job id") final String id, AppJobManager jobManager) {
return jobManager.jobById(id);
}
@Command(value = "act.job.progress", help = "Show job progress")
public int getJobProgress(@Required("specify job id") final String id, AppJobManager jobManager) {
Job job = jobManager.jobById(id);
return null == job ? -1 : job.getProgressInPercent();
}
@Command(name = "act.job.cancel", help = "Cancel a job")
public void cancel(@Required("specify job id") String id, AppJobManager jobManager) {
jobManager.cancel(id);
}
@Command(value = "act.job.scheduler", help = "Show Job manager scheduler status")
public String getSchedulerStatus(AppJobManager jobManager) {
ScheduledThreadPoolExecutor executor = jobManager.executor();
JSONObject json = new JSONObject();
json.put("is terminating", executor.isTerminating());
json.put("is terminated", executor.isTerminated());
json.put("is shutdown", executor.isShutdown());
json.put("# of runnable in the queue", executor.getQueue().size());
json.put("active count", executor.getActiveCount());
json.put("# of completed tasks", executor.getActiveCount());
json.put("core pool size", executor.getCorePoolSize());
json.put("pool size", executor.getPoolSize());
return json.toJSONString();
}
public static class WebsocketEndpoints {
@WsEndpoint("/~/job/{id}/progress")
public static class JobProgress {
@Inject
private WebSocketConnectionManager connectionManager;
@OnEvent
public void onConnect(WebSocketConnectEvent event) {
WebSocketContext context = event.source();
String jobId = context.actionContext().paramVal("id");
String tag = SimpleProgressGauge.wsJobProgressTag(jobId);
connectionManager.subscribe(context.actionContext().session(), tag);
}
}
}
}