guru.qas.martini.jmeter.preprocessor.DefaultMartiniSuitePreProcessorBean.java Source code

Introduction

Here is the source code for guru.qas.martini.jmeter.preprocessor.DefaultMartiniSuitePreProcessorBean.java

Source

/*
Copyright 2018 Penny Rohr Curich
    
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 guru.qas.martini.jmeter.preprocessor;

import java.util.concurrent.atomic.AtomicBoolean;

import org.slf4j.cal10n.LocLogger;
import org.slf4j.cal10n.LocLoggerFactory;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Configurable;

import com.google.common.util.concurrent.Monitor;

import ch.qos.cal10n.IMessageConveyor;
import guru.qas.martini.Messages;
import guru.qas.martini.event.SuiteIdentifier;
import guru.qas.martini.runtime.event.EventManager;

import static guru.qas.martini.jmeter.preprocessor.DefaultMartiniSuitePreProcesorBeanMessages.*;

@SuppressWarnings("WeakerAccess")
@Configurable
public class DefaultMartiniSuitePreProcessorBean implements InitializingBean, MartiniSuitePreProcessorBean {

    protected final EventManager eventManager;
    protected final SuiteIdentifier suiteIdentifier;
    protected final AtomicBoolean beforeSuitePublished;
    protected final AtomicBoolean afterSuitePublished;

    protected LocLogger logger;
    protected Monitor monitor;

    @Autowired
    protected DefaultMartiniSuitePreProcessorBean(EventManager eventManager, SuiteIdentifier suiteIdentifier) {
        this.eventManager = eventManager;
        this.suiteIdentifier = suiteIdentifier;
        beforeSuitePublished = new AtomicBoolean(false);
        afterSuitePublished = new AtomicBoolean(false);
    }

    @Override
    public void afterPropertiesSet() {
        setUpLogger();
        setUpMonitor();
    }

    protected void setUpLogger() {
        IMessageConveyor messageConveyor = Messages.getMessageConveyor();
        Class<? extends DefaultMartiniSuitePreProcessorBean> implementation = getClass();
        logger = new LocLoggerFactory(messageConveyor).getLocLogger(implementation);
    }

    protected void setUpMonitor() {
        monitor = new Monitor();
    }

    @Override
    public void publishBeforeSuite() {
        try {
            monitor.enterInterruptibly();
            try {
                if (beforeSuitePublished.compareAndSet(false, true)) {
                    logger.debug(BEFORE_SUITE);
                    eventManager.publishBeforeSuite(this, suiteIdentifier);
                } else {
                    logger.warn(BEFORE_SUITE_ALREADY_PUBLISHED);
                }
            } finally {
                monitor.leave();
            }
        } catch (InterruptedException e) {
            logger.warn(INTERRUPTED, e);
        }
    }

    @Override
    public void publishAfterSuite() {
        try {
            monitor.enterInterruptibly();
            try {
                if (!beforeSuitePublished.get()) {
                    logger.warn(BEFORE_SUITE_NOT_PUBLISHED);
                } else if (afterSuitePublished.compareAndSet(false, true)) {
                    logger.info(AFTER_SUITE);
                    eventManager.publishAfterSuite(this, suiteIdentifier);
                }
            } finally {
                monitor.leave();
            }
        } catch (InterruptedException e) {
            logger.warn(INTERRUPTED, e);
        }
    }

    @Override
    public void destroy() {
        publishAfterSuite();
    }
}