refactor: extract func to getting hostname from host

Author: marjune163

src/app/main.go

@@ -3,13 +3,13 @@ package app
 import (
 	"../param"
 	"../serverErrHandler"
+	"../util"
 	"../vhost"
 	"crypto/tls"
 	"errors"
 	"net"
 	"net/http"
 	"os"
-	"strings"
 	"sync"
 )
 
@@ -211,11 +211,7 @@ func NewApp(params []*param.Param) *App {
 		l.handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 			var serveVHost *vhost.VHost
 
-			hostname := r.Host
-			colonIndex := strings.LastIndexByte(hostname, ':')
-			if colonIndex >= 0 {
-				hostname = hostname[:colonIndex]
-			}
+			hostname := util.ExtractHostname(r.Host)
 
 			for _, vh := range l.vhosts {
 				if vh.MatchHostname(hostname) {

src/util/extractHostname.go

@@ -0,0 +1,11 @@
+package util
+
+import "strings"
+
+func ExtractHostname(host string) string {
+	colonIndex := strings.LastIndexByte(host, ':')
+	if colonIndex >= 0 {
+		return host[:colonIndex]
+	}
+	return host
+}

src/util/extractHostname_test.go

@@ -0,0 +1,19 @@
+package util
+
+import "testing"
+
+func TestExtractHostname(t *testing.T) {
+	var host, hostname string
+
+	host = "example.com"
+	hostname = ExtractHostname(host)
+	if hostname != "example.com" {
+		t.Error(hostname)
+	}
+
+	host = "example.com:8080"
+	hostname = ExtractHostname(host)
+	if hostname != "example.com" {
+		t.Error(hostname)
+	}
+}