com.centurylink.mdw.timer.startup.UserGroupMonitor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mdw-services Show documentation
MDW is a workflow framework specializing in microservice orchestration
There is a newer version: 6.1.39
/*
 * Copyright (C) 2017 CenturyLink, Inc.
 *
 * 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.
 */
package com.centurylink.mdw.timer.startup;

import java.sql.PreparedStatement;
import java.sql.ResultSet;

import com.centurylink.mdw.config.PropertyManager;
import com.centurylink.mdw.constant.PropertyNames;
import com.centurylink.mdw.dataaccess.DbAccess;
import com.centurylink.mdw.service.data.user.UserGroupCache;
import com.centurylink.mdw.services.cache.CacheRegistration;
import com.centurylink.mdw.startup.StartupException;
import com.centurylink.mdw.startup.StartupService;
import com.centurylink.mdw.util.log.LoggerUtil;
import com.centurylink.mdw.util.log.StandardLogger;

/**
 * Checks for asset imports performed on other instances
 */
public class UserGroupMonitor implements StartupService {

    private static StandardLogger logger = LoggerUtil.getStandardLogger();

    private static boolean _terminating;
    private static UserGroupMonitor monitor = null;
    private static Thread thread = null;

    /**
     * Invoked when the server starts up.
     */
    public void onStartup() throws StartupException {
        if (monitor == null) {
            monitor = this;
            thread = new Thread() {
                @Override
                public void run() {
                    this.setName("UserGroupMonitor-thread");
                    monitor.start();
                }
            };
            thread.start();
        }
    }

    public void onShutdown() {
        _terminating = true;
        thread.interrupt();
    }

    public void start() {
        try {
            Long interval = PropertyManager.getLongProperty(PropertyNames.MDW_USERGROUP_MONITOR_INTERVAL, 120000); //Defaults to checking every 120 seconds
            _terminating = false;

            while (!_terminating) {
                try {
                    Thread.sleep(interval);

                    // Check if it needs to trigger a UserGroup cache refresh
                    String select = "select mod_dt from VALUE where name= ? and owner_type= ? and owner_id= ?";
                    try (DbAccess dbAccess = new DbAccess(); PreparedStatement stmt = dbAccess.getConnection().prepareStatement(select)) {
                        stmt.setString(1, "LastUserGroupChange");
                        stmt.setString(2, "UserGroupAdmin");
                        stmt.setString(3, "0");
                        long latestChange = 0;
                        try (ResultSet rs = stmt.executeQuery()) {
                            if (rs.next()) {
                                latestChange = rs.getTimestamp("mod_dt").getTime();
                            }
                        }
                        // Proceed if user changes have occurred on another instance
                        if (latestChange > 0 && latestChange > UserGroupCache.getLastUserSync()) {
                            logger.info("Detected User/Group changes.  Refreshing...");
                            CacheRegistration.getInstance().refreshCache("UserGroupCache");
                        }
                    }
                }
                catch (InterruptedException e) {
                    if (!_terminating) throw e;
                    logger.info(this.getClass().getName() + " stopping.");
                }
            }
        }
        catch (Exception e) {
            logger.error(e.getMessage(), e);
        }
        finally {
            if (!_terminating) this.start();  // Restart if a failure occurred, besides instance is shutting down
        }
    }
}