/*
 * SoapUI, Copyright (C) 2004-2016 SmartBear Software 
 *
 * Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent 
 * versions of the EUPL (the "Licence"); 
 * You may not use this work except in compliance with the Licence. 
 * You may obtain a copy of the Licence at: 
 * 
 * http://ec.europa.eu/idabc/eupl 
 * 
 * Unless required by applicable law or agreed to in writing, software distributed under the Licence is 
 * distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 
 * express or implied. See the Licence for the specific language governing permissions and limitations 
 * under the Licence. 
 */

package com.eviware.soapui.impl.wsdl.loadtest.data;

import com.eviware.soapui.model.support.LoadTestRunListenerAdapter;
import com.eviware.soapui.model.support.TestSuiteListenerAdapter;
import com.eviware.soapui.model.testsuite.LoadTest;
import com.eviware.soapui.model.testsuite.LoadTestRunContext;
import com.eviware.soapui.model.testsuite.LoadTestRunner;
import com.eviware.soapui.model.testsuite.TestCaseRunContext;
import com.eviware.soapui.model.testsuite.TestCaseRunner;
import com.eviware.soapui.model.testsuite.TestStep;
import com.eviware.soapui.model.testsuite.TestStepResult;
import org.apache.log4j.Logger;

import javax.swing.table.AbstractTableModel;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * Collector of samples from a loadtest, exposed as TableModel
 *
 * @author Ole.Matzura
 */

public class LoadTestSamples extends AbstractTableModel {
    private final LoadTest loadTest;
    private List<List<LoadTestStepSample[]>> samples = new ArrayList<List<LoadTestStepSample[]>>();
    private List<Long> timestamps = new ArrayList<Long>();
    private InternalLoadTestRunListener loadTestRunListener = new InternalLoadTestRunListener();
    private InternalTestSuiteListener testSuiteListener = new InternalTestSuiteListener();
    private final static Logger log = Logger.getLogger(LoadTestSamples.class);

    public LoadTestSamples(LoadTest loadTest) {
        this.loadTest = loadTest;

        loadTest.addLoadTestRunListener(loadTestRunListener);
        loadTest.getTestCase().getTestSuite().addTestSuiteListener(testSuiteListener);
    }

    public int getRowCount() {
        return samples.size();
    }

    public void release() {
        loadTest.removeLoadTestRunListener(loadTestRunListener);
        loadTest.getTestCase().getTestSuite().removeTestSuiteListener(testSuiteListener);
    }

    public int getColumnCount() {
        return loadTest.getTestCase().getTestStepCount() + 1;
    }

    public Object getValueAt(int rowIndex, int columnIndex) {
        return columnIndex == 0 ? new Date(timestamps.get(rowIndex)) : samples.get(rowIndex).get(columnIndex - 1);
    }

    public Class<?> getColumnClass(int columnIndex) {
        return columnIndex == 0 ? Date.class : LoadTestStepSample[].class;
    }

    public String getColumnName(int columnIndex) {
        return columnIndex == 0 ? "Timestamp" : loadTest.getTestCase().getTestStepAt(columnIndex - 1).getName();
    }

    public synchronized void clear() {
        int size = samples.size();
        samples.clear();
        timestamps.clear();
        fireTableRowsDeleted(0, size - 1);
    }

    private final class InternalTestSuiteListener extends TestSuiteListenerAdapter {
        public void loadTestRemoved(LoadTest loadTest) {
            if (loadTest.equals(LoadTestSamples.this.loadTest)) {
                loadTest.removeLoadTestRunListener(loadTestRunListener);
            }
        }

        public void testStepAdded(TestStep testStep, int index) {
            if (testStep.getTestCase() == loadTest.getTestCase()) {
                for (List<LoadTestStepSample[]> values : samples) {
                    values.add(index, new LoadTestStepSample[0]);
                }
            }
        }

        public void testStepMoved(TestStep testStep, int fromIndex, int offset) {
            if (testStep.getTestCase() == loadTest.getTestCase()) {
                for (List<LoadTestStepSample[]> values : samples) {
                    LoadTestStepSample[] s = values.remove(fromIndex);
                    values.add(offset, s);
                }
            }
        }

        public void testStepRemoved(TestStep testStep, int index) {
            if (testStep.getTestCase() == loadTest.getTestCase()) {
                for (List<LoadTestStepSample[]> values : samples) {
                    values.remove(index);
                }
            }
        }
    }

    private class InternalLoadTestRunListener extends LoadTestRunListenerAdapter {
        public void afterTestCase(LoadTestRunner loadTestRunner, LoadTestRunContext context, TestCaseRunner testRunner,
                                  TestCaseRunContext runContext) {
            long timestamp = System.currentTimeMillis();
            List<LoadTestStepSample[]> s = new ArrayList<LoadTestStepSample[]>();
            List<TestStepResult> testResults = testRunner.getResults();

            for (int c = 0; c < loadTest.getTestCase().getTestStepCount(); c++) {
                TestStep testStep = loadTest.getTestCase().getTestStepAt(c);
                List<LoadTestStepSample> results = new ArrayList<LoadTestStepSample>();

                for (int i = 0; i < testResults.size(); i++) {
                    TestStepResult stepResult = testResults.get(i);
                    if (stepResult == null) {
                        log.warn("Result [" + c + "] is null in TestCase [" + testRunner.getTestCase().getName() + "]");
                        continue;
                    }

                    if (stepResult.getTestStep().equals(testStep)) {
                        results.add(new LoadTestStepSample(stepResult));
                    }
                }

                s.add(results.toArray(new LoadTestStepSample[results.size()]));
            }

            synchronized (this) {
                samples.add(s);
                timestamps.add(timestamp);
                fireTableRowsInserted(samples.size() - 1, samples.size() - 1);
            }
        }
    }
}