diff --git a/models/response.go b/models/response.go index e5ac64e..5303a06 100644 --- a/models/response.go +++ b/models/response.go @@ -19,6 +19,10 @@ func (r *Response) SendError(msg string) { r.Message = msg } +func (r *Response) SendMessage(msg string) { + r.Message = msg +} + func (r *Response) AddGet(get Get) { get.Query = nil r.Get = append(r.Get, get) @@ -32,6 +36,13 @@ func (r *Response) AddSubscription(sub Subscribe) { r.Subscribe = append(r.Subscribe, sub) } +func (r *Response) AddUnsubscription(sub Subscribe) { + r.Subscribe = append(r.Subscribe, sub) +} + +func (r *Response) AddUPublish(pub Publish) { + r.Publish = append(r.Publish, pub) +} func (r *Response) IsValid() bool { return !r.Error }