-
Notifications
You must be signed in to change notification settings - Fork 39
/
event_swap_out.go
123 lines (109 loc) · 3.29 KB
/
event_swap_out.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package bsdb
import (
"errors"
"fmt"
"time"
"gorm.io/gorm"
)
// ListSwapOutEvents list swap out events
func (b *BsDBImpl) ListSwapOutEvents(blockID uint64, spID uint32) ([]*EventSwapOut, []*EventCompleteSwapOut, []*EventCancelSwapOut, error) {
var (
events []*EventSwapOut
completeEvents []*EventCompleteSwapOut
cancelEvents []*EventCancelSwapOut
err error
)
startTime := time.Now()
methodName := currentFunction()
defer func() {
if err != nil {
MetadataDatabaseFailureMetrics(err, startTime, methodName)
} else {
MetadataDatabaseSuccessMetrics(startTime, methodName)
}
}()
err = b.db.Table((&EventSwapOut{}).TableName()).
Select("*").
Where("storage_provider_id = ? and create_at <= ?", spID, blockID).
Find(&events).Error
if err != nil {
return nil, nil, nil, err
}
err = b.db.Table((&EventCompleteSwapOut{}).TableName()).
Select("*").
Where("src_storage_provider_id = ? and create_at <= ?", spID, blockID).
Find(&completeEvents).Error
if err != nil {
return nil, nil, nil, err
}
err = b.db.Table((&EventCancelSwapOut{}).TableName()).
Select("*").
Where("storage_provider_id = ? and create_at <= ?", spID, blockID).
Find(&cancelEvents).Error
if err != nil {
return nil, nil, nil, err
}
return events, completeEvents, cancelEvents, err
}
func CreateSwapOutIdx(vgfID uint32, spID uint32, gvgIDS []uint32) string {
var (
idx string
ids string
)
// StorageProviderId and GlobalVirtualGroupFamilyId ensure event continuity when GlobalVirtualGroupFamilyId != 0.
// If it's 0, 'StorageProviderId' with GlobalVirtualGroupIds serve the same purpose.
if vgfID == 0 {
for j, id := range gvgIDS {
if j != 0 {
ids += "+"
}
ids += fmt.Sprintf("%d", id)
}
idx = fmt.Sprintf("%d+%s", spID, ids)
} else {
idx = fmt.Sprintf("%d+%d", spID, vgfID)
}
return idx
}
// GetEventSwapOutByGvgID get swap out event by gvg id
func (b *BsDBImpl) GetEventSwapOutByGvgID(gvgID uint32) (*EventSwapOut, error) {
var (
event *EventSwapOut
cancelEvent *EventCancelSwapOut
query string
gvg *GlobalVirtualGroup
err error
)
startTime := time.Now()
methodName := currentFunction()
defer func() {
if err != nil {
MetadataDatabaseFailureMetrics(err, startTime, methodName)
} else {
MetadataDatabaseSuccessMetrics(startTime, methodName)
}
}()
gvg, err = b.GetGlobalVirtualGroupByGvgID(gvgID)
if err != nil || gvg == nil {
return nil, err
}
query = fmt.Sprintf("select * from event_swap_out where FIND_IN_SET('%d', global_virtual_group_ids) > 0 or global_virtual_group_family_id = %d order by create_at desc;", gvgID, gvg.FamilyId)
err = b.db.Raw(query).Take(&event).Error
if err != nil {
return nil, err
}
query = fmt.Sprintf("select * from event_cancel_swap_out where FIND_IN_SET('%d', global_virtual_group_ids) > 0 or global_virtual_group_family_id = %d order by create_at desc;", gvgID, gvg.FamilyId)
err = b.db.Raw(query).Take(&cancelEvent).Error
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return event, nil
}
return nil, err
}
// check if the latest cancel event create at is larger than swap out
// it means the gvg is not in swap out status
if cancelEvent.CreateAt > event.CreateAt {
return nil, errors.New("the gvg is not in swap out status")
}
return event, nil
}