vodkas

commit c7ccc3dce979ede6c1971cfee9d0e780fdf18f15
parent c39ed80b38272c7a8344c45714004b29b6070cca
Author: vh <vetle.haflan@gmail.com>
Date:   Sun, 23 Feb 2020 17:14:09 +0100

Store data as []byte instead of string

And now all types of data is supported (:D)

Closes #2

Diffstat:
M
vodkas.go
 | 
13
++++++-------

1 file changed, 6 insertions(+), 7 deletions(-)
diff --git a/vodkas.go b/vodkas.go
@@ -26,19 +26,19 @@ are removed completely.
 `
 
 type SyncedStorage struct{
-        links map[string]string
+        links map[string][]byte
         sync.Mutex
 }
 
 var storage SyncedStorage
 
-func (storage *SyncedStorage) push(code string, contents string) {
+func (storage *SyncedStorage) push(code string, contents []byte) {
         storage.Lock()
         defer storage.Unlock()
         storage.links[code] = contents
 }
 
-func (storage *SyncedStorage) pop(code string) (contents string, found bool) {
+func (storage *SyncedStorage) pop(code string) (contents []byte, found bool) {
         storage.Lock()
         defer storage.Unlock()
         contents, found = storage.links[code]
@@ -53,7 +53,7 @@ func MainHandler(res http.ResponseWriter, r *http.Request) {
                 res.WriteHeader(http.StatusInternalServerError)
                 return
         }
-        storage.push("sorandom", string(b))
+        storage.push("sorandom", b)
         textOnly := r.Header.Get("Simple") != ""    // Should be able to force simple
         textOnly = textOnly || strings.Contains(r.Header.Get("User-Agent"), "curl")
         var responseText string
@@ -93,8 +93,7 @@ func ShotHandler(res http.ResponseWriter, r *http.Request) {
                         res.WriteHeader(http.StatusInternalServerError)
                         log.Panicln("Error when trying to read body")
                 }
-                contents = string(b)
-                storage.push(code, contents)
+                storage.push(code, b)
                 if _, err := res.Write([]byte(fmt.Sprint("Contents stored in given link\n"))); err != nil {
                         log.Panicln("Error when trying to write response")
                 }
@@ -105,7 +104,7 @@ func ShotHandler(res http.ResponseWriter, r *http.Request) {
 
 /**************** Main ****************/
 func main(){
-        storage.links = make(map[string]string)
+        storage.links = make(map[string][]byte)
         port := flag.Int("p", 8080, "Port")
         flag.Parse()
         defer fmt.Println("Server shutting down")