net.unicon.sakora.impl.csv.CsvCommonHandlerService.java Source code

Java tutorial

Introduction

Here is the source code for net.unicon.sakora.impl.csv.CsvCommonHandlerService.java

Source

/*
 * Licensed to the Sakai Foundation under one or more contributor
 * license agreements. See the NOTICE file distributed with this
 * work for additional information regarding copyright ownership.
 * The Sakai Foundation 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 net.unicon.sakora.impl.csv;

import java.text.DateFormat;
import java.util.Date;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import net.unicon.sakora.api.csv.CsvHandler;
import net.unicon.sakora.api.csv.CsvSyncContext;

import org.apache.commons.lang.ArrayUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import org.sakaiproject.component.api.ServerConfigurationService;
import org.sakaiproject.coursemanagement.api.CourseManagementAdministration;
import org.sakaiproject.coursemanagement.api.CourseManagementService;

/**
 * Handled any common processing that needs to be chared between the CSV handlers
 * during processing (or idle time)
 * 
 * @author Aaron Zeckoski azeckoski@unicon.net
 */
public class CsvCommonHandlerService {
    static final Log log = LogFactory.getLog(CsvCommonHandlerService.class);

    public static final String SYNC_VAR_ID = "id";
    public static final String SYNC_VAR_CONTEXT = "context";
    public static final String SYNC_VAR_STATUS = "status";
    public static final String SYNC_VAR_HANDLER = "handler";
    public static final String SYNC_VAR_HANDLER_STATE = "handler_state";
    public static final String SYNC_VAR_HANDLER_STATS = "handler_stats";
    public static final String SYNC_VAR_STARTDATE = "start_date";
    public static final String SYNC_VAR_SUMMARY = "summary";

    public static final String SYNC_STATE_RUNNING = "running";
    public static final String SYNC_STATE_FAILED = "failed";
    public static final String SYNC_STATE_COMPLETE = "complete";

    public static final String STATE_START = "start";
    public static final String STATE_READ = "read";
    public static final String STATE_PROCESS = "process";
    public static final String STATE_CLEANUP = "cleanup";
    public static final String STATE_FAIL = "fail";
    public static final String STATE_DONE = "done";

    public static final String URM_DELETE = "delete";
    public static final String URM_DISABLE = "disable";
    public static final String URM_IGNORE = "ignore";
    private static final String[] URM_VALIDS = { URM_DISABLE, URM_DELETE, URM_IGNORE };

    private static final String CURRENT_ENROLLMENT_SET_EIDS = "currentEnrollmentSetEids";
    private static final String CURRENT_SECTION_EIDS = "currentSectionEids";
    private static final String CURRENT_COURSE_OFFERING_EIDS = "currentCourseOfferingEids";
    private static final String CURRENT_SESSION_EIDS = "currentSessionEids";

    private static final String IGNORE_MEMBERSHIP_REMOVALS = "ignoreMembershipRemovals";
    private static final String IGNORE_MISSING_SESSIONS = "ignoreMissingSessions";
    private static final String USER_REMOVAL_MODE = "userRemoveMode";

    protected ServerConfigurationService configurationService;
    protected CourseManagementAdministration cmAdmin;
    protected CourseManagementService cmService;

    private volatile int runCounter = 0;
    private ConcurrentHashMap<String, Object> syncVars = new ConcurrentHashMap<String, Object>();

    public void init() {
        // load common config values
        ignoreMissingSessions = configurationService.getBoolean("net.unicon.sakora.csv.ignoreMissingSessions",
                ignoreMissingSessions);
        if (ignoreMissingSessions()) {
            log.info(
                    "SakoraCSV ignoreMissingSessions is enabled: all data related to sessions not included in sessions.csv will be left skipped or otherwise unchanged");
        } else {
            log.info(
                    "SakoraCSV set to process missing sessions (ignoreMissingSessions=false): all data related to sessions not included in sessions.csv will be processed and REMOVED");
        }
        ignoreMembershipRemovals = configurationService.getBoolean("net.unicon.sakora.csv.ignoreMembershipRemovals",
                ignoreMembershipRemovals);
        if (ignoreMembershipRemovals()) {
            log.info(
                    "SakoraCSV ignoreMembershipRemovals is enabled: all enrollment removal processing will be skipped");
        }
        setUserRemoveMode(configurationService.getString("net.unicon.sakora.csv.userRemovalMode", userRemoveMode));
        log.info("SakoraCSV userRemoveMode is set to " + userRemoveMode);
    }

