diff options
author | William Carroll <wpcarro@gmail.com> | 2020-02-09T13·47+0000 |
---|---|---|
committer | William Carroll <wpcarro@gmail.com> | 2020-02-10T10·06+0000 |
commit | 64654d1d6d1f36a20e00ee6cdc3866ff009c3a0a (patch) | |
tree | 3a764c88197e704a86ef527467b49a933c9f242d /gopkgs/utils/utils.go | |
parent | ec4c8472ca8eaed9a40d5decf5f909bdda96ec62 (diff) |
Create gopkgs directory for golang libs
- Created a gopkgs directory and registered it with default.nix's readTree - Moved monzo_ynab/utils -> gopkgs - Consumed utils.go in main.go - Renamed monzo_ynab -> job
Diffstat (limited to 'gopkgs/utils/utils.go')
-rw-r--r-- | gopkgs/utils/utils.go | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/gopkgs/utils/utils.go b/gopkgs/utils/utils.go new file mode 100644 index 000000000000..24a01cea9d1d --- /dev/null +++ b/gopkgs/utils/utils.go @@ -0,0 +1,56 @@ +// Some utility functions to tidy up my Golang. +package utils + +import ( + "log" + "io/ioutil" + "net/http" + "net/http/httputil" +) + +// Call log.Fatal with `err` when it's not nil. +func FailOn(err error) { + if err != nil { + log.Fatal(err) + } +} + +// Make a simple GET request to `url`. Fail if anything returns an error. I'd +// like to accumulate a library of these, so that I can write scrappy Go +// quickly. For now, this function just returns the body of the response back as +// a string. +func SimpleGet(url string, headers map[string]string, debug bool) string { + client := &http.Client{} + req, err := http.NewRequest("GET", url, nil) + if err != nil { + log.Fatal(err) + } + for k, v := range headers { + req.Header.Add(k, v) + } + + res, err := client.Do(req) + if err != nil { + log.Fatal(err) + } + defer res.Body.Close() + + if debug { + bytes, _ := httputil.DumpRequest(req, true) + log.Println(string(bytes)) + bytes, _ = httputil.DumpResponse(res, true) + log.Println(string(bytes)) + } + + if res.StatusCode == http.StatusOK { + bytes, err := ioutil.ReadAll(res.Body) + if err != nil { + log.Fatal(err) + } + return string(bytes) + } else { + log.Println(res) + log.Fatalf("HTTP status code of response not OK: %v\n", res.StatusCode) + return "" + } +} |