-
Notifications
You must be signed in to change notification settings - Fork 142
/
event.go
60 lines (55 loc) · 1.85 KB
/
event.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
// SPDX-License-Identifier: LGPL-3.0-or-later
// Copyright 2019 DNA Dev team
//
/*
* Copyright (C) 2018 The ontology Authors
* This file is part of The ontology library.
*
* The ontology is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* The ontology is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with The ontology. If not, see <http://www.gnu.org/licenses/>.
*/
package actor
import (
"github.com/DNAProject/DNA/events"
"github.com/DNAProject/DNA/events/message"
"github.com/ontio/ontology-eventbus/actor"
)
type EventActor struct {
blockPersistCompleted func(v interface{})
smartCodeEvt func(v interface{})
}
//receive from subscribed actor
func (t *EventActor) Receive(c actor.Context) {
switch msg := c.Message().(type) {
case *message.SaveBlockCompleteMsg:
t.blockPersistCompleted(*msg.Block)
case *message.SmartCodeEventMsg:
t.smartCodeEvt(*msg.Event)
default:
}
}
//Subscribe save block complete and smartcontract Event
func SubscribeEvent(topic string, handler func(v interface{})) {
var props = actor.FromProducer(func() actor.Actor {
if topic == message.TOPIC_SAVE_BLOCK_COMPLETE {
return &EventActor{blockPersistCompleted: handler}
} else if topic == message.TOPIC_SMART_CODE_EVENT {
return &EventActor{smartCodeEvt: handler}
} else {
return &EventActor{}
}
})
var pid = actor.Spawn(props)
var sub = events.NewActorSubscriber(pid)
sub.Subscribe(topic)
}