    public void destroy() {
        syncVars = null;
    }

    /* We won't allow multiple syncs to run so no need for the thread safety -AZ
    private static final ThreadLocal<HashMap<String, Object>> runVars = new ThreadLocal<HashMap<String,Object>>() {
    @Override
    protected HashMap<String, Object> initialValue() {
        return new HashMap<String, Object>();
    }
    };*/

    public synchronized String initRun(CsvSyncContext context) {
        // run this before a run starts
        syncVars.clear(); //runVars.remove(); // reset the run vars
        // setup the run id
        String runId = ++runCounter + ":" + (new Date().getTime() / 1000);
        syncVars.put(SYNC_VAR_ID, runId);
        syncVars.put(SYNC_VAR_STATUS, SYNC_STATE_RUNNING);
        syncVars.put(SYNC_VAR_CONTEXT, context);
        syncVars.put(SYNC_VAR_STARTDATE, new Date());
        // Do some logging
        log.info("SakoraCSV sync run (" + runId + ") starting: " + context);
        // process context overrides
        if (context.getProperties().containsKey(IGNORE_MISSING_SESSIONS)) {
            Boolean ims = Boolean.parseBoolean(context.getProperties().get(IGNORE_MISSING_SESSIONS));
            overrideIgnoreMissingSessions(ims);
            log.info("SakoraCSV sync run (" + runId + ") overriding " + IGNORE_MISSING_SESSIONS + ": " + ims);
        }
        if (context.getProperties().containsKey(IGNORE_MEMBERSHIP_REMOVALS)) {
            Boolean ier = Boolean.parseBoolean(context.getProperties().get(IGNORE_MEMBERSHIP_REMOVALS));
            overrideIgnoreMembershipRemovals(ier);
            log.info("SakoraCSV sync run (" + runId + ") overriding " + IGNORE_MEMBERSHIP_REMOVALS + ": " + ier);
        }
        if (context.getProperties().containsKey(USER_REMOVAL_MODE)) {
            String urm = context.getProperties().get(USER_REMOVAL_MODE);
            overrideUserRemoveMode(urm);
            log.info("SakoraCSV sync run (" + runId + ") overriding " + USER_REMOVAL_MODE + ": " + urm);
        }
        return runId;
    }

    public synchronized void completeRun(boolean success) {
        // run this after a run completes
        String runId = getCurrentSyncRunId();
        Date start = (Date) syncVars.get(SYNC_VAR_STARTDATE);
        log.info("SakoraCSV sync complete (" + (success ? "success" : "FAILED") + ") for run (" + runId
                + ") started on " + DateFormat.getDateTimeInstance().format(start));
        syncVars.put(SYNC_VAR_STATUS, success ? SYNC_STATE_COMPLETE : SYNC_STATE_FAILED);
        if (success) {
            StringBuilder sb = new StringBuilder();
            int total_lines = 0;
            int total_errors = 0;
            int total_adds = 0;
            int total_updates = 0;
            int total_deletes = 0;
            int total_seconds = 0;
            // compile and output the stats data to the logs
            @SuppressWarnings("unchecked")
            Map<String, Map<String, Integer>> stats = getCurrentSyncVar(SYNC_VAR_HANDLER_STATS, Map.class);
            for (Map.Entry<String, Map<String, Integer>> entry : stats.entrySet()) {
                String handler = entry.getKey();
                Map<String, Integer> handlerStats = entry.getValue();
                int lines = handlerStats.get("lines");
                int errors = handlerStats.get("errors");
                int adds = handlerStats.get("adds");
                int updates = handlerStats.get("updates");
                int deletes = handlerStats.get("deletes");
                int seconds = handlerStats.get("seconds");
                total_lines += lines;
                total_errors += errors;
                total_adds += adds;
                total_updates += updates;
                total_deletes += deletes;
                total_seconds += seconds;
                sb.append("  - ");
                sb.append(StringUtils.rightPad(handler, 20));
                sb.append(": processed ");
                sb.append(String.format("%6d", lines));
                sb.append(" lines with ");
                sb.append(String.format("%4d", errors));
                sb.append(" errors in ");
                sb.append(String.format("%4d", seconds));
                sb.append(" seconds: ");
                sb.append(String.format("%4d", adds));
                sb.append(" adds, ");
                sb.append(String.format("%4d", updates));
                sb.append(" updates, ");
                sb.append(String.format("%4d", deletes));
                sb.append(" deletes\n");
            }
            // total summary (start, end, totals)
            sb.append("  --- TOTAL:         processed ");
            sb.append(String.format("%6d", total_lines));
            sb.append(" lines with ");
            sb.append(String.format("%5d", total_errors));
            sb.append(" errors in ");
            sb.append(String.format("%5d", total_seconds));
            sb.append(" seconds: ");
            sb.append(String.format("%5d", total_adds));
            sb.append(" adds, ");
            sb.append(String.format("%5d", total_updates));
            sb.append(" updates, ");
            sb.append(String.format("%5d", total_deletes));
            sb.append(" deletes\n");
            syncVars.put(SYNC_VAR_SUMMARY, sb.toString());
            log.info("SakoraCSV sync statistics for run (" + runId + "):\n" + sb.toString());
        }
    }

