diff --git a/pubsub/lib.go b/pubsub/lib.go index 38f8773..8289dea 100644 --- a/pubsub/lib.go +++ b/pubsub/lib.go @@ -95,6 +95,15 @@ func (p *PSCli) Read(i interface{}) error { return json.Unmarshal(bs, i) } +func (p *PSCli) Close() error { + if !p.closed { + p.closed = true + close(p.ch) + return p.con.Close() + } + return nil +} + func NewCli(q string) (*PSCli, error) { ret := &PSCli{} ret.q = q