Skip to content

Commit e7c5fff

Browse files
authored
[#101] Added new base message function. (#102)
Added a function for NewBaseMessage()
1 parent 40be6a3 commit e7c5fff

File tree

2 files changed

+21
-19
lines changed

2 files changed

+21
-19
lines changed

messaging/base_message.go

+15
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import (
77

88
"oss.nandlabs.io/golly/codec"
99
"oss.nandlabs.io/golly/ioutils"
10+
"oss.nandlabs.io/golly/uuid"
1011
)
1112

1213
type BaseMessage struct {
@@ -16,6 +17,20 @@ type BaseMessage struct {
1617
body *bytes.Buffer
1718
}
1819

20+
func NewBaseMessage() (baseMsg *BaseMessage, err error) {
21+
var uid *uuid.UUID
22+
uid, err = uuid.V4()
23+
if err == nil {
24+
baseMsg = &BaseMessage{
25+
id: uid.String(),
26+
headers: make(map[string]interface{}),
27+
headerTypes: make(map[string]reflect.Kind),
28+
body: &bytes.Buffer{},
29+
}
30+
}
31+
return
32+
}
33+
1934
func (bm *BaseMessage) Id() string {
2035
return bm.id
2136
}

messaging/local_message.go

+6-19
Original file line numberDiff line numberDiff line change
@@ -1,29 +1,16 @@
11
package messaging
22

3-
import (
4-
"bytes"
5-
"reflect"
6-
7-
"oss.nandlabs.io/golly/uuid"
8-
)
9-
103
type LocalMessage struct {
114
*BaseMessage
125
}
136

147
func NewLocalMessage() (msg Message, err error) {
15-
var uid *uuid.UUID
16-
uid, err = uuid.V4()
17-
if err != nil {
18-
return
19-
}
20-
msg = &LocalMessage{
21-
BaseMessage: &BaseMessage{
22-
id: uid.String(),
23-
headers: make(map[string]interface{}),
24-
headerTypes: make(map[string]reflect.Kind),
25-
body: &bytes.Buffer{},
26-
},
8+
var baseMsg *BaseMessage
9+
baseMsg, err = NewBaseMessage()
10+
if err == nil {
11+
msg = &LocalMessage{
12+
BaseMessage: baseMsg,
13+
}
2714
}
2815
return
2916
}

0 commit comments

Comments
 (0)