diff --git a/main.go b/main.go index c21becf..c717f1b 100644 --- a/main.go +++ b/main.go @@ -139,6 +139,32 @@ func indexHandler(w http.ResponseWriter, r *http.Request) { w.Write([]byte("500 Internal Server Error")) } } + + if command == "list" { + ctx := context.Background() + keys, err := client.Keys(ctx, "*").Result() + if err != nil { + fmt.Println(err) + w.WriteHeader(500) + w.Write([]byte("500 Internal Server Error")) + return + } + + resp := "" + for _, key := range keys { + value, err = client.Get(ctx, key).Result() + if err == redis.Nil { + w.WriteHeader(500) + w.Write([]byte("500 Internal Server Error")) + return + } + resp += key + ":" + value + "\n" + } + + w.WriteHeader(200) + w.Write([]byte(resp)) + } + } func shortlinkHandler(w http.ResponseWriter, r *http.Request) { diff --git a/susmng.py b/susmng.py index 9b1b0b5..259e905 100755 --- a/susmng.py +++ b/susmng.py @@ -7,6 +7,7 @@ import hashlib import pathlib import os import json +import sys def get_args(): @@ -64,8 +65,8 @@ def main(args): ).hexdigest() } ) - print(r) - print(r.content) + print(r, file=sys.stderr) + print(r.content.decode().strip()) return 0