Ejemplo n.º 1
0
  public final boolean unify(final JIPCons params, Hashtable varsTbl) {
    JIPTerm output = params.getNth(1);
    JIPTerm term = params.getNth(2);

    // check if input is a variable
    if (output instanceof JIPVariable) {
      // try to extract the term
      if (!((JIPVariable) output).isBounded()) {
        throw new JIPInstantiationException(1);
      } else {
        // extracts the term
        output = ((JIPVariable) output).getValue();
      }
    }

    // Get the stream
    StreamInfo sinfo = JIPio.getOutputStreamInfo(output, false);

    //        String mode = sinfo.getProperties().getProperty("mode");
    //        if(!(mode.equals("mode(write)") || mode.equals("mode(append)")))
    //        	throw new JIPPermissionException("output", "stream", sinfo.getAlias());
    //        if(!sinfo.getProperties().getProperty("type").equals("type(text)"))
    //        	throw new JIPPermissionException("output", "binary_stream", sinfo.getAlias());

    OutputStream writer = JIPio.getOutputStream(sinfo.getHandle(), getJIPEngine());

    try {
      writer.write(term.toString(getJIPEngine()).getBytes(getJIPEngine().getEncoding()));
      writer.flush();
    } catch (IOException ex) {
      throw new JIPRuntimeException(JIPio.ERR_IOEXCEPTION, ex.getMessage());
    }

    return true;
  }
  @Override
  public AggregationInfo chronology(String dir, String targetFile) throws IOException {
    log.info("Try to aggregate {} into file {}", dir, targetFile);
    Collection<Hessian2Input> inputStreams = new ArrayList<Hessian2Input>();
    Set<String> fileNameList = fileStorage.getFileNameList(dir);
    if (fileNameList.isEmpty()) {
      log.info("Nothing to aggregate. Directory {} is empty.", dir);
      new Hessian2Output(fileStorage.create(targetFile)).close();
      return new AggregationInfo(0, 0, 0);
    }
    for (String fileName : fileNameList) {
      try {
        InputStream in = fileStorage.open(fileName);
        inputStreams.add(new Hessian2Input(in));
      } catch (FileNotFoundException e) {
        log.warn(e.getMessage(), e);
      }
    }

    int count = 0;
    long minTime = 0;
    long maxTime = 0;

    Hessian2Output out = null;
    OutputStream os = null;
    try {
      if (fileStorage.delete(targetFile, false)) {
        log.warn("Target file {} did not deleted!", targetFile);
      }
      os = fileStorage.create(targetFile);
      out = new Hessian2Output(os);
      MinMaxPriorityQueue<StreamInfo> queue = MinMaxPriorityQueue.create();
      for (Hessian2Input inputStream : inputStreams) {
        LogEntry logEntry;
        try {
          logEntry = (LogEntry) inputStream.readObject();
        } catch (EOFException e) {
          continue;
        }
        queue.add(new StreamInfo(inputStream, logEntry));
      }

      while (!queue.isEmpty()) {
        StreamInfo<LogEntry> streamInfo = queue.removeFirst();
        out.writeObject(streamInfo.lastLogEntry);

        if (count == 0) {
          minTime = streamInfo.lastLogEntry.getTime();
          maxTime = streamInfo.lastLogEntry.getTime();
        } else {
          maxTime = streamInfo.lastLogEntry.getTime();
        }

        count++;
        LogEntry logEntry;
        try {
          logEntry = (LogEntry) streamInfo.stream.readObject();
        } catch (EOFException e) {
          continue;
        }
        streamInfo.lastLogEntry = logEntry;
        queue.add(streamInfo);
      }
    } finally {
      if (out != null) {
        out.close();
        os.close();
      }
    }

    return new AggregationInfo(minTime, maxTime, count);
  }