コード例 #1
0
  /** {@inheritDoc} */
  @Override
  public OperationResult execute(
      OperationContext context, ModelNode operation, ResultHandler resultHandler)
      throws OperationFailedException {

    validator.validate(operation);

    String name = operation.require(NAME).asString();
    String runtimeName =
        operation.hasDefined(RUNTIME_NAME) ? operation.get(RUNTIME_NAME).asString() : name;
    byte[] hash;
    if (operation.hasDefined(INPUT_STREAM_INDEX) && operation.hasDefined(HASH)) {
      throw new OperationFailedException(
          new ModelNode().set("Can't pass in both an input-stream-index and a hash"));
    } else if (operation.hasDefined(INPUT_STREAM_INDEX)) {
      InputStream in = getContents(context, operation);
      try {
        hash = deploymentRepository.addDeploymentContent(in);
      } catch (IOException e) {
        throw new OperationFailedException(new ModelNode().set(e.toString()));
      }
    } else if (operation.hasDefined(HASH)) {

      hash = operation.get(HASH).asBytes();
      if (!deploymentRepository.hasDeploymentContent(hash)) {
        throw new OperationFailedException(
            new ModelNode()
                .set(
                    String.format(
                        "No deployment content with hash %s is available in the deployment content repository.",
                        HashUtil.bytesToHexString(hash))));
      }
    } else {
      throw new OperationFailedException(
          new ModelNode().set("Neither an attachment or a hash were passed in"));
    }

    ModelNode rootModel = context.getSubModel();
    ModelNode deployments = rootModel.get(DEPLOYMENT);

    ModelNode replaceNode = deployments.hasDefined(name) ? deployments.get(name) : null;
    if (replaceNode == null) {
      throw new OperationFailedException(
          new ModelNode().set(String.format("No deployment with name %s found", name)));
    }

    boolean start = replaceNode.get(ENABLED).asBoolean();

    ModelNode deployNode = new ModelNode();
    deployNode.get(NAME).set(name);
    deployNode.get(RUNTIME_NAME).set(runtimeName);
    deployNode.get(HASH).set(hash);
    deployNode.get(ENABLED).set(start);

    deployments.get(name).set(deployNode);

    ModelNode compensatingOp = operation.clone();
    compensatingOp.get(RUNTIME_NAME).set(replaceNode.get(RUNTIME_NAME).asString());
    compensatingOp.get(HASH).set(replaceNode.get(HASH).asBytes());
    if (operation.hasDefined(INPUT_STREAM_INDEX)) {
      operation.remove(INPUT_STREAM_INDEX);
    }

    if (start) {
      DeploymentHandlerUtil.replace(deployNode, name, context, resultHandler);
    } else {
      resultHandler.handleResultComplete();
    }

    return new BasicOperationResult(compensatingOp);
  }
コード例 #2
0
  /**
   * Scan the given directory for content changes.
   *
   * @param directory the directory to scan
   * @param updates the update list;
   * @param foundDeployed place to store marker files found in the directory; key is the name of the
   *     deployment, value is the marker file
   * @param newlyAdded place to store names of newly added content
   * @param registeredDeployments TODO
   * @return the builder the current builder following any changes
   */
  private void scanDirectory(
      File directory,
      final List<ModelNode> updates,
      Map<String, File> foundDeployed,
      Set<String> newlyAdded,
      Set<String> registeredDeployments) {

    // TODO externalize config of filter?
    File[] children = directory.listFiles(filter);
    if (children == null) {
      return;
    }

    for (File child : children) {

      String fileName = child.getName();

      if (fileName.endsWith(DEPLOYED)) {
        String origName = fileName.substring(0, fileName.length() - DEPLOYED.length());
        foundDeployed.put(origName, child);
      } else if (child.isDirectory()) {
        int idx = fileName.lastIndexOf('.');
        if (idx > -1 && ARCHIVES.contains(fileName.substring(idx))) {
          // FIXME handle exploded deployments
          log.warnf(
              "%s is an exploded deployment and exploded deployments are not currently handled by %s",
              child.getName(), getClass().getSimpleName());
        } else {
          // It's just a dir for organizing content. Recurse
          scanDirectory(child, updates, foundDeployed, newlyAdded, registeredDeployments);
        }
      } else {
        // Found a single non-marker file
        boolean uploaded = false;
        if (registeredDeployments.contains(fileName)) {

          byte[] hash = new byte[0];
          try {
            final InputStream inputStream = new FileInputStream(child);
            hash = deploymentRepository.addDeploymentContent(fileName, fileName, inputStream);
          } catch (IOException e) {
            log.error("Failed to add content to deployment repository for [" + fileName + "]", e);
            continue;
          }
          updates.add(getFullReplaceOperation(fileName, hash));
          uploaded = true;
        } else {
          byte[] hash = new byte[0];
          try {
            final InputStream inputStream = new FileInputStream(child);
            hash = deploymentRepository.addDeploymentContent(fileName, fileName, inputStream);
          } catch (IOException e) {
            log.error("Failed to add content to deployment repository for [" + fileName + "]", e);
            continue;
          }
          updates.add(getAddOperation(fileName, hash));
          updates.add(getDeployOperation(fileName));
          uploaded = true;
        }

        if (uploaded && replaceWithDeployedMarker(child)) {
          newlyAdded.add(fileName);
        }
      }
    }
  }