com.gemstone.gemfire.admin.internal.LogCollator Maven / Gradle / Ivy

/*
 * Copyright (c) 2010-2015 Pivotal Software, Inc. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you
 * may not use this file except in compliance with the License. You
 * may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * permissions and limitations under the License. See accompanying
 * LICENSE file.
 */

package com.gemstone.gemfire.admin.internal;

import com.gemstone.gemfire.internal.admin.GfManagerAgent;
import com.gemstone.gemfire.internal.admin.GemFireVM;
import com.gemstone.gemfire.internal.admin.ApplicationVM;
import com.gemstone.gemfire.internal.MergeLogFiles;

import java.io.ByteArrayInputStream;
import java.io.InputStream;  
import java.io.PrintWriter;  
import java.io.StringWriter;  
import java.util.ArrayList;
import java.util.List;

public class LogCollator {
  
  private GfManagerAgent system;
  private List logTails;
    
  public LogCollator() {
  }
  
  public String collateLogs(GfManagerAgent system) {
    try {
      if (system == null) {
        return "";
      }
      this.system = system;
      this.logTails = new ArrayList();
      gatherActiveLogs();
      gatherInactiveLogs();
      return mergeLogs();
    }
    finally {
      this.system = null;
      this.logTails = null;
    }
  }

  // -------------------------------------------------------------------------
  
  private String mergeLogs() {
    // combine logs...
    InputStream[] logFiles = new InputStream[this.logTails.size()];
    String[] logFileNames = new String[logFiles.length];
    for (int i = 0; i < this.logTails.size(); i++) {
      Loglet loglet = (Loglet) this.logTails.get(i);
      logFiles[i] = new ByteArrayInputStream(loglet.tail.getBytes());
      logFileNames[i] = loglet.name;
    }
    
    // delegate to MergeLogFiles...
    StringWriter writer = new StringWriter();
    PrintWriter mergedLog = new PrintWriter(writer);
    if (!MergeLogFiles.mergeLogFiles(logFiles, logFileNames, mergedLog)) {
      return writer.toString();
    } 
    else {
      return "";
    }
  }

  private void gatherActiveLogs() {
    ApplicationVM[] runningsApps = this.system.listApplications();
    for (int i = 0; i < runningsApps.length; i++) {
      addLogFrom(runningsApps[i]);
    }
  }
  
  private void gatherInactiveLogs() {
    /* not yet supported....
    if (useStopped) {
      LogViewHelper helper = new LogViewHelper();
      for (Iterator iter = stoppedNodes.iterator(); iter.hasNext(); ) {
        Object adminEntity = iter.next();
        helper.setAdminEntity(adminEntity);
        try {
          if (helper.logViewAvailable()) {
            String[] logs = helper.getSystemLogs();
            addTail(allTails, logs, adminEntity.toString());
          }
        } catch (Exception e) {
          Service.getService().reportSystemError(e);
        }
      }
    }
    */
  }
  
  private void addLogFrom(GemFireVM vm) {
    String name = null;
    name = vm.toString();
    String[] logs = vm.getSystemLogs();
    addTail(name, logs);
  }

  private void addTail(String logName, String[] logs) {
    if (logs.length > 0) {
      String tail = (logs.length > 1) ? logs[1] : logs[0];      
      this.logTails.add(new Loglet(logName, tail));
    }
  }

  /*
  public void setUseStoppedManagers(boolean useStopped) {
    this.useStopped = useStopped;
  }
  */

  private static class Loglet {
    String name;
    String tail;
    Loglet(String name, String tail) {
      this.name = name;
      this.tail = tail;
    }
  }
  
}