    /**
     * Retrieve a variable stored for the current sync processing if it is set
     * 
     * @param name string key
     * @param type the type of the variable (for compiler happiness only, will not be converted)
     * @return the value OR null if it is not set
     */
    @SuppressWarnings("unchecked")
    public <T> T getCurrentSyncVar(String name, Class<T> type) {
        if (syncVars.containsKey(name)) {
            return (T) syncVars.get(name);
        }
        return null;
    }

    /**
     * Add a variable and value to the thread for this sync run,
     * using null value will remove the variable,
     * adding existing will replace
     * 
     * @param name any string name
     * @param value any object, if null then the var is removed
     */
    public void setCurrentSyncVar(String name, Object value) {
        if (value == null) {
            syncVars.remove(name);
        } else {
            syncVars.put(name, value);
        }
    }

    /**
     * @return the current sync run for this thread OR null if there is not one (no sync is running)
     */
    public String getCurrentSyncRunId() {
        return getCurrentSyncVar(SYNC_VAR_ID, String.class);
    }

    /**
     * @return the current sync context for this thread OR null if there is not one (no sync is running)
     */
    public CsvSyncContext getCurrentSyncRunContext() {
        return getCurrentSyncVar(SYNC_VAR_CONTEXT, CsvSyncContext.class);
    }

    public String getCurrentSyncState() {
        String status = getCurrentSyncVar(SYNC_VAR_STATUS, String.class);
        if (status.equals(SYNC_STATE_RUNNING)) {
            String state = getCurrentSyncVar(SYNC_VAR_HANDLER_STATE, String.class);
            CsvHandler handler = getCurrentSyncVar(SYNC_VAR_HANDLER, CsvHandler.class);
            if (handler != null) {
                String handlerName = handler.getName();
                status = "Sync (" + getCurrentSyncRunId() + "): " + handlerName + " state is: " + state;
            }
        }
        return status;
    }

    public void setCurrentHandlerState(String state, CsvHandler handler) {
        /* Allows us to take actions when the state changes (like logging for example)
         */
        setCurrentSyncVar(SYNC_VAR_HANDLER_STATE, state);
        setCurrentSyncVar(SYNC_VAR_HANDLER, handler);
        String handlerName = handler.getName();
        log.info("SakoraCSV: Sync (" + getCurrentSyncRunId() + "): " + handlerName + " state is: " + state);
        if (STATE_DONE.equals(state)) {
            // store the stats in the overall set
            @SuppressWarnings("unchecked")
            Map<String, Map<String, Integer>> stats = getCurrentSyncVar(SYNC_VAR_HANDLER_STATS, Map.class);
            if (stats == null) {
                stats = new LinkedHashMap<String, Map<String, Integer>>(); // want to maintain the ordering
            }
            stats.put(handlerName, handler.getStats());
            setCurrentSyncVar(SYNC_VAR_HANDLER_STATS, stats);
        }
    }

