yggdrasil-network/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2025-04-29 22:55:06 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add empty lines

Browse source
This commit is contained in:
Alex Kotov 2021-07-22 23:30:14 +05:00
parent 215351aefc
commit 7e98367d43
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
cmd/yggdrasilctl/main.go
View file

@ -32,6 +32,7 @@ func main() {
func run() int { func run() int {
logbuffer := &bytes.Buffer{} logbuffer := &bytes.Buffer{}
logger := log.New(logbuffer, "", log.Flags()) logger := log.New(logbuffer, "", log.Flags())
defer func() int { defer func() int {
if r := recover(); r != nil { if r := recover(); r != nil {
logger.Println("Fatal error:", r) logger.Println("Fatal error:", r)
@ -60,6 +61,7 @@ func run() int {
fmt.Println(" - ", os.Args[0], "-endpoint=tcp://localhost:9001 getDHT") fmt.Println(" - ", os.Args[0], "-endpoint=tcp://localhost:9001 getDHT")
fmt.Println(" - ", os.Args[0], "-endpoint=unix:///var/run/ygg.sock getDHT") fmt.Println(" - ", os.Args[0], "-endpoint=unix:///var/run/ygg.sock getDHT")
} }
server := flag.String("endpoint", endpoint, "Admin socket endpoint") server := flag.String("endpoint", endpoint, "Admin socket endpoint")
injson := flag.Bool("json", false, "Output in JSON format (as opposed to pretty-print)") injson := flag.Bool("json", false, "Output in JSON format (as opposed to pretty-print)")
verbose := flag.Bool("v", false, "Verbose output (includes public keys)") verbose := flag.Bool("v", false, "Verbose output (includes public keys)")
@ -113,6 +115,7 @@ func run() int {
var conn net.Conn var conn net.Conn
u, err := url.Parse(endpoint) u, err := url.Parse(endpoint)
if err == nil { if err == nil {
switch strings.ToLower(u.Scheme) { switch strings.ToLower(u.Scheme) {
case "unix": case "unix":
@ -129,9 +132,11 @@ func run() int {
logger.Println("Connecting to TCP socket", u.Host) logger.Println("Connecting to TCP socket", u.Host)
conn, err = net.Dial("tcp", endpoint) conn, err = net.Dial("tcp", endpoint)
} }
if err != nil { if err != nil {
panic(err) panic(err)
} }
logger.Println("Connected") logger.Println("Connected")
defer conn.Close() defer conn.Close()
@ -176,7 +181,9 @@ func run() int {
if err := encoder.Encode(&send); err != nil { if err := encoder.Encode(&send); err != nil {
panic(err) panic(err)
} }
logger.Printf("Request sent") logger.Printf("Request sent")
if err := decoder.Decode(&recv); err == nil { if err := decoder.Decode(&recv); err == nil {
logger.Printf("Response received") logger.Printf("Response received")
if recv["status"] == "error" { if recv["status"] == "error" {
@ -212,6 +219,7 @@ func run() int {
if v, ok := recv["status"]; ok && v != "success" { if v, ok := recv["status"]; ok && v != "success" {
return 1 return 1
} }
return 0 return 0
} }