ee.jakarta.tck.concurrent.common.context.providers.IntContextSnapshot Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2021, 2023 Contributors to the Eclipse Foundation
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */
package ee.jakarta.tck.concurrent.common.context.providers;

import ee.jakarta.tck.concurrent.common.context.IntContext;
import jakarta.enterprise.concurrent.spi.ThreadContextRestorer;
import jakarta.enterprise.concurrent.spi.ThreadContextSnapshot;

/**
 * Thread context snapshot for a mock context type that consists of an int
 * value.
 */
public class IntContextSnapshot implements ThreadContextSnapshot {
    private final int contextSnapshot;

    IntContextSnapshot(final int i) {
        contextSnapshot = i;
    }

    public ThreadContextRestorer begin() {
        ThreadContextRestorer restorer = new IntContextRestorer(IntContext.get());
        IntContext.set(contextSnapshot);
        return restorer;
    }

    public String toString() {
        return "IntContextSnapshot@" + Integer.toHexString(hashCode()) + "(" + contextSnapshot + ")";
    }
}