    /**
     * REMOVAL MODE handling
     * If false (DEFAULT), remove all memberships which are missing from current feed.
     * If true, no memberships removals are processed for feed (all removal processing is skipped)
     * 
     * controlled by net.unicon.sakora.csv.ignoreMembershipRemovals, Default: false
     */
    protected boolean ignoreMembershipRemovals = false;

    public void setIgnoreMembershipRemovals(boolean ignoreMembershipRemovals) {
        this.ignoreMembershipRemovals = ignoreMembershipRemovals;
    }

    public boolean isIgnoreMembershipRemovals() {
        return ignoreMembershipRemovals;
    }

    /**
     * Allows the current setting to be overridden for the current sync run only
     * @param ier null clears the override, see {@link #ignoreMembershipRemovals}
     */
    public void overrideIgnoreMembershipRemovals(Boolean ier) {
        setCurrentSyncVar(IGNORE_MEMBERSHIP_REMOVALS, ier);
        if (ier != null) {
            log.info("Overriding the ignoreMembershipRemovals value of " + ignoreMembershipRemovals + " with "
                    + ier.booleanValue() + " for current sync: " + getCurrentSyncRunId());
        }
    }

    public boolean ignoreMembershipRemovals() {
        Boolean ier = getCurrentSyncVar(IGNORE_MEMBERSHIP_REMOVALS, Boolean.class);
        if (ier != null) {
            // override from the current run
            return ier.booleanValue();
        }
        return ignoreMembershipRemovals;
    }

    /**
     * USER REMOVAL handling
     * disable: (DEFAULT): assign the user a type which matches the "suspended" key in the PersonHandler (defaults to "suspended")
     * delete: removes the user from the system
     * ignore: skips over the user removal processing entirely
     */
    protected String userRemoveMode = "disable";

    public void setUserRemoveMode(String userRemoveMode) {
        if (ArrayUtils.contains(URM_VALIDS, userRemoveMode)) {
            this.userRemoveMode = userRemoveMode;
        } else {
            this.userRemoveMode = URM_DISABLE;
            log.warn("SakoraCSV userRemoveMode (" + userRemoveMode
                    + ") is invalid: resetting to default (disable), must match one of these: "
                    + ArrayUtils.toString(URM_VALIDS));
        }
    }

    public String getUserRemoveMode() {
        return userRemoveMode;
    }

    /**
     * Allows the current setting to be overridden for the current sync run only
     * @param urm null clears the override, see {@link #userRemoveMode}
     */
    public void overrideUserRemoveMode(String urm) {
        if (ArrayUtils.contains(URM_VALIDS, userRemoveMode) || urm == null) {
            setCurrentSyncVar(USER_REMOVAL_MODE, urm);
            if (urm != null) {
                log.info("Overriding the " + USER_REMOVAL_MODE + " value of " + userRemoveMode + " with " + urm
                        + " for current sync: " + getCurrentSyncRunId());
            }
        } else {
            log.warn("SakoraCSV userRemoveMode override (" + urm
                    + ") is invalid: ignoring the override, must match one of these: "
                    + ArrayUtils.toString(URM_VALIDS));
        }
    }

    public String userRemoveMode() {
        String urm = getCurrentSyncVar(USER_REMOVAL_MODE, String.class);
        if (urm != null) {
            // override from the current run
            return urm;
        }
        return userRemoveMode;
    }

    /**
     * Academic Session Skip handling:
     * If the flag is set then all data processing for anything not in the current (i.e. included in the feed) academic sessions
     * will be skipped. This includes the sessions themselves (they will not be disabled or removed or flagged), 
     * the course offerings, the enrollment sets, and the enrollments.
     * 
     * controlled by net.unicon.sakora.csv.ignoreMissingSessions config, Default: false
     */
    protected boolean ignoreMissingSessions = false;

    public void setIgnoreMissingSessions(boolean ignoreMissingSessions) {
        this.ignoreMissingSessions = ignoreMissingSessions;
    }

