wolvie/short
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: change shortener input type from []byte to string

Browse Source
This commit is contained in:
Tom Andrade 2022-05-17 14:16:23 +02:00
parent 267d2805ab
commit a31df48b23
Signed by: wolvie
GPG Key ID: 31AAB07872E82669
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
internal/shortie/shortie.go
View File

@ -113,11 +113,9 @@ func redirect(k string) (string, error) {
// shortener receive a url, validates the url, generate a random suffix string // shortener receive a url, validates the url, generate a random suffix string
// of urlSize size, checks if the suffix string is ensure on the kv database // of urlSize size, checks if the suffix string is ensure on the kv database
// and then writes the kv pair (suffix, url) to the database, returning the suffix // and then writes the kv pair (suffix, url) to the database, returning the suffix
func shortener(u []byte, s int) (string, error) { func shortener(u string, s int) (string, error) {
var su string var su string
us := string(u) if !govalidator.IsURL(string(u)) {
if !govalidator.IsURL(string(us)) {
return su, fmt.Errorf("Bad Request")
return su, ErrBadRequest return su, ErrBadRequest
} }
pu, _ := url.Parse(u) pu, _ := url.Parse(u)
@ -199,7 +197,7 @@ func IndexHandler(t *template.Template) func(ctx *fasthttp.RequestCtx) {
func Short(t *template.Template) func(ctx *fasthttp.RequestCtx) { func Short(t *template.Template) func(ctx *fasthttp.RequestCtx) {
return func(ctx *fasthttp.RequestCtx) { return func(ctx *fasthttp.RequestCtx) {
ctx.Response.Header.SetCanonical([]byte("Content-Type"), []byte("text/html")) ctx.Response.Header.SetCanonical([]byte("Content-Type"), []byte("text/html"))
suf, err := shortener(ctx.FormValue("url"), URLSize) suf, err := shortener(string(ctx.FormValue("url")), URLSize)
if err != nil { if err != nil {
ctx.SetStatusCode(fasthttp.StatusBadRequest) ctx.SetStatusCode(fasthttp.StatusBadRequest)
t.Execute(ctx, body{ t.Execute(ctx, body{