feat: Unified API management using ProtocolConstant #209

This commit is contained in:
tjq 2021-02-08 22:22:17 +08:00
parent eda39a6372
commit a575b65320
2 changed files with 10 additions and 12 deletions

View File

@ -8,11 +8,11 @@ package com.github.kfcfans.powerjob.common;
*/
public class ProtocolConstant {
public static final String SERVER_PATH_HEARTBEAT = "heartbeat";
public static final String SERVER_PATH_STATUS_REPORT = "statusReport";
public static final String SERVER_PATH_LOG_REPORT = "logReport";
public static final String SERVER_PATH_HEARTBEAT = "/server/heartbeat";
public static final String SERVER_PATH_STATUS_REPORT = "/server/statusReport";
public static final String SERVER_PATH_LOG_REPORT = "/server/logReport";
public static final String WORKER_PATH_DISPATCH_JOB = "/worker/dispatchJob";
public static final String WORKER_PATH_DISPATCH_JOB = "/worker/runJob";
public static final String WORKER_PATH_STOP_INSTANCE = "/worker/stopInstance";
public static final String WORKER_PATH_QUERY_INSTANCE_INFO = "/worker/queryInstanceInfo";
}

View File

@ -1,6 +1,7 @@
package com.github.kfcfans.powerjob.server.handler.outer;
import com.github.kfcfans.powerjob.common.OmsConstant;
import com.github.kfcfans.powerjob.common.ProtocolConstant;
import com.github.kfcfans.powerjob.common.request.TaskTrackerReportInstanceStatusReq;
import com.github.kfcfans.powerjob.common.request.WorkerHeartbeat;
import com.github.kfcfans.powerjob.common.request.WorkerLogReportReq;
@ -15,7 +16,6 @@ import io.vertx.ext.web.Router;
import io.vertx.ext.web.RoutingContext;
import io.vertx.ext.web.handler.BodyHandler;
import java.util.Optional;
import java.util.Properties;
import static com.github.kfcfans.powerjob.server.handler.outer.WorkerRequestHandler.getWorkerRequestHandler;
@ -28,8 +28,6 @@ import static com.github.kfcfans.powerjob.server.handler.outer.WorkerRequestHand
*/
public class WorkerRequestHttpHandler extends AbstractVerticle {
private static final String HTTP_PREFIX = "/wrh/v1/";
@Override
public void start() throws Exception {
@ -41,18 +39,18 @@ public class WorkerRequestHttpHandler extends AbstractVerticle {
Router router = Router.router(vertx);
router.route().handler(BodyHandler.create());
router.post(HTTP_PREFIX + "heartbeat")
router.post(ProtocolConstant.SERVER_PATH_HEARTBEAT)
.handler(ctx -> {
WorkerHeartbeat heartbeat = ctx.getBodyAsJson().mapTo(WorkerHeartbeat.class);
getWorkerRequestHandler().onReceiveWorkerHeartbeat(heartbeat);
});
router.post(HTTP_PREFIX + "instanceStatusReport")
router.post(ProtocolConstant.SERVER_PATH_STATUS_REPORT)
.blockingHandler(ctx -> {
TaskTrackerReportInstanceStatusReq req = ctx.getBodyAsJson().mapTo(TaskTrackerReportInstanceStatusReq.class);
Optional<AskResponse> askResponseOpt = getWorkerRequestHandler().onReceiveTaskTrackerReportInstanceStatusReq(req);
askResponseOpt.ifPresent(askResponse -> out(ctx, askResponse));
getWorkerRequestHandler().onReceiveTaskTrackerReportInstanceStatusReq(req);
out(ctx, AskResponse.succeed(null));
});
router.post(HTTP_PREFIX + "logReport")
router.post(ProtocolConstant.SERVER_PATH_LOG_REPORT)
.blockingHandler(ctx -> {
WorkerLogReportReq req = ctx.getBodyAsJson().mapTo(WorkerLogReportReq.class);
getWorkerRequestHandler().onReceiveWorkerLogReportReq(req);