Browse Source

fixed global flags

pull/55/head
Jae Kwon 10 years ago
parent
commit
adccd8f878
2 changed files with 117 additions and 30 deletions
  1. +30
    -9
      cmd/debora/commands.go
  2. +87
    -21
      cmd/debora/main.go

+ 30
- 9
cmd/debora/commands.go View File

@ -1,6 +1,7 @@
package main package main
import ( import (
"fmt"
acm "github.com/tendermint/tendermint/account" acm "github.com/tendermint/tendermint/account"
"github.com/tendermint/tendermint/binary" "github.com/tendermint/tendermint/binary"
btypes "github.com/tendermint/tendermint/cmd/barak/types" btypes "github.com/tendermint/tendermint/cmd/barak/types"
@ -8,13 +9,33 @@ import (
"github.com/tendermint/tendermint/rpc" "github.com/tendermint/tendermint/rpc"
) )
// Convenience function for a single validator.
func ListProcesses(privKey acm.PrivKey, remote string) (btypes.ResponseListProcesses, error) {
command := btypes.CommandListProcesses{}
nonce := GetNonce(remote)
func RunProcess(privKey acm.PrivKey, remote string, command btypes.CommandRunProcess) (response btypes.ResponseRunProcess, err error) {
nonce, err := GetNonce(remote)
if err != nil {
return response, err
}
commandBytes, signature := SignCommand(privKey, nonce+1, command)
_, err = RunAuthCommand(remote, commandBytes, []acm.Signature{signature}, &response)
return response, err
}
func StopProcess(privKey acm.PrivKey, remote string, command btypes.CommandStopProcess) (response btypes.ResponseStopProcess, err error) {
nonce, err := GetNonce(remote)
if err != nil {
return response, err
}
commandBytes, signature := SignCommand(privKey, nonce+1, command)
_, err = RunAuthCommand(remote, commandBytes, []acm.Signature{signature}, &response)
return response, err
}
func ListProcesses(privKey acm.PrivKey, remote string, command btypes.CommandListProcesses) (response btypes.ResponseListProcesses, err error) {
nonce, err := GetNonce(remote)
if err != nil {
return response, err
}
commandBytes, signature := SignCommand(privKey, nonce+1, command) commandBytes, signature := SignCommand(privKey, nonce+1, command)
response := btypes.ResponseListProcesses{}
_, err := RunAuthCommand(remote, commandBytes, []acm.Signature{signature}, &response)
_, err = RunAuthCommand(remote, commandBytes, []acm.Signature{signature}, &response)
return response, err return response, err
} }
@ -22,14 +43,14 @@ func ListProcesses(privKey acm.PrivKey, remote string) (btypes.ResponseListProce
// Utility method to get nonce from the remote. // Utility method to get nonce from the remote.
// The next command should include the returned nonce+1 as nonce. // The next command should include the returned nonce+1 as nonce.
func GetNonce(remote string) uint64 {
func GetNonce(remote string) (uint64, error) {
var err error var err error
response := btypes.ResponseStatus{} response := btypes.ResponseStatus{}
_, err = rpc.Call(remote, "status", Arr(), &response) _, err = rpc.Call(remote, "status", Arr(), &response)
if err != nil { if err != nil {
Exit(Fmt("Error fetching nonce from remote %v: %v", remote, err))
return 0, fmt.Errorf("Error fetching nonce from remote %v:\n %v", remote, err)
} }
return response.Nonce
return response.Nonce, nil
} }
// Each developer runs this // Each developer runs this


+ 87
- 21
cmd/debora/main.go View File

@ -9,9 +9,25 @@ import (
acm "github.com/tendermint/tendermint/account" acm "github.com/tendermint/tendermint/account"
"github.com/tendermint/tendermint/binary" "github.com/tendermint/tendermint/binary"
btypes "github.com/tendermint/tendermint/cmd/barak/types"
. "github.com/tendermint/tendermint/common" . "github.com/tendermint/tendermint/common"
) )
var (
remotes []string
privKey acm.PrivKey
remotesFlag = cli.StringFlag{
Name: "remotes",
Value: "http://127.0.0.1:8082",
Usage: "comma separated list of remote baraks",
}
privKeyFlag = cli.StringFlag{
Name: "privkey-file",
Value: "privkey",
Usage: "file containing private key json",
}
)
func main() { func main() {
fmt.Printf("New Debora Process (PID: %d)\n", os.Getpid()) fmt.Printf("New Debora Process (PID: %d)\n", os.Getpid())
app := cli.NewApp() app := cli.NewApp()
@ -19,34 +35,46 @@ func main() {
app.Usage = "summons commands to barak" app.Usage = "summons commands to barak"
app.Version = "0.0.1" app.Version = "0.0.1"
app.Email = "ethan@erisindustries.com,jae@tendermint.com" app.Email = "ethan@erisindustries.com,jae@tendermint.com"
app.Flags = []cli.Flag{}
app.Flags = []cli.Flag{
remotesFlag,
privKeyFlag,
}
app.Before = func(c *cli.Context) error {
remotes, privKey = ParseFlags(c)
return nil
}
app.Commands = []cli.Command{ app.Commands = []cli.Command{
cli.Command{
Name: "run",
Usage: "run process",
Action: cliRunProcess,
Flags: []cli.Flag{
//remotesFlag,
//privKeyFlag,
},
},
cli.Command{
Name: "stop",
Usage: "stop process",
Action: cliStopProcess,
Flags: []cli.Flag{
//remotesFlag,
//privKeyFlag,
},
},
cli.Command{ cli.Command{
Name: "list", Name: "list",
Usage: "list processes", Usage: "list processes",
Action: cliListProcesses, Action: cliListProcesses,
Flags: []cli.Flag{
remotesFlag,
privKeyFlag,
Flags: []cli.Flag{
//remotesFlag,
//privKeyFlag,
}, },
}, },
} }
app.Run(os.Args) app.Run(os.Args)
} }
var (
remotesFlag = cli.StringFlag{
Name: "remotes",
Value: "http://127.0.0.1:8082",
Usage: "comma separated list of remote baraks",
}
privKeyFlag = cli.StringFlag{
Name: "privkey-file",
Value: "privkey",
Usage: "file containing private key json",
}
)
func ParseFlags(c *cli.Context) (remotes []string, privKey acm.PrivKey) { func ParseFlags(c *cli.Context) (remotes []string, privKey acm.PrivKey) {
remotesStr := c.String("remotes") remotesStr := c.String("remotes")
remotes = strings.Split(remotesStr, ",") remotes = strings.Split(remotesStr, ",")
@ -62,8 +90,45 @@ func ParseFlags(c *cli.Context) (remotes []string, privKey acm.PrivKey) {
return remotes, privKey return remotes, privKey
} }
func cliRunProcess(c *cli.Context) {
/*
args := c.Args()
if len(args) == 0 {
log.Fatal("Must specify application name")
}
app := args[0]
*/
command := btypes.CommandRunProcess{}
for _, remote := range remotes {
response, err := RunProcess(privKey, remote, command)
if err != nil {
fmt.Printf("%v failure. %v\n", remote, err)
} else {
fmt.Printf("%v success. %v\n", remote, response)
}
}
}
func cliStopProcess(c *cli.Context) {
/*
args := c.Args()
if len(args) == 0 {
log.Fatal("Must specify application name")
}
app := args[0]
*/
command := btypes.CommandStopProcess{}
for _, remote := range remotes {
response, err := StopProcess(privKey, remote, command)
if err != nil {
fmt.Printf("%v failure. %v\n", remote, err)
} else {
fmt.Printf("%v success. %v\n", remote, response)
}
}
}
func cliListProcesses(c *cli.Context) { func cliListProcesses(c *cli.Context) {
remotes, privKey := ParseFlags(c)
/* /*
args := c.Args() args := c.Args()
if len(args) == 0 { if len(args) == 0 {
@ -71,12 +136,13 @@ func cliListProcesses(c *cli.Context) {
} }
app := args[0] app := args[0]
*/ */
command := btypes.CommandListProcesses{}
for _, remote := range remotes { for _, remote := range remotes {
response, err := ListProcesses(privKey, remote)
response, err := ListProcesses(privKey, remote, command)
if err != nil { if err != nil {
fmt.Printf("%v failed. %v\n", remote, err)
fmt.Printf("%v failure. %v\n", remote, err)
} else { } else {
fmt.Printf("%v processes: %v\n", remote, response.Processes)
fmt.Printf("%v success: %v\n", remote, response)
} }
} }
} }

Loading…
Cancel
Save