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

optimize port check

Browse Source
This commit is contained in:
Tom Andrade 2019-05-01 00:54:20 +02:00
parent 4a69686f81
commit a834bfab9f
Signed by: wolvie
GPG Key ID: 31AAB07872E82669
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
main.go
View File

@ -102,14 +102,6 @@ func shortner(w http.ResponseWriter, r *http.Request) {
break break
} }
} }
var hostSuf string
if port != "80" && proto == "http" {
hostSuf = ":" + port + "/"
} else if port != "443" && proto == "https" {
hostSuf = ":" + port + "/"
} else if port == "443" || port == "80" {
hostSuf = "/"
}
set(u.String(), suffix) set(u.String(), suffix)
shortend := proto + "://" + domain + hostSuf + path + suffix shortend := proto + "://" + domain + hostSuf + path + suffix
returnTmpl.Execute(w, shortend) returnTmpl.Execute(w, shortend)
@ -219,8 +211,12 @@ func main() {
if port > 65535 || port < 1 { if port > 65535 || port < 1 {
} }
if path != "" && !strings.HasSuffix(path, "/") { if port != 80 && proto == "http" {
path = path + "/" hostSuf = ":" + strconv.Itoa(port) + "/"
} else if port != 443 && proto == "https" {
hostSuf = ":" + strconv.Itoa(port) + "/"
} else if port == 443 || port == 80 {
hostSuf = "/"
} }
ip := net.ParseIP(addr) ip := net.ParseIP(addr)
if ip != nil { if ip != nil {