io.cucumber.spring.GlueCodeContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cucumber-spring Show documentation

There is a newer version: 7.21.1

package io.cucumber.spring;

import java.util.HashMap;
import java.util.Map;

class GlueCodeContext {

    private static final ThreadLocal localContext = new ThreadLocal() {
        protected GlueCodeContext initialValue() {
            return new GlueCodeContext();
        }
    };

    private final Map objects = new HashMap();
    private final Map callbacks = new HashMap();
    private int counter;

    private GlueCodeContext() {
    }

    public static GlueCodeContext getInstance() {
        return localContext.get();
    }

    public void start() {
        cleanUp();
        counter++;
    }

    public String getId() {
        return "cucumber_glue_" + counter;
    }

    public void stop() {
        for (Runnable callback : callbacks.values()) {
            callback.run();
        }
        cleanUp();
    }

    public Object get(String name) {
        return objects.get(name);
    }

    public void put(String name, Object object) {
        objects.put(name, object);
    }

    public Object remove(String name) {
        callbacks.remove(name);
        return objects.remove(name);
    }

    private void cleanUp() {
        objects.clear();
        callbacks.clear();
    }

    public void registerDestructionCallback(String name, Runnable callback) {
        callbacks.put(name, callback);
    }
}