cubefs

1
// Copyright 2018 The CubeFS Authors.
2
//
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
6
//
7
//     http://www.apache.org/licenses/LICENSE-2.0
8
//
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
12
// implied. See the License for the specific language governing
13
// permissions and limitations under the License.k
14

15
package metanode
16

17
import (
18
	"sync"
19
	"sync/atomic"
20
)
21

22
// TxIDAllocator generates and allocates ids
23
type TxIDAllocator struct {
24
	mpTxID   uint64
25
	txIDLock sync.RWMutex
26
}
27

28
// func newTxIDAllocator(mpID uint64, partition raftstore.Partition) (alloc *TxIDAllocator) {
29
func newTxIDAllocator() (alloc *TxIDAllocator) {
30
	alloc = new(TxIDAllocator)
31
	return
32
}
33

34
func (alloc *TxIDAllocator) Reset() {
35
	atomic.StoreUint64(&alloc.mpTxID, 0)
36
}
37

38
func (alloc *TxIDAllocator) setTransactionID(id uint64) {
39
	atomic.StoreUint64(&alloc.mpTxID, id)
40
}
41

42
func (alloc *TxIDAllocator) getTransactionID() uint64 {
43
	return atomic.LoadUint64(&alloc.mpTxID)
44
}
45

46
func (alloc *TxIDAllocator) allocateTransactionID() (mpTxID uint64) {
47
	alloc.txIDLock.Lock()
48
	defer alloc.txIDLock.Unlock()
49
	mpTxID = atomic.LoadUint64(&alloc.mpTxID) + 1
50
	alloc.setTransactionID(mpTxID)
51
	return
52
}
53