blob: f80f1abcb32667784167f7349bed14b65c03979e (
plain)
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
|
package psmb
import (
"bytes"
"fmt"
"github.com/hit-mc/psmb-go/protocol"
"io"
)
// Publish a message. Panic if client mode is not Publisher. Block if send queue is full.
func (c *Client) Publish(msg []byte) error {
if c.mode.Type() != protocol.ModePublish {
panic(fmt.Errorf("invalid operation"))
}
c.txQueue <- bytesOutboundMessage{
b: msg,
waitChan: make(chan error),
}
return nil
}
type OutboundMessage interface {
// getContent return the content reader and length in bytes
getContent() (io.Reader, int64)
// Wait returns a channel which blocks until this message is successfully sent to the server.
Wait() <-chan error
}
type bytesOutboundMessage struct {
b []byte
waitChan chan error
}
func (b bytesOutboundMessage) getContent() (io.Reader, int64) {
return bytes.NewReader(b.b), int64(len(b.b))
}
func (b bytesOutboundMessage) Wait() <-chan error {
return b.waitChan
}
|
