From 1b9721825d20d9c5dfe6d35bc8721d6c1be75e66 Mon Sep 17 00:00:00 2001
From: Gregory Eremin <g@erem.in>
Date: Wed, 11 Feb 2015 02:40:21 +0700
Subject: [PATCH] Unit debug tool

---
 socket_send.go | 16 ----------------
 unit_send.go   | 25 +++++++++++++++++++++++++
 2 files changed, 25 insertions(+), 16 deletions(-)
 delete mode 100644 socket_send.go
 create mode 100644 unit_send.go

diff --git a/socket_send.go b/socket_send.go
deleted file mode 100644
index 5a0fafa..0000000
--- a/socket_send.go
+++ /dev/null
@@ -1,16 +0,0 @@
-package main
-
-import (
-	"fmt"
-
-	"github.com/localhots/yeast/unit"
-)
-
-func main() {
-	u := unit.New("uuid")
-	msg := []byte("{}")
-
-	fmt.Println("Sending message:", string(msg))
-	resp, _ := u.Send(msg)
-	fmt.Println("Reply recieved:", string(resp))
-}
diff --git a/unit_send.go b/unit_send.go
new file mode 100644
index 0000000..4e1e99d
--- /dev/null
+++ b/unit_send.go
@@ -0,0 +1,25 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+
+	"github.com/localhots/yeast/unit"
+)
+
+func main() {
+	var (
+		name string
+		msg  string
+	)
+
+	flag.StringVar(&name, "unit", "", "Unit name")
+	flag.StringVar(&msg, "msg", "", "Message")
+	flag.Parse()
+
+	u := unit.New(name)
+
+	fmt.Println("Sending message:", msg)
+	resp, _ := u.Call([]byte(msg))
+	fmt.Println("Reply recieved:", string(resp))
+}