AssociateDataFileBean.java

package com.intuit.tank.project;

/*
 * #%L
 * JSF Support Beans
 * %%
 * Copyright (C) 2011 - 2015 Intuit Inc.
 * %%
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * #L%
 */

import java.io.Serializable;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import javax.enterprise.context.ConversationScoped;
import javax.enterprise.event.Observes;
import javax.inject.Inject;
import javax.inject.Named;

import org.primefaces.model.DualListModel;

import com.intuit.tank.ModifiedDatafileMessage;
import com.intuit.tank.ProjectBean;
import com.intuit.tank.dao.DataFileDao;
import com.intuit.tank.project.DataFile;
import com.intuit.tank.util.CreateDateComparator;
import com.intuit.tank.util.CreateDateComparator.SortOrder;

@Named
@ConversationScoped
public class AssociateDataFileBean implements Serializable {

    private static final long serialVersionUID = 1L;

    @Inject
    private ProjectBean projectBean;

    private DualListModel<DataFile> selectionModel;

    /**
     * Initializes the instance variables. It also performs database operations to fetch the file items from the
     * database and perform check for the files that are associated with the given project and populate the instance
     * variables accordingly.
     * 
     * @param project
     *            the project value to be set
     * @param workload
     *            the workload value to be set.
     */
    public void init() {
        initScriptSelectionModel();
    }

    /**
     * @return the selectionModel
     */
    public DualListModel<DataFile> getSelectionModel() {
        return selectionModel;
    }

    /**
     * @param selectionModel
     *            the selectionModel to set
     */
    public void setSelectionModel(DualListModel<DataFile> selectionModel) {
        if (selectionModel != null && (!selectionModel.getSource().isEmpty() || !selectionModel.getTarget().isEmpty())) {
            this.selectionModel = selectionModel;
        }
    }

    private void initScriptSelectionModel() {
        selectionModel = new DualListModel<DataFile>();
        List<DataFile> files = new DataFileDao().findAll();
        Collections.sort(files, new CreateDateComparator(SortOrder.DESCENDING));
        Set<Integer> dataFileIds = projectBean.getJobConfiguration().getDataFileIds();
        for (DataFile d : files) {
            if (dataFileIds.contains(d.getId())) {
                selectionModel.getTarget().add(d);
            } else {
                selectionModel.getSource().add(d);
            }
        }
    }

    public void observerUpload(@Observes ModifiedDatafileMessage msg) {
        DataFile df = msg.getModified();
        if (df != null && selectionModel != null) {
            selectionModel.getTarget().add(df);
        }
    }

    /**
     * Persist the job configuration to the database.
     */
    public void save() {
        projectBean.getJobConfiguration().getDataFileIds().clear();
        projectBean.getJobConfiguration().getDataFileIds().addAll(getDataFileIds(selectionModel.getTarget()));
    }

    /**
     * Get the set of ids of the data files of the selected file items.
     * 
     * @param list
     * 
     * @param selectedFiles
     *            selected file items.
     * @param fileNameMap
     *            Map that would return a data file for a given name.
     * @return
     */
    private static Set<Integer> getDataFileIds(List<DataFile> list) {
        Set<Integer> set = new HashSet<Integer>();
        for (DataFile df : list) {
            set.add(df.getId());
        }
        return set;
    }

}