56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package stomp
|
|
|
|
import (
|
|
"github.com/go-stomp/stomp/v3/frame"
|
|
)
|
|
|
|
// SendOpt contains options for for the Conn.Send and Transaction.Send functions.
|
|
var SendOpt struct {
|
|
// Receipt specifies that the client should request acknowledgement
|
|
// from the server before the send operation successfully completes.
|
|
Receipt func(*frame.Frame) error
|
|
|
|
// NoContentLength specifies that the SEND frame should not include
|
|
// a content-length header entry. By default the content-length header
|
|
// entry is always included, but some message brokers assign special
|
|
// meaning to STOMP frames that do not contain a content-length
|
|
// header entry. (In particular ActiveMQ interprets STOMP frames
|
|
// with no content-length as being a text message)
|
|
NoContentLength func(*frame.Frame) error
|
|
|
|
// Header provides the opportunity to include custom header entries
|
|
// in the SEND frame that the client sends to the server. This option
|
|
// can be specified multiple times if multiple custom header entries
|
|
// are required.
|
|
Header func(key, value string) func(*frame.Frame) error
|
|
}
|
|
|
|
func init() {
|
|
SendOpt.Receipt = func(f *frame.Frame) error {
|
|
if f.Command != frame.SEND {
|
|
return ErrInvalidCommand
|
|
}
|
|
id := allocateId()
|
|
f.Header.Set(frame.Receipt, id)
|
|
return nil
|
|
}
|
|
|
|
SendOpt.NoContentLength = func(f *frame.Frame) error {
|
|
if f.Command != frame.SEND {
|
|
return ErrInvalidCommand
|
|
}
|
|
f.Header.Del(frame.ContentLength)
|
|
return nil
|
|
}
|
|
|
|
SendOpt.Header = func(key, value string) func(*frame.Frame) error {
|
|
return func(f *frame.Frame) error {
|
|
if f.Command != frame.SEND {
|
|
return ErrInvalidCommand
|
|
}
|
|
f.Header.Add(key, value)
|
|
return nil
|
|
}
|
|
}
|
|
}
|