edu.usu.sdl.openstorefront.report.ComponentOrganizationReport.java Source code

Introduction

Here is the source code for edu.usu.sdl.openstorefront.report.ComponentOrganizationReport.java

Source

/*
 * Copyright 2015 Space Dynamics Laboratory - Utah State University Research Foundation.
 *
 * 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 edu.usu.sdl.openstorefront.report;

import com.orientechnologies.orient.core.record.impl.ODocument;
import edu.usu.sdl.openstorefront.common.util.TimeUtil;
import edu.usu.sdl.openstorefront.core.entity.ApprovalStatus;
import edu.usu.sdl.openstorefront.core.entity.Component;
import edu.usu.sdl.openstorefront.core.entity.Report;
import edu.usu.sdl.openstorefront.report.generator.CSVGenerator;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.commons.lang3.StringUtils;

/**
 *
 * @author dshurtleff
 */
public class ComponentOrganizationReport extends BaseReport {

    public ComponentOrganizationReport(Report report) {
        super(report);
    }

    @Override
    protected void gatherData() {
    }

    @Override
    protected void writeReport() {
        CSVGenerator cvsGenerator = (CSVGenerator) generator;

        //write header
        cvsGenerator.addLine("Component Organization Report", sdf.format(TimeUtil.currentDate()));
        cvsGenerator.addLine("Organization", "Component Name",
                //"Security Classification",
                "Last Update Date", "Approve Status");

        List<ODocument> documents = service.getPersistenceService().query(
                "Select DISTINCT(organization) as organization, name, name.toLowerCase() as sortname, securityMarkingType, lastActivityDts, approvalState from "
                        + Component.class.getSimpleName() + " where approvalState='" + ApprovalStatus.APPROVED
                        + "' and " + " activeStatus= '" + Component.ACTIVE_STATUS + "' order by sortname",
                new HashMap<>());

        //group by org
        Map<String, List<ODocument>> orgMap = new HashMap<>();
        documents.forEach(document -> {
            String org = document.field("organization");
            if (StringUtils.isBlank(org)) {
                org = "No Organization Specified";
            }
            if (orgMap.containsKey(org)) {
                orgMap.get(org).add(document);
            } else {
                List<ODocument> records = new ArrayList<>();
                records.add(document);
                orgMap.put(org, records);
            }
        });

        long totalComponents = 0;
        List<String> sortedOrganizations = new ArrayList<>(orgMap.keySet());
        sortedOrganizations.sort(null);

        for (String organization : sortedOrganizations) {
            cvsGenerator.addLine(organization);
            for (ODocument document : orgMap.get(organization)) {

                //String securityMarking = document.field("securityMarkingType");
                cvsGenerator.addLine("", document.field("name"),
                        //securityMarking == null ? "" : "(" + securityMarking + ") - " + TranslateUtil.translate(SecurityMarkingType.class, securityMarking),
                        document.field("lastActivityDts"), document.field("approvalState"));

                totalComponents++;
            }
            cvsGenerator.addLine("Total", orgMap.get(organization).size());
            cvsGenerator.addLine("");
        }
        cvsGenerator.addLine("");
        cvsGenerator.addLine("Report Totals");
        cvsGenerator.addLine("Total Organizations: " + orgMap.keySet().size());
        cvsGenerator.addLine("Total Component: " + totalComponents);
    }

}