InjectionUnitOfWork.java
01 /*
02  * SPDX-License-Identifier: Apache-2.0
03  *
04  * Copyright 2008-2017 the original author or authors.
05  *
06  * Licensed under the Apache License, Version 2.0 (the "License");
07  * you may not use this file except in compliance with the License.
08  * You may obtain a copy of the License at
09  *
10  *     http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing, software
13  * distributed under the License is distributed on an "AS IS" BASIS,
14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  * See the License for the specific language governing permissions and
16  * limitations under the License.
17  */
18 package org.codehaus.griffon.runtime.core.injection;
19 
20 import javax.annotation.Nonnull;
21 import java.util.ArrayList;
22 import java.util.List;
23 
24 import static griffon.util.GriffonClassUtils.requireState;
25 import static java.util.Objects.requireNonNull;
26 
27 /**
28  @author Andres Almiray
29  @since 2.6.0
30  */
31 public class InjectionUnitOfWork {
32     private static final ThreadLocal<List<Object>> CONTEXT = new ThreadLocal<>();
33     private static final String ERROR_NO_UNITOFWORK_IN_PROCESS = "There is no InjectionUnitOfWork in process!";
34 
35     private InjectionUnitOfWork() {
36         // prevent instantiation
37     }
38 
39     public static void start() {
40         requireState(CONTEXT.get() == null, "There is already an existing InjectionUnitOfWork in process!");
41         CONTEXT.set(new ArrayList<>());
42     }
43 
44     @Nonnull
45     public static List<Object> finish() {
46         List<Object> instances = CONTEXT.get();
47         requireState(instances != null, ERROR_NO_UNITOFWORK_IN_PROCESS);
48         CONTEXT.set(null);
49         return instances;
50     }
51 
52     public static void track(@Nonnull Object instance) {
53         requireNonNull(instance, "Argument 'instance' must not be null");
54         List<Object> instances = CONTEXT.get();
55         requireState(instances != null, ERROR_NO_UNITOFWORK_IN_PROCESS);
56         instances.add(instance);
57     }
58 }