up

serve.go

@@ -22,13 +22,16 @@ func (s *Server) Admin() *fiber.App {
 	})
 
 	a.Post("/shorten", func(c *fiber.Ctx) error {
+		longUrl := string(c.Body())
 		shortUrl := c.Query("s")
 		if shortUrl == "" {
 			shortUrl = s.serverGen.NextID()
 		}
-		longUrl := string(c.Body())
+		err := s.storage.Put(shortUrl, longUrl)
-
+		if err != nil {
-		return s.storage.Put(shortUrl, longUrl)
+			return err
+		}
+		return c.SendString(shortUrl)
 	})
 
 	return a