edit doc

gitgitcode · gitgitcode · commit 265680625309 · 2020-06-15T08:10:29.000+08:00
diff --git a/alink.go b/alink.go
@@ -116,7 +116,7 @@ func Title(httpBody *bytes.Reader) (t string, err error) {
 	title := ""
 	node, err := html.Parse(httpBody)
 	if err != nil {
-		return title,err
+		return title, err
 	}
 
 	title, _ = titleText(node)
@@ -130,24 +130,24 @@ func Alink(httpBody *bytes.Reader) (l *[]string, b bool) {
 	var links []string
 	node, err := html.Parse(httpBody)
 	if err != nil {
-		return &links,false
+		return &links, false
 	}
 	ff, _ := alLink(node, &links)
 	return ff, true
 }
 
 // alLink Get href url
-func alLink(node *html.Node, h *[]string) ( f *[]string,n bool) {
+func alLink(node *html.Node, h *[]string) (f *[]string, n bool) {
 	b := false
 
 	if isAHrefElement(node) {
 		for _, a := range node.Attr {
 			if a.Key == "href" {
 				s := trimHash(a.Val)
-				if check(h,s)==false{
+				if check(h, s) == false {
 					//*h = make([]string,0,len(*h)+len(s))
 					//log.Println(cap(f))
-					*h= append(*h, s)
+					*h = append(*h, s)
 				}
 
 				return h, true
@@ -177,7 +177,6 @@ func trimHash(l string) string {
 	return l
 }
 
-
 // Check url exits
 func check(sl *[]string, s string) bool {
 	var check bool
@@ -188,4 +187,4 @@ func check(sl *[]string, s string) bool {
 		}
 	}
 	return check
-}
+}
diff --git a/alink_test.go b/alink_test.go
@@ -85,23 +85,21 @@ func TestNewRespBody(t *testing.T) {
 	str := []byte(s)
 	reader := bytes.NewReader(str)
 	//str := strings.NewReader(s)
-	abc ,err := NewRespBody(reader)
-	if err !=nil{
+	abc, err := NewRespBody(reader)
+	if err != nil {
 		t.Error(err)
 	}
 	cc := html.NewTokenizer(abc)
 
 	tokenType := cc.Next()
 
 	token := cc.Token()
-	log.Print(tokenType,token)
+	log.Print(tokenType, token)
 	if token.DataAtom.String() != "div" {
 		t.Error("Wrong body ")
 	}
 }
 
-
-
 func TestVideo(t *testing.T) {
 	var reader = `<a href="http://jjjj.com">1</a>
    <video src="http://abc.com/ab.mp4">
@@ -126,7 +124,7 @@ func TestVideo(t *testing.T) {
 		wantS   [] string
 		wantErr bool
 	}{
-		{"video", f, []string{"http://abc.com/ab.mp4"}, false },
+		{"video", f, []string{"http://abc.com/ab.mp4"}, false},
 	}
 
 	//log.Print(tests[0].wantS)
@@ -160,15 +158,14 @@ func TestTitle(t *testing.T) {
 	p1 := []byte(page1)
 	a1 := args{bytes.NewReader(p1)}
 
-
 	tests := []struct {
 		name    string
 		args    args
 		wantT   string
 		wantErr bool
 	}{
-		{"title",a,"the title", false},
-		{"title",a1,"test1", false},
+		{"title", a, "the title", false},
+		{"title", a1, "test1", false},
 		// TODO: Add test cases.
 	}
 	for _, tt := range tests {
@@ -189,9 +186,9 @@ func BenchmarkAlink(b *testing.B) {
 	page := "<html><header><title>the title</title></header><body></body></html>"
 	p := []byte(page)
 
-	for i := 0;i<b.N ;i++{
+	for i := 0; i < b.N; i++ {
 		Alink(bytes.NewReader(p))
 	}
 }
 
-//go test -cover -v -coverprofile=c.out
+//go test -cover -v -coverprofile=c.out
