org.apache.usergrid.chop.webapp.view.runner.RunnersLayout.java Source code

Introduction

Here is the source code for org.apache.usergrid.chop.webapp.view.runner.RunnersLayout.java

Source

/*
 * 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.usergrid.chop.webapp.view.runner;

import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Map;

import com.vaadin.ui.Accordion;
import com.vaadin.ui.Alignment;
import com.vaadin.ui.Button;
import com.vaadin.ui.VerticalLayout;
import com.vaadin.ui.Table;
import org.apache.commons.lang.StringUtils;

import org.apache.usergrid.chop.api.Module;
import org.apache.usergrid.chop.api.Runner;
import org.apache.usergrid.chop.api.State;
import org.apache.usergrid.chop.api.StatsSnapshot;
import org.apache.usergrid.chop.webapp.dao.ModuleDao;
import org.apache.usergrid.chop.webapp.dao.RunnerDao;
import org.apache.usergrid.chop.webapp.dao.model.RunnerGroup;
import org.apache.usergrid.chop.webapp.service.InjectorFactory;
import org.apache.usergrid.chop.webapp.service.runner.RunnerService;
import org.apache.usergrid.chop.webapp.service.runner.RunnerServiceImpl;

public class RunnersLayout extends VerticalLayout {

    private static final DateFormat DATE_FORMAT = new SimpleDateFormat("HH:mm:ss");

    private final RunnerDao runnerDao = InjectorFactory.getInstance(RunnerDao.class);
    private final ModuleDao moduleDao = InjectorFactory.getInstance(ModuleDao.class);

    // Use RunnerServiceMock for testing
    private final RunnerService runnerService = InjectorFactory.getInstance(RunnerServiceImpl.class);

    private final Accordion accordion = new Accordion();

    public RunnersLayout() {
        addItems();
        loadData();
    }

    private void addItems() {
        addAccordion();
        addRefreshButton();
    }

    private void addRefreshButton() {

        Button button = new Button("Refresh");
        button.setWidth("100px");

        button.addClickListener(new Button.ClickListener() {
            public void buttonClick(Button.ClickEvent event) {
                loadData();
            }
        });

        addComponent(button);
        this.setComponentAlignment(button, Alignment.BOTTOM_CENTER);
    }

    private void addAccordion() {
        accordion.setWidth("800px");
        accordion.setHeight("530px");
        addComponent(accordion);
        this.setComponentAlignment(accordion, Alignment.MIDDLE_CENTER);
    }

    private void loadData() {

        accordion.removeAllComponents();
        Map<RunnerGroup, List<Runner>> runnerGroups = runnerDao.getRunnersGrouped();

        for (RunnerGroup group : runnerGroups.keySet()) {
            Table table = getRunnersTable(runnerGroups.get(group));
            addGroup(group, table);
        }
    }

    private Table getRunnersTable(List<Runner> runners) {

        Table table = getTable();

        for (Runner runner : runners) {
            addRunnerToTable(table, runner);
        }

        return table;
    }

    private void addRunnerToTable(Table table, Runner runner) {

        State state = runnerService.getState(runner);
        StatsSnapshot stats = state == State.RUNNING ? runnerService.getStats(runner) : null;

        String percentageComplete = stats != null ? stats.getPercentageComplete() + "%" : "";
        String startTime = stats != null ? DATE_FORMAT.format(new Date(stats.getStartTime())) : "";

        Object[] cells = new Object[] { runner.getUrl(), state.toString(), percentageComplete, startTime };

        table.addItem(cells, runner.getUrl());
    }

    private void addGroup(RunnerGroup group, Table runnersTable) {

        Module module = moduleDao.get(group.getModuleId());

        if (module == null) {
            return;
        }

        String caption = String.format("%s / %s / %s: commit[%s], user[%s]", module.getGroupId(),
                module.getArtifactId(), module.getVersion(), StringUtils.abbreviate(group.getCommitId(), 10),
                group.getUser());

        accordion.addTab(runnersTable, caption);
    }

    private static Table getTable() {

        Table table = new Table();
        table.setSizeFull();

        table.addContainerProperty("URL", String.class, null);
        table.addContainerProperty("State", String.class, null);
        table.addContainerProperty("Complete %", String.class, null);
        table.addContainerProperty("Start Time", String.class, null);

        return table;
    }

}