// Publish sends a single MQTT PUBLISH message to the server. On completion, the
// supplied OnCompleteFunc is called. For QOS 0 messages, onComplete is called
// immediately after the message is sent to the outgoing buffer. For QOS 1 messages,
// onComplete is called when PUBACK is received. For QOS 2 messages, onComplete is
// called after the PUBCOMP message is received.
func (this *Server) Publish(msg *message.PublishMessage, onComplete OnCompleteFunc) error {
if err := this.checkConfiguration(); err != nil {
return err
}
if msg.Retain() {
if err := this.topicsMgr.Retain(msg); err != nil {
glog.Errorf("Error retaining message: %v", err)
}
}
if err := this.topicsMgr.Subscribers(msg.Topic(), msg.QoS(), &this.subs, &this.qoss); err != nil {
return err
}
msg.SetRetain(false)
//glog.Debugf("(server) Publishing to topic %q and %d subscribers", string(msg.Topic()), len(this.subs))
for _, s := range this.subs {
if s != nil {
fn, ok := s.(*OnPublishFunc)
if !ok {
glog.Errorf("Invalid onPublish Function")
} else {
(*fn)(msg)
}
}
}
return nil
}
// For PUBLISH message, we should figure out what QoS it is and process accordingly
// If QoS == 0, we should just take the next step, no ack required
// If QoS == 1, we should send back PUBACK, then take the next step
// If QoS == 2, we need to put it in the ack queue, send back PUBREC
func (this *service) processPublish(msg *message.PublishMessage) error {
switch msg.QoS() {
case message.QosExactlyOnce:
this.sess.Pub2in.Wait(msg, nil)
resp := message.NewPubrecMessage()
resp.SetPacketId(msg.PacketId())
_, err := this.writeMessage(resp)
return err
case message.QosAtLeastOnce:
resp := message.NewPubackMessage()
resp.SetPacketId(msg.PacketId())
if _, err := this.writeMessage(resp); err != nil {
return err
}
return this.onPublish(msg)
case message.QosAtMostOnce:
return this.onPublish(msg)
}
return fmt.Errorf("(%s) invalid message QoS %d.", this.cid(), msg.QoS())
}
// onPublish() is called when the server receives a PUBLISH message AND have completed
// the ack cycle. This method will get the list of subscribers based on the publish
// topic, and publishes the message to the list of subscribers.
func (this *service) onPublish(msg *message.PublishMessage) error {
if msg.Retain() {
if err := this.topicsMgr.Retain(msg); err != nil {
glog.Errorf("(%s) Error retaining message: %v", this.cid(), err)
}
}
err := this.topicsMgr.Subscribers(msg.Topic(), msg.QoS(), &this.subs, &this.qoss)
if err != nil {
glog.Errorf("(%s) Error retrieving subscribers list: %v", this.cid(), err)
return err
}
msg.SetRetain(false)
//glog.Debugf("(%s) Publishing to topic %q and %d subscribers", this.cid(), string(msg.Topic()), len(this.subs))
for _, s := range this.subs {
if s != nil {
fn, ok := s.(*OnPublishFunc)
if !ok {
glog.Errorf("Invalid onPublish Function")
return fmt.Errorf("Invalid onPublish Function")
} else {
(*fn)(msg)
}
}
}
return nil
}
func (this *service) publish(msg *message.PublishMessage, onComplete OnCompleteFunc) error {
//glog.Debugf("service/publish: Publishing %s", msg)
_, err := this.writeMessage(msg)
if err != nil {
return fmt.Errorf("(%s) Error sending %s message: %v", this.cid(), msg.Name(), err)
}
switch msg.QoS() {
case message.QosAtMostOnce:
if onComplete != nil {
return onComplete(msg, nil, nil)
}
return nil
case message.QosAtLeastOnce:
return this.sess.Pub1ack.Wait(msg, onComplete)
case message.QosExactlyOnce:
return this.sess.Pub2out.Wait(msg, onComplete)
}
return nil
}