allow specific targets

Author: Gabriel Monroy

client/create.go

@@ -3,6 +3,8 @@ package client
 import (
 	"fmt"
 	"os"
+	"regexp"
+	"strconv"
 
 	"github.com/coreos/fleet/job"
 	"github.com/coreos/fleet/unit"
@@ -42,15 +44,34 @@ func (c *FleetClient) Create(component string, data bool) (err error) {
 }
 
 // Create normal service unit
-func (c *FleetClient) createServiceUnit(component string) (unitName string, unitPtr *unit.Unit, err error) {
-	num, err := c.nextUnit(component)
-	if err != nil {
-		return
-	}
-	unitName, err = formatUnitName(component, num)
-	if err != nil {
-		return
+func (c *FleetClient) createServiceUnit(target string) (unitName string, unitPtr *unit.Unit, err error) {
+
+	// see if we were provided a specific target
+	r := regexp.MustCompile(`([a-z-]+)\.([\d]+)`)
+	match := r.FindStringSubmatch(target)
+	var (
+		num int
+		component string
+	)
+	if len(match) == 3 {
+		component = match[1]
+		num, err = strconv.Atoi(match[2])
+		if err != nil {
+			return "", nil, err
+		}
+		unitName, err = formatUnitName(component, num)
+	} else {
+		component = target
+		num, err := c.nextUnit(component)
+		if err != nil {
+			return "", nil, err
+		}
+		unitName, err = formatUnitName(component, num)
+		if err != nil {
+			return "", nil, err
+		}
 	}
+
 	unitPtr, err = NewUnit(component)
 	if err != nil {
 		return

client/destroy.go

@@ -1,20 +1,40 @@
 package client
 
-import "fmt"
+import (
+	"fmt"
+	"regexp"
+	"strconv"
+)
 
 // Destroy unschedules one unit for a given component type
 func (c *FleetClient) Destroy(component string) (err error) {
-	num, err := c.lastUnit(component)
-	if err != nil {
-		return
+
+	// see if we were provided a specific target
+	r := regexp.MustCompile(`([a-z-]+)\.([\d]+)`)
+	match := r.FindStringSubmatch(component)
+	var (
+		num int
+		unitName string
+	)
+	if len(match) == 3 {
+		num, err = strconv.Atoi(match[2])
+		if err != nil {
+			return err
+		}
+		unitName, err = formatUnitName(component, 0)
+	} else {
+		num, err = c.lastUnit(component)
+		if err != nil {
+			return err
+		}
+		unitName, err = formatUnitName(component, num)
+		if err != nil {
+			return err
+		}
 	}
 	if num == 0 {
 		return fmt.Errorf("no units to destroy")
 	}
-	unitName, err := formatUnitName(component, num)
-	if err != nil {
-		return
-	}
 	_, err = c.Fleet.Job(unitName)
 	if err != nil {
 		return

client/start.go

@@ -3,13 +3,24 @@ package client
 import (
 	"fmt"
 	"os"
+	"regexp"
 
 	"github.com/coreos/fleet/job"
 )
 
 // Start launches target units and blocks until active
 func (c *FleetClient) Start(target string, data bool) (err error) {
-	units, err := c.getUnits(target)
+
+	// see if we were provided a specific target
+	r := regexp.MustCompile(`([a-z-]+)\.([\d]+)`)
+	match := r.FindStringSubmatch(target)
+	var component string
+	if len(match) == 3 {
+		component = match[1]
+	} else {
+		component = target
+	}
+	units, err := c.getUnits(component)
 	if err != nil {
 		return
 	}