diff --git a/serve.go b/serve.go index 95e8f13..733b445 100644 --- a/serve.go +++ b/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()) - - return s.storage.Put(shortUrl, longUrl) + err := s.storage.Put(shortUrl, longUrl) + if err != nil { + return err + } + return c.SendString(shortUrl) }) return a