ugsync/src/main/java/org/apache/ranger/usergroupsync/UserSyncMetricsProducer.java

/*
 * 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.ranger.usergroupsync;

import java.io.File;

import org.apache.log4j.Logger;
import org.apache.ranger.plugin.util.RangerMetricsUtil;
import org.apache.ranger.unixusersync.config.UserGroupSyncConfig;

public class UserSyncMetricsProducer implements Runnable {

	private static final Logger LOG = Logger.getLogger(UserSyncMetricsProducer.class);
	private boolean shutdownFlag = false;

	public static void main(String[] args) {
		UserSyncMetricsProducer userSyncMetrics = new UserSyncMetricsProducer();
		userSyncMetrics.run();
		/*
		 * try { userSyncMetrics.writeJVMMetrics(); } catch (Throwable e) { // TODO
		 * Auto-generated catch block e.printStackTrace(); }
		 */
	}

	@Override
	public void run() {
		try {
			UserGroupSyncConfig config = UserGroupSyncConfig.getInstance();
			long sleepTimeBetweenCycleInMillis = config.getUserSyncMetricsFrequency();
			String logFileNameWithPath = config.getUserSyncMetricsFileName();
			LOG.info("user sync metrics frequency :  " + sleepTimeBetweenCycleInMillis + " and metrics file : "
					+ logFileNameWithPath);
			if (null != logFileNameWithPath) {
				while (!shutdownFlag) {
					try {
						if (LOG.isDebugEnabled()) {
							LOG.debug("Sleeping metrics for [" + sleepTimeBetweenCycleInMillis + "] milliSeconds");
						}
						Thread.sleep(sleepTimeBetweenCycleInMillis);
					} catch (InterruptedException e) {
						LOG.error("Failed to wait for [" + sleepTimeBetweenCycleInMillis
								+ "] milliseconds before attempting to fetch userSync metrics information", e);
					}
					try {
						writeJVMMetrics(logFileNameWithPath);
					} catch (Throwable t) {
						LOG.error("Failed to write user sync metrics into file. Error details: ", t);
					}
				}
			} else {
				LOG.info("No file directory found for usersync metrics log ");
			}
		} catch (Throwable t) {
			LOG.error("Failed to start user sync metrics. Error details: ", t);
		} finally {
			LOG.info("Shutting down the User Sync metrics producer thread");
		}

	}

	private void writeJVMMetrics(String logFileNameWithPath) throws Throwable {
		try {
			File userMetricFile = null;
			userMetricFile = new File(logFileNameWithPath);
			if (!userMetricFile.exists()) {
				userMetricFile.createNewFile();
			}
			RangerMetricsUtil rangerMetricsUtil = new RangerMetricsUtil();
			rangerMetricsUtil.writeMetricsToFile(userMetricFile);
		} catch (Throwable t) {
			LOG.error("UserSyncMetricsProducer.writeJVMMetrics() failed to write metrics into file. Error details: ", t);
			throw t;
		}
	}
}