    public boolean isIgnoreMissingSessions() {
        return ignoreMissingSessions;
    }

    /**
     * Allows the current setting to be overridden for the current sync run only
     * @param ims null clears the override, see {@link #ignoreMissingSessions}
     */
    public void overrideIgnoreMissingSessions(Boolean ims) {
        setCurrentSyncVar(IGNORE_MISSING_SESSIONS, ims);
        if (ims != null) {
            log.info("Overriding the ignoreMissingSessions value of " + ignoreMissingSessions + " with "
                    + ims.booleanValue() + " for current sync: " + getCurrentSyncRunId());
        }
    }

    public boolean ignoreMissingSessions() {
        Boolean ims = getCurrentSyncVar(IGNORE_MISSING_SESSIONS, Boolean.class);
        if (ims != null) {
            // override from the current run
            return ims.booleanValue();
        }
        return ignoreMissingSessions;
    }

    // ACADEMIC SESSIONS
    protected int setCurrentAcademicSessions(String[] sessions) {
        HashSet<String> currentAcademicSessionEids;
        if (sessions != null) {
            currentAcademicSessionEids = new HashSet<String>(sessions.length);
            for (int i = 0; i < sessions.length; i++) {
                currentAcademicSessionEids.add(sessions[i]);
            }
        } else {
            currentAcademicSessionEids = new HashSet<String>(0);
        }
        if (currentAcademicSessionEids.isEmpty()) {
            log.warn(
                    "SakoraCSV has no current academic sessions, this is typically not a valid state, please check your sessions.csv file");
        }
        setCurrentSyncVar(CURRENT_SESSION_EIDS, currentAcademicSessionEids);
        return currentAcademicSessionEids.size();
    }

    protected boolean processAcademicSession(String academicSessionEid) {
        boolean process;
        if (ignoreMissingSessions()) {
            // check the list of sessions which are current and if this is not in that set then false
            @SuppressWarnings("unchecked")
            Set<String> currentAcademicSessionEids = (Set<String>) getCurrentSyncVar(CURRENT_SESSION_EIDS,
                    Set.class);
            if (currentAcademicSessionEids != null && currentAcademicSessionEids.contains(academicSessionEid)) {
                process = true;
            } else {
                process = false;
            }
        } else {
            // standard processing, process all sessions
            process = true;
        }
        return process;
    }

    protected Set<String> getCurrentAcademicSessionEids() {
        @SuppressWarnings("unchecked")
        Set<String> currentSessionEids = (Set<String>) getCurrentSyncVar(CURRENT_SESSION_EIDS, Set.class);
        if (currentSessionEids == null) {
            currentSessionEids = new HashSet<String>(0);
        }
        return currentSessionEids;
    }

    // COURSE OFFERINGS
    protected int addCurrentCourseOffering(String courseOfferingEid) {
        @SuppressWarnings("unchecked")
        Set<String> currentCourseOfferingEids = (Set<String>) getCurrentSyncVar(CURRENT_COURSE_OFFERING_EIDS,
                Set.class);
        if (currentCourseOfferingEids == null) {
            currentCourseOfferingEids = new HashSet<String>();
            setCurrentSyncVar(CURRENT_COURSE_OFFERING_EIDS, currentCourseOfferingEids);
        }
        if (courseOfferingEid != null) {
            currentCourseOfferingEids.add(courseOfferingEid);
        }
        return currentCourseOfferingEids.size();
    }

    protected boolean processCourseOffering(String courseOfferingEid) {
        boolean process;
        if (ignoreMissingSessions()) {
            // check the list of offerings which are current and if course offering is not in that then skip it
            @SuppressWarnings("unchecked")
            Set<String> currentCourseOfferingEids = (Set<String>) getCurrentSyncVar(CURRENT_COURSE_OFFERING_EIDS,
                    Set.class);
            if (currentCourseOfferingEids != null && currentCourseOfferingEids.contains(courseOfferingEid)) {
                process = true;
            } else {
                process = false;
            }
        } else {
            // standard processing, process all course offerings
            process = true;
        }
        return process;
    }

