Add UnsignedLongBitmap
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / main / java / org / opendaylight / controller / cluster / datastore / FrontendHistoryMetadataBuilder.java
1 /*
2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.controller.cluster.datastore;
9
10 import static com.google.common.base.Preconditions.checkState;
11 import static java.util.Objects.requireNonNull;
12
13 import com.google.common.primitives.UnsignedLong;
14 import java.util.HashMap;
15 import java.util.Map;
16 import org.eclipse.jdt.annotation.NonNull;
17 import org.opendaylight.controller.cluster.access.concepts.ClientIdentifier;
18 import org.opendaylight.controller.cluster.access.concepts.LocalHistoryIdentifier;
19 import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
20 import org.opendaylight.controller.cluster.datastore.persisted.FrontendHistoryMetadata;
21 import org.opendaylight.controller.cluster.datastore.utils.MutableUnsignedLongSet;
22 import org.opendaylight.controller.cluster.datastore.utils.UnsignedLongBitmap;
23 import org.opendaylight.yangtools.concepts.Builder;
24 import org.opendaylight.yangtools.concepts.Identifiable;
25
26 final class FrontendHistoryMetadataBuilder implements Builder<FrontendHistoryMetadata>,
27         Identifiable<LocalHistoryIdentifier> {
28
29     private final @NonNull Map<UnsignedLong, Boolean> closedTransactions;
30     private final @NonNull MutableUnsignedLongSet purgedTransactions;
31     private final @NonNull LocalHistoryIdentifier identifier;
32
33     private boolean closed;
34
35     FrontendHistoryMetadataBuilder(final LocalHistoryIdentifier identifier) {
36         this.identifier = requireNonNull(identifier);
37         purgedTransactions = MutableUnsignedLongSet.of();
38         closedTransactions = new HashMap<>(2);
39     }
40
41     FrontendHistoryMetadataBuilder(final ClientIdentifier clientId, final FrontendHistoryMetadata meta) {
42         identifier = new LocalHistoryIdentifier(clientId, meta.getHistoryId(), meta.getCookie());
43         closedTransactions = meta.getClosedTransactions().mutableCopy();
44         purgedTransactions = meta.getPurgedTransactions().mutableCopy();
45         closed = meta.isClosed();
46     }
47
48     @Override
49     public LocalHistoryIdentifier getIdentifier() {
50         return identifier;
51     }
52
53     @Override
54     public FrontendHistoryMetadata build() {
55         return new FrontendHistoryMetadata(identifier.getHistoryId(), identifier.getCookie(), closed,
56             UnsignedLongBitmap.copyOf(closedTransactions), purgedTransactions.immutableCopy());
57     }
58
59     void onHistoryClosed() {
60         checkState(identifier.getHistoryId() != 0);
61         closed = true;
62     }
63
64     void onTransactionAborted(final TransactionIdentifier txId) {
65         closedTransactions.put(UnsignedLong.fromLongBits(txId.getTransactionId()), Boolean.FALSE);
66     }
67
68     void onTransactionCommitted(final TransactionIdentifier txId) {
69         closedTransactions.put(UnsignedLong.fromLongBits(txId.getTransactionId()), Boolean.TRUE);
70     }
71
72     void onTransactionPurged(final TransactionIdentifier txId) {
73         final long txidBits = txId.getTransactionId();
74         closedTransactions.remove(UnsignedLong.fromLongBits(txidBits));
75         purgedTransactions.add(txidBits);
76     }
77
78     /**
79      * Transform frontend metadata for a particular client history into its {@link LocalFrontendHistory} counterpart.
80      *
81      * @param shard parent shard
82      * @return Leader history state
83      */
84     @NonNull AbstractFrontendHistory toLeaderState(final @NonNull Shard shard) {
85         if (identifier.getHistoryId() == 0) {
86             return StandaloneFrontendHistory.recreate(shard.persistenceId(), identifier.getClientId(),
87                 shard.getDataStore(), closedTransactions, purgedTransactions);
88         }
89
90         return LocalFrontendHistory.recreate(shard.persistenceId(), shard.getDataStore(),
91             shard.getDataStore().recreateTransactionChain(identifier, closed), closedTransactions, purgedTransactions);
92     }
93 }