Merge pull request #2 from TransactPRO/add-apply-tube

Added apply tube to beanstalk Conn

conn.go

@@ -41,6 +41,52 @@ func NewConn(conn io.ReadWriteCloser) *Conn {
 	return c
 }
 
+// ApplyTube: call UseTube and WatchTube
+func (c *Conn) ApplyTube(tubeName string) error {
+	var err error
+	if err = c.UseTube(tubeName); err != nil {
+		return err
+	}
+	err = c.WatchTube(tubeName)
+	return err
+}
+
+// UseTube: set tube to send data into
+func (c *Conn) UseTube(tubeName string) error {
+	var err error
+	if err = checkName(tubeName); err != nil {
+		return err
+	}
+	c.Tube = Tube{c, tubeName}
+	c.used = tubeName
+	c.printLine("use", tubeName)
+	return err
+}
+
+// WatchTube: add tubeName to listen from
+func (c *Conn) WatchTube(tubeName string) error {
+	var err error
+	if err = checkName(tubeName); err != nil {
+		return err
+	}
+	c.TubeSet.Name[tubeName] = true
+	c.watched[tubeName] = true
+	c.printLine("watch", tubeName)
+	return err
+}
+
+// UnwatchTube: undo WatchTube
+func (c *Conn) UnwatchTube(tubeName string) error {
+	var err error
+	if err = checkName(tubeName); err != nil {
+		return err
+	}
+	c.TubeSet.Name[tubeName] = false
+	c.watched[tubeName] = false
+	c.printLine("ignore", tubeName)
+	return err
+}
+
 // Dial connects to the given address on the given network using net.Dial
 // and then returns a new Conn for the connection.
 func Dial(network, addr string) (*Conn, error) {