    protected Set<String> getCurrentCourseOfferingEids() {
        @SuppressWarnings("unchecked")
        Set<String> currentCourseOfferingEids = (Set<String>) getCurrentSyncVar(CURRENT_COURSE_OFFERING_EIDS,
                Set.class);
        if (currentCourseOfferingEids == null) {
            currentCourseOfferingEids = new HashSet<String>(0);
        }
        return currentCourseOfferingEids;
    }

    // SECTIONS
    protected int addCurrentSection(String sectionEid) {
        @SuppressWarnings("unchecked")
        Set<String> currentSectionEids = (Set<String>) getCurrentSyncVar(CURRENT_SECTION_EIDS, Set.class);
        if (currentSectionEids == null) {
            currentSectionEids = new HashSet<String>();
            setCurrentSyncVar(CURRENT_SECTION_EIDS, currentSectionEids);
        }
        if (sectionEid != null) {
            currentSectionEids.add(sectionEid);
        }
        return currentSectionEids.size();
    }

    protected boolean processSection(String sectionEid) {
        boolean process;
        if (ignoreMissingSessions()) {
            // check the list of sections which are current and if section is not in that then skip it
            @SuppressWarnings("unchecked")
            Set<String> currentSectionEids = (Set<String>) getCurrentSyncVar(CURRENT_SECTION_EIDS, Set.class);
            if (currentSectionEids != null && currentSectionEids.contains(sectionEid)) {
                process = true;
            } else {
                process = false;
            }
        } else {
            // standard processing, process all course offerings
            process = true;
        }
        return process;
    }

    protected Set<String> getCurrentSectionEids() {
        @SuppressWarnings("unchecked")
        Set<String> currentSectionEids = (Set<String>) getCurrentSyncVar(CURRENT_SECTION_EIDS, Set.class);
        if (currentSectionEids == null) {
            currentSectionEids = new HashSet<String>(0);
        }
        return currentSectionEids;
    }

    // ENROLLMENT SETS
    protected int addCurrentEnrollmentSet(String enrollmentSetEid) {
        @SuppressWarnings("unchecked")
        Set<String> currentEnrollmentSetEids = (Set<String>) getCurrentSyncVar(CURRENT_ENROLLMENT_SET_EIDS,
                Set.class);
        if (currentEnrollmentSetEids == null) {
            currentEnrollmentSetEids = new HashSet<String>();
            setCurrentSyncVar(CURRENT_ENROLLMENT_SET_EIDS, currentEnrollmentSetEids);
        }
        if (enrollmentSetEid != null) {
            currentEnrollmentSetEids.add(enrollmentSetEid);
        }
        return currentEnrollmentSetEids.size();
    }

    protected boolean processEnrollmentSet(String enrollmentSetEid) {
        boolean process;
        if (ignoreMissingSessions()) {
            // check the list of offerings which are current and if course offering is not in that then skip it
            @SuppressWarnings("unchecked")
            Set<String> currentEnrollmentSetEids = (Set<String>) getCurrentSyncVar(CURRENT_ENROLLMENT_SET_EIDS,
                    Set.class);
            if (currentEnrollmentSetEids != null && currentEnrollmentSetEids.contains(enrollmentSetEid)) {
                process = true;
            } else {
                process = false;
            }
        } else {
            // standard processing, process all course offerings
            process = true;
        }
        return process;
    }

    protected Set<String> getCurrentEnrollmentSets() {
        @SuppressWarnings("unchecked")
        Set<String> currentEnrollmentSetEids = (Set<String>) getCurrentSyncVar(CURRENT_ENROLLMENT_SET_EIDS,
                Set.class);
        if (currentEnrollmentSetEids == null) {
            currentEnrollmentSetEids = new HashSet<String>(0);
        }
        return currentEnrollmentSetEids;
    }

    public void setConfigurationService(ServerConfigurationService configurationService) {
        this.configurationService = configurationService;
    }

    public void setCmAdmin(CourseManagementAdministration cmAdmin) {
        this.cmAdmin = cmAdmin;
    }

    public void setCmService(CourseManagementService cmService) {
        this.cmService = cmService;
    }

}