Refactor: encapsulate custom url manipulation function into object method.

2 files changed, 14 insertions, 8 deletions

diff --git a/common/urlparse.go b/common/myurl/urlparse.go
index dc72cee..69695d1 100644
--- a/common/urlparse.go
+++ b/common/myurl/urlparse.go
@@ -1,4 +1,4 @@
-package common
+package myurl
 
 import (
 	"errors"
@@ -6,10 +6,16 @@ import (
 	"strings"
 )
 
-// GetFileExtensionFromUrl
+type Url string
+
+func (o Url) Url() string {
+	return string(o)
+}
+
+// FileExtension returns file extension of file name from this url.
 // copied from https://elisegev.medium.com/get-a-file-extension-from-a-url-in-golang-5061d4a298a
-func GetFileExtensionFromUrl(rawUrl string) (string, error) {
-	u, err := url.Parse(rawUrl)
+func (o Url) FileExtension() (string, error) {
+	u, err := url.Parse(o.Url())
 	if err != nil {
 		return "", err
 	}
diff --git a/common/urlparse_test.go b/common/myurl/urlparse_test.go
index 9ed4449..b7df7bd 100644
--- a/common/urlparse_test.go
+++ b/common/myurl/urlparse_test.go
@@ -1,4 +1,4 @@
-package common
+package myurl
 
 import "testing"
 
@@ -12,15 +12,15 @@ func TestGetFileExtensionFromUrl(t *testing.T) {
 		{Expected: "flv"},
 	}
 	var err error
-	tests[0].Actual, err = GetFileExtensionFromUrl("http://www.example.com/index.html")
+	tests[0].Actual, err = Url("http://www.example.com/index.html").FileExtension()
 	if err != nil {
 		t.Fatalf("GetFileExtensionFromUrl: %v", err)
 	}
-	tests[1].Actual, err = GetFileExtensionFromUrl("https://www.example.com/index.htm")
+	tests[1].Actual, err = Url("https://www.example.com/index.htm").FileExtension()
 	if err != nil {
 		t.Fatalf("GetFileExtensionFromUrl: %v", err)
 	}
-	tests[2].Actual, err = GetFileExtensionFromUrl("https://www.example.com/video.flv?a=1&b=2flv")
+	tests[2].Actual, err = Url("https://www.example.com/video.flv?a=1&b=2flv").FileExtension()
 	if err != nil {
 		t.Fatalf("GetFileExtensionFromUrl: %v", err)
 	}