Пример #1
0
  public WebResponse beWebExecute(MasterContext context, WebRequest req) {
    // 判断job是否已经在运行中,或者在队列中
    // 如果在,抛出异常,已经在执行中
    // 如果不在,将该job放入等待队列
    SocketLog.info("receive web debug request,rid=" + req.getRid() + ",debugId=" + req.getId());
    String debugId = req.getId();
    for (JobElement e : new ArrayList<JobElement>(context.getDebugQueue())) {
      if (e.getJobID().equals(debugId)) {
        WebResponse resp =
            WebResponse.newBuilder()
                .setRid(req.getRid())
                .setOperate(WebOperate.ExecuteDebug)
                .setStatus(Status.ERROR)
                .setErrorText("已经在队列中,无法再次运行")
                .build();
        return resp;
      }
    }

    DebugHistory debug = context.getDebugHistoryManager().findDebugHistory(debugId);
    context.getMaster().debug(debug);

    WebResponse resp =
        WebResponse.newBuilder()
            .setRid(req.getRid())
            .setOperate(WebOperate.ExecuteDebug)
            .setStatus(Status.OK)
            .build();
    SocketLog.info("send web debug response,rid=" + req.getRid() + ",debugId=" + debugId);
    return resp;
  }
Пример #2
0
  @Override
  public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) throws Exception {
    final Channel channel = ctx.getChannel();
    SocketMessage sm = (SocketMessage) e.getMessage();
    if (sm.getKind() == Kind.REQUEST) {
      final Request request = Request.newBuilder().mergeFrom(sm.getBody()).build();
      if (request.getOperate() == Operate.HeartBeat) {
        beHeartBeat.beHeartBeat(context, channel, request);
      }
    } else if (sm.getKind() == Kind.WEB_REUQEST) {
      final WebRequest request = WebRequest.newBuilder().mergeFrom(sm.getBody()).build();
      if (request.getOperate() == WebOperate.ExecuteJob) {
        completionService.submit(
            new Callable<ChannelResponse>() {
              public ChannelResponse call() throws Exception {
                return new ChannelResponse(channel, beWebExecute.beWebExecute(context, request));
              }
            });
      } else if (request.getOperate() == WebOperate.CancelJob) {
        completionService.submit(
            new Callable<ChannelResponse>() {
              public ChannelResponse call() throws Exception {
                return new ChannelResponse(channel, beWebCancel.beWebCancel(context, request));
              }
            });
      } else if (request.getOperate() == WebOperate.UpdateJob) {
        completionService.submit(
            new Callable<ChannelResponse>() {
              public ChannelResponse call() throws Exception {
                return new ChannelResponse(channel, beUpdate.beWebUpdate(context, request));
              }
            });
      } else if (request.getOperate() == WebOperate.ExecuteDebug) {
        completionService.submit(
            new Callable<ChannelResponse>() {
              public ChannelResponse call() throws Exception {
                return new ChannelResponse(channel, beDebug.beWebExecute(context, request));
              }
            });
      }
    } else if (sm.getKind() == Kind.RESPONSE) {
      for (ResponseListener lis : new ArrayList<ResponseListener>(listeners)) {
        lis.onResponse(Response.newBuilder().mergeFrom(sm.getBody()).build());
      }
    } else if (sm.getKind() == Kind.WEB_RESPONSE) {
      for (ResponseListener lis : new ArrayList<ResponseListener>(listeners)) {
        lis.onWebResponse(WebResponse.newBuilder().mergeFrom(sm.getBody()).build());
      }
    }

    super.messageReceived(ctx, e);
  }
Пример #3
0
 private SocketMessage wapper(WebResponse resp) {
   return SocketMessage.newBuilder()
       .setKind(Kind.WEB_RESPONSE)
       .setBody(resp.toByteString())
       .build();
 }