Java Pattern: Use Atomic Boolean to Return Single Usage Object

  • 时间:2020-09-07 13:13:13
  • 分类:网络文摘
  • 阅读:159 次

Sometimes, you want the data to be accessed only once. For example, when the HTTP data stream is open for reading, once completed, you don’t want to expose the data again. This is to reduce the risks of leaking data.

It turns out we can easily use the following Design Pattern to achieve this – with the help of the Atomic Boolean types in Java. The Atomic data types are thread safe in Java – allowing multiple threads to access and modifying.

We will have a AtomicBoolean field e.g. pending to indicate whether the data is available for access. Once it has been retrieved, we set it to false via compareAndSet.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
package com.helloacm;
 
import java.util.concurrent.atomic.AtomicBoolean;
 
public class Example {
    // is the data available?
    private final AtomicBoolean pending = new AtomicBoolean(true);
 
    // data field
    private final Object data = new Object();
 
    public Object get() {
        // when it is true, return the data and set to false
        if (pending.compareAndSet(true, false)) {
            return data;
        }
        return null;
    }
}
package com.helloacm;

import java.util.concurrent.atomic.AtomicBoolean;

public class Example {
    // is the data available?
    private final AtomicBoolean pending = new AtomicBoolean(true);

    // data field
    private final Object data = new Object();

    public Object get() {
        // when it is true, return the data and set to false
        if (pending.compareAndSet(true, false)) {
            return data;
        }
        return null;
    }
}

Let’s unit test the above pattern:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
package com.helloacm;
 
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
 
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;
 
public class ExampleTest {
    // test instance
    private Example instance;
 
    @BeforeEach
    public void setup() {
        instance = new Example();
    }
 
    @Test
    public void test_get_twice_should_return_null() {
        assertNotNull(instance.get());
        assertNull(instance.get());
    }
}
package com.helloacm;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;

import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;

public class ExampleTest {
    // test instance
    private Example instance;

    @BeforeEach
    public void setup() {
        instance = new Example();
    }

    @Test
    public void test_get_twice_should_return_null() {
        assertNotNull(instance.get());
        assertNull(instance.get());
    }
}

The first time, the data should be available for retrieval, then once it is invoked, the get() will return NULL. This allows data to be only accessed once.

–EOF (The Ultimate Computing & Technology Blog) —

推荐阅读:
爷爷——我因你而骄傲|小学作文  再隔多远又有一根电线杆不必移动  小明房间的地面至少是多少平方米  公倍数应用题  5天完成了6%  圆柱体切成6个相等的小圆柱体  压路机的滚筒是一个圆柱  两个关于质数的问题  正方体铁块熔铸成圆锥形问题  甲乙应合作几天 
评论列表
添加评论