send_options.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
		}
	}
}