浏览代码

Expose `Close()` and `Drain()`.

Arto Bendiken 3 年之前
父节点
当前提交
f4c969d60b
共有 1 个文件被更改,包括 8 次插入0 次删除
  1. 8 0
      bus.go

+ 8 - 0
bus.go

@@ -25,6 +25,14 @@ func Connect(url string) (*Bus, error) {
 	return &Bus{NATS: encodedConn}, nil
 }
 
+func (bus *Bus) Close() {
+	bus.NATS.Close()
+}
+
+func (bus *Bus) Drain() error {
+	return bus.NATS.Drain()
+}
+
 func (bus *Bus) SendEmail(request *SendEmail) error {
 	return bus.Publish("email.outbound", request)
 }