-
Notifications
You must be signed in to change notification settings - Fork 46
/
payload.go
125 lines (114 loc) · 3.22 KB
/
payload.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
124
125
package payload
import (
"errors"
"fmt"
"io/ioutil"
"os"
"time"
"github.com/rsocket/rsocket-go/common"
"github.com/rsocket/rsocket-go/framing"
)
var (
errNotFile = errors.New("target is not file")
errTooLargePooledPayload = fmt.Sprintf("too large pooled payload: maximum size is %d", common.MaxUint24-3)
)
type (
// Payload is a stream message (upstream or downstream).
// It contains data associated with a stream created by a previous request.
// In Reactive Streams and Rx this is the 'onNext' event.
Payload interface {
// Metadata returns raw metadata bytes.
// The ok result indicates whether metadata exists.
Metadata() (metadata []byte, ok bool)
// MetadataUTF8 returns metadata as UTF8 string.
// The ok result indicates whether metadata exists.
MetadataUTF8() (metadata string, ok bool)
// Data returns raw data bytes.
Data() []byte
// DataUTF8 returns data as UTF8 string.
DataUTF8() string
// Release release all resources of payload.
// Some payload implements is pooled, so you must release resoures after using it.
Release()
}
// SetupPayload is particular payload for RSocket Setup.
SetupPayload interface {
Payload
// DataMimeType returns MIME type of data.
DataMimeType() string
// MetadataMimeType returns MIME type of metadata.
MetadataMimeType() string
// TimeBetweenKeepalive returns interval duration of keepalive.
TimeBetweenKeepalive() time.Duration
// MaxLifetime returns max lifetime of RSocket connection.
MaxLifetime() time.Duration
// Version return RSocket protocol version.
Version() common.Version
}
)
// Clone create a copy of original payload.
func Clone(payload Payload) Payload {
ret := &rawPayload{}
if d := payload.Data(); len(d) > 0 {
clone := make([]byte, len(d))
copy(clone, d)
ret.data = clone
}
if m, ok := payload.Metadata(); ok && len(m) > 0 {
clone := make([]byte, len(m))
copy(clone, m)
ret.metadata = clone
}
return ret
}
// New create a new payload with bytes.
func New(data []byte, metadata []byte) Payload {
return &rawPayload{
data: data,
metadata: metadata,
}
}
// NewPooled returns a pooled payload.
// Remember call Release() at last.
// Sometimes payload will be released automatically. (eg: sent by a requester).
func NewPooled(data, metadata []byte) Payload {
size := framing.CalcPayloadFrameSize(data, metadata)
if size > common.MaxUint24-3 {
panic(errTooLargePooledPayload)
}
return framing.NewFramePayload(0, data, metadata)
}
// NewString create a new payload with strings.
func NewString(data, metadata string) Payload {
return &strPayload{
data: data,
metadata: metadata,
}
}
// NewFile create a new payload from file.
func NewFile(filename string, metadata []byte) (Payload, error) {
fi, err := os.Stat(filename)
if err != nil {
return nil, err
}
if fi.IsDir() {
return nil, errNotFile
}
// Check file size
//if fi.Size() > common.MaxUint24 {
// return nil, errTooLargeFile
//}
bs, err := ioutil.ReadFile(filename)
if err != nil {
return nil, err
}
return New(bs, metadata), nil
}
// MustNewFile create a new payload from file.
func MustNewFile(filename string, metadata []byte) Payload {
foo, err := NewFile(filename, metadata)
if err != nil {
panic(err)
}
return foo
}