org.apache.geode.management.internal.cli.commands.ExportLogsInterceptor.java Source code

Introduction

Here is the source code for org.apache.geode.management.internal.cli.commands.ExportLogsInterceptor.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more contributor license
 * agreements. See the NOTICE file distributed with this work for additional information regarding
 * copyright ownership. The ASF licenses this file to You 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.
 */
package org.apache.geode.management.internal.cli.commands;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.geode.internal.logging.LogService;
import org.apache.geode.internal.logging.log4j.LogLevel;
import org.apache.geode.management.cli.Result;
import org.apache.geode.management.internal.cli.AbstractCliAroundInterceptor;
import org.apache.geode.management.internal.cli.GfshParseResult;
import org.apache.geode.management.internal.cli.functions.ExportLogsFunction;
import org.apache.geode.management.internal.cli.result.ResultBuilder;
import org.apache.logging.log4j.Level;
import org.apache.logging.log4j.Logger;

import java.io.File;
import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.time.LocalDateTime;
import java.util.Map;

/**
 * after the export logs, will need to copy the tempFile to the desired location and delete the temp
 * file.
 */
public class ExportLogsInterceptor extends AbstractCliAroundInterceptor {
    private static final Logger logger = LogService.getLogger();

    @Override
    public Result preExecution(GfshParseResult parseResult) {
        // the arguments are in the order of it's being declared
        Map<String, String> arguments = parseResult.getParamValueStrings();

        // validates groupId and memberIds not both set
        if (arguments.get("group") != null && arguments.get("member") != null) {
            return ResultBuilder.createUserErrorResult("Can't specify both group and member.");
        }

        // validate log level
        String logLevel = arguments.get("log-level");
        if (StringUtils.isBlank(logLevel) || LogLevel.getLevel(logLevel) == null) {
            return ResultBuilder.createUserErrorResult("Invalid log level: " + logLevel);
        }

        // validate start date and end date
        String start = arguments.get("start-time");
        String end = arguments.get("end-time");
        if (start != null && end != null) {
            // need to make sure end is later than start
            LocalDateTime startTime = ExportLogsFunction.parseTime(start);
            LocalDateTime endTime = ExportLogsFunction.parseTime(end);
            if (startTime.isAfter(endTime)) {
                return ResultBuilder.createUserErrorResult("start-time has to be earlier than end-time.");
            }
        }

        // validate onlyLogs and onlyStats
        boolean onlyLogs = Boolean.parseBoolean(arguments.get("logs-only"));
        boolean onlyStats = Boolean.parseBoolean(arguments.get("stats-only"));
        if (onlyLogs && onlyStats) {
            return ResultBuilder.createUserErrorResult("logs-only and stats-only can't both be true");
        }

        return ResultBuilder.createInfoResult("");
    }

    @Override
    public Result postExecution(GfshParseResult parseResult, Result commandResult, Path tempFile) {
        // in the command over http case, the command result is in the downloaded temp file
        if (tempFile != null) {
            Path dirPath;
            String dirName = parseResult.getParamValueStrings().get("dir");
            if (StringUtils.isBlank(dirName)) {
                dirPath = Paths.get(System.getProperty("user.dir")).toAbsolutePath();
            } else {
                dirPath = Paths.get(dirName).toAbsolutePath();
            }
            String fileName = "exportedLogs_" + System.currentTimeMillis() + ".zip";
            File exportedLogFile = dirPath.resolve(fileName).toFile();
            try {
                FileUtils.copyFile(tempFile.toFile(), exportedLogFile);
                FileUtils.deleteQuietly(tempFile.toFile());
                commandResult = ResultBuilder
                        .createInfoResult("Logs exported to: " + exportedLogFile.getAbsolutePath());
            } catch (IOException e) {
                logger.error(e.getMessage(), e);
                commandResult = ResultBuilder.createGemFireErrorResult(e.getMessage());
            }
        } else if (commandResult.getStatus() == Result.Status.OK) {
            commandResult = ResultBuilder.createInfoResult(
                    "Logs exported to the connected member's file system: " + commandResult.nextLine());
        }
        return commandResult;
    }
}