This repository has been archived by the owner on Nov 5, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
builtin.go
108 lines (98 loc) · 2.57 KB
/
builtin.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
package edge
import (
"github.com/dgraph-io/badger/v3"
"github.com/ronaksoft/rony"
"github.com/ronaksoft/rony/internal/cluster"
"github.com/ronaksoft/rony/internal/gateway"
"github.com/ronaksoft/rony/store"
"google.golang.org/protobuf/proto"
)
/*
Creation Time: 2021 - Jan - 11
Created by: (ehsan)
Maintainers:
1. Ehsan N. Moosa (E2)
Auditor: Ehsan N. Moosa (E2)
Copyright Ronak Software Group 2020
*/
// Builtin keep track of pages distribution over Edge servers.
type Builtin struct {
cluster cluster.Cluster
gateway gateway.Gateway
serverID string
rs uint64
}
func newBuiltin(serverID string, gw gateway.Gateway, c cluster.Cluster) *Builtin {
return &Builtin{
cluster: c,
gateway: gw,
serverID: serverID,
}
}
func (pm *Builtin) GetNodes(ctx *RequestCtx, in *rony.MessageEnvelope) {
req := rony.PoolGetNodes.Get()
defer rony.PoolGetNodes.Put(req)
res := rony.PoolEdges.Get()
defer rony.PoolEdges.Put(res)
err := req.Unmarshal(in.Message)
if err != nil {
ctx.PushError(rony.ErrCodeInvalid, rony.ErrItemRequest)
return
}
if pm.cluster == nil {
res.Nodes = append(res.Nodes, &rony.Edge{
ReplicaSet: 0,
ServerID: pm.serverID,
HostPorts: pm.gateway.Addr(),
Leader: true,
})
} else if len(req.ReplicaSet) == 0 {
members := pm.cluster.RaftMembers(pm.cluster.ReplicaSet())
for _, m := range members {
res.Nodes = append(res.Nodes, m.Proto(nil))
}
} else {
for _, rs := range req.ReplicaSet {
members := pm.cluster.RaftMembers(rs)
for _, m := range members {
res.Nodes = append(res.Nodes, m.Proto(nil))
}
}
}
ctx.PushMessage(rony.C_Edges, res)
return
}
func (pm *Builtin) GetPage(ctx *RequestCtx, in *rony.MessageEnvelope) {
if pm.cluster.ReplicaSet() != 1 {
ctx.PushError(rony.ErrCodeUnavailable, rony.ErrItemRequest)
return
}
req := rony.PoolGetPage.Get()
defer rony.PoolGetPage.Put(req)
res := rony.PoolPage.Get()
defer rony.PoolPage.Put(res)
err := proto.UnmarshalOptions{Merge: true}.Unmarshal(in.Message, req)
if err != nil {
ctx.PushError(rony.ErrCodeInvalid, rony.ErrItemRequest)
return
}
alloc := store.NewAllocator()
defer alloc.ReleaseAll()
err = store.Update(func(txn *badger.Txn) (err error) {
_, err = rony.ReadPageWithTxn(txn, alloc, req.GetPageID(), res)
if err == nil {
return
}
if req.GetReplicaSet() == 0 {
return err
}
res.ReplicaSet = req.GetReplicaSet()
res.ID = req.GetPageID()
return rony.SavePageWithTxn(txn, alloc, res)
})
if err != nil {
ctx.PushError(rony.ErrCodeInternal, err.Error())
return
}
ctx.PushMessage(rony.C_Page, res)
}