コード例 #1
0
  public OMElement getCEPPublisher(OMElement request) throws XMLStreamException {
    request.build();
    request.detach();

    OMElement tenantId =
        request.getFirstChildWithName(
            new QName(
                ManagerServiceConstants.NAMESPACE, ManagerServiceConstants.ELEMENT_TENANT_ID));
    OMElement executionPlan =
        request.getFirstChildWithName(
            new QName(
                ManagerServiceConstants.NAMESPACE, ManagerServiceConstants.ELEMENT_EXEC_PLAN));
    OMElement requesterIp =
        request.getFirstChildWithName(
            new QName(
                ManagerServiceConstants.NAMESPACE, ManagerServiceConstants.ELEMENT_REQUESTER_IP));
    String key = getKey(executionPlan.getText(), tenantId.getText());
    log.info("CEP Publisher requested for  " + key);

    Set<Endpoint> endpointSet = cepPublishers.get(key);
    Endpoint selectedEndpoint = selectEndpoint(endpointSet, requesterIp.getText());
    OMElement response =
        factory.createOMElement(
            ManagerServiceConstants.ELEMENT_CEP_PUBLISHER_RESPONSE, OMNamespace);
    if (selectedEndpoint != null) {
      OMElement hostNameElement =
          factory.createOMElement(ManagerServiceConstants.ELEMENT_HOST_NAME, OMNamespace);
      OMElement portElement =
          factory.createOMElement(ManagerServiceConstants.ELEMENT_PORT, OMNamespace);

      OMText hostNameText = factory.createOMText(hostNameElement, selectedEndpoint.getHostName());
      OMText portText =
          factory.createOMText(portElement, Integer.toString(selectedEndpoint.getPort()));

      hostNameElement.addChild(hostNameText);
      portElement.addChild(portText);
      response.addChild(hostNameElement);
      response.addChild(portElement);
      log.info(
          "Returning CEP Publisher:"
              + selectedEndpoint.getHostName()
              + ":"
              + selectedEndpoint.getPort());
    } else {
      log.warn("No CEP publishers registered " + key);
    }

    return response;
  }
コード例 #2
0
  private Endpoint selectEndpoint(Set<Endpoint> endpointSet, String requesterIp) {
    Endpoint selectedEndpoint = null;

    if (endpointSet != null && !endpointSet.isEmpty()) {
      // If  there's a storm receiver/cep publisher in the same host as requester IP select it
      if ("".equals(requesterIp) == false) {
        for (Endpoint endpoint : endpointSet) {
          if (endpoint.getHostName().equals(requesterIp)) {
            selectedEndpoint = endpoint;

            if (log.isDebugEnabled()) {
              log.debug(
                  "Selecting"
                      + endpoint.toString()
                      + " since it's in the same host as the requester");
            }
            break;
          }
        }
      }
      // If there are no endpoints in the same host. Select the endpoint with lease number of
      // connections
      if (selectedEndpoint == null) {
        int minConnectionCount = Integer.MAX_VALUE;

        for (Endpoint endpoint : endpointSet) {
          if (log.isDebugEnabled()) {
            log.debug(
                "Endpoint "
                    + endpoint.toString()
                    + " has "
                    + endpoint.getConnectionCount()
                    + " connections.");
          }

          if (endpoint.getConnectionCount() < minConnectionCount) {
            minConnectionCount = endpoint.getConnectionCount();
            selectedEndpoint = endpoint;
          }
        }
      }

      selectedEndpoint.setConnectionCount(selectedEndpoint.getConnectionCount() + 1);
    }
    return selectedEndpoint;
  }