Madan Jampani | 04aeb45 | 2015-05-02 16:12:24 -0700 | [diff] [blame] | 1 | package org.onosproject.store.core.impl; |
| 2 | |
Thomas Vachuska | d0d5854 | 2015-06-03 12:38:44 -0700 | [diff] [blame] | 3 | import com.google.common.collect.Maps; |
Madan Jampani | 04aeb45 | 2015-05-02 16:12:24 -0700 | [diff] [blame] | 4 | import org.apache.felix.scr.annotations.Activate; |
| 5 | import org.apache.felix.scr.annotations.Component; |
| 6 | import org.apache.felix.scr.annotations.Deactivate; |
| 7 | import org.apache.felix.scr.annotations.Reference; |
| 8 | import org.apache.felix.scr.annotations.ReferenceCardinality; |
| 9 | import org.apache.felix.scr.annotations.Service; |
| 10 | import org.onosproject.core.IdBlock; |
| 11 | import org.onosproject.core.IdBlockStore; |
| 12 | import org.onosproject.store.service.AtomicCounter; |
Thomas Vachuska | f64c077 | 2015-06-01 10:32:14 -0700 | [diff] [blame] | 13 | import org.onosproject.store.service.StorageException; |
Madan Jampani | 04aeb45 | 2015-05-02 16:12:24 -0700 | [diff] [blame] | 14 | import org.onosproject.store.service.StorageService; |
| 15 | import org.slf4j.Logger; |
| 16 | |
Thomas Vachuska | d0d5854 | 2015-06-03 12:38:44 -0700 | [diff] [blame] | 17 | import java.util.Map; |
| 18 | |
| 19 | import static org.onlab.util.Tools.delay; |
| 20 | import static org.slf4j.LoggerFactory.getLogger; |
Madan Jampani | 04aeb45 | 2015-05-02 16:12:24 -0700 | [diff] [blame] | 21 | |
| 22 | /** |
| 23 | * Implementation of {@code IdBlockStore} using {@code AtomicCounter}. |
| 24 | */ |
| 25 | @Component(immediate = true, enabled = true) |
| 26 | @Service |
| 27 | public class ConsistentIdBlockStore implements IdBlockStore { |
| 28 | |
Thomas Vachuska | f64c077 | 2015-06-01 10:32:14 -0700 | [diff] [blame] | 29 | private static final int MAX_TRIES = 3; |
Thomas Vachuska | d0d5854 | 2015-06-03 12:38:44 -0700 | [diff] [blame] | 30 | private static final int RETRY_DELAY_MS = 2_000; |
Thomas Vachuska | f64c077 | 2015-06-01 10:32:14 -0700 | [diff] [blame] | 31 | |
Madan Jampani | 04aeb45 | 2015-05-02 16:12:24 -0700 | [diff] [blame] | 32 | private final Logger log = getLogger(getClass()); |
| 33 | private final Map<String, AtomicCounter> topicCounters = Maps.newConcurrentMap(); |
| 34 | |
| 35 | private static final long DEFAULT_BLOCK_SIZE = 0x100000L; |
| 36 | |
| 37 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 38 | protected StorageService storageService; |
| 39 | |
| 40 | @Activate |
| 41 | public void activate() { |
| 42 | log.info("Started"); |
| 43 | } |
| 44 | |
| 45 | @Deactivate |
| 46 | public void deactivate() { |
| 47 | log.info("Stopped"); |
| 48 | } |
| 49 | |
| 50 | @Override |
| 51 | public IdBlock getIdBlock(String topic) { |
Thomas Vachuska | f64c077 | 2015-06-01 10:32:14 -0700 | [diff] [blame] | 52 | AtomicCounter counter = topicCounters |
| 53 | .computeIfAbsent(topic, |
| 54 | name -> storageService.atomicCounterBuilder() |
| 55 | .withName(name) |
| 56 | .build()); |
| 57 | Throwable exc = null; |
| 58 | for (int i = 0; i < MAX_TRIES; i++) { |
| 59 | try { |
| 60 | Long blockBase = counter.getAndAdd(DEFAULT_BLOCK_SIZE); |
| 61 | return new IdBlock(blockBase, DEFAULT_BLOCK_SIZE); |
| 62 | } catch (StorageException e) { |
| 63 | log.warn("Unable to allocate ID block due to {}; retrying...", |
| 64 | e.getMessage()); |
| 65 | exc = e; |
Thomas Vachuska | d0d5854 | 2015-06-03 12:38:44 -0700 | [diff] [blame] | 66 | delay(RETRY_DELAY_MS); // FIXME: This is a deliberate hack; fix in Drake |
Thomas Vachuska | f64c077 | 2015-06-01 10:32:14 -0700 | [diff] [blame] | 67 | } |
| 68 | } |
| 69 | throw new IllegalStateException("Unable to allocate ID block", exc); |
Madan Jampani | 04aeb45 | 2015-05-02 16:12:24 -0700 | [diff] [blame] | 70 | } |
Thomas Vachuska | f64c077 | 2015-06-01 10:32:14 -0700 | [diff] [blame] | 71 | |
Madan Jampani | 04aeb45 | 2015-05-02 16:12:24 -0700 | [diff] [blame] | 72 | } |