Ayaka Koshibe | 3a32156 | 2015-04-29 13:24:07 -0700 | [diff] [blame] | 1 | package org.onosproject.store.consistent.impl; |
| 2 | |
| 3 | import org.onosproject.store.service.TransactionContext; |
| 4 | import org.onosproject.store.service.TransactionContextBuilder; |
| 5 | |
| 6 | /** |
| 7 | * The default implementation of a transaction context builder. This builder |
| 8 | * generates a {@link DefaultTransactionContext}. |
| 9 | */ |
| 10 | public class DefaultTransactionContextBuilder implements TransactionContextBuilder { |
| 11 | |
| 12 | private boolean partitionsEnabled = true; |
Madan Jampani | 50589ac | 2015-06-08 11:38:46 -0700 | [diff] [blame] | 13 | private final DatabaseManager manager; |
Ayaka Koshibe | 3a32156 | 2015-04-29 13:24:07 -0700 | [diff] [blame] | 14 | private final long transactionId; |
| 15 | |
Madan Jampani | 50589ac | 2015-06-08 11:38:46 -0700 | [diff] [blame] | 16 | public DefaultTransactionContextBuilder(DatabaseManager manager, long transactionId) { |
| 17 | this.manager = manager; |
Ayaka Koshibe | 3a32156 | 2015-04-29 13:24:07 -0700 | [diff] [blame] | 18 | this.transactionId = transactionId; |
| 19 | } |
| 20 | |
| 21 | @Override |
| 22 | public TransactionContextBuilder withPartitionsDisabled() { |
| 23 | partitionsEnabled = false; |
| 24 | return this; |
| 25 | } |
| 26 | |
| 27 | @Override |
| 28 | public TransactionContext build() { |
| 29 | return new DefaultTransactionContext( |
Madan Jampani | 50589ac | 2015-06-08 11:38:46 -0700 | [diff] [blame] | 30 | transactionId, |
| 31 | partitionsEnabled ? manager.partitionedDatabase : manager.inMemoryDatabase, |
| 32 | () -> partitionsEnabled ? manager.consistentMapBuilder() |
| 33 | : manager.consistentMapBuilder().withPartitionsDisabled()); |
Ayaka Koshibe | 3a32156 | 2015-04-29 13:24:07 -0700 | [diff] [blame] | 34 | } |
Ayaka Koshibe | 3a32156 | 2015-04-29 13:24:07 -0700 | [diff] [blame] | 35 | } |