对结果进行URL排序
(arr []mode.Link, url, host string)
| 66 | |
| 67 | // 对结果进行URL排序 |
| 68 | func UrlDispose(arr []mode.Link, url, host string) ([]mode.Link, []mode.Link) { |
| 69 | var urls []mode.Link |
| 70 | var urlts []mode.Link |
| 71 | var other []mode.Link |
| 72 | for _, v := range arr { |
| 73 | if strings.Contains(v.Url, url) { |
| 74 | urls = append(urls, v) |
| 75 | } else { |
| 76 | if host != "" && regexp.MustCompile(host).MatchString(v.Url) { |
| 77 | urlts = append(urlts, v) |
| 78 | } else { |
| 79 | other = append(other, v) |
| 80 | } |
| 81 | } |
| 82 | } |
| 83 | |
| 84 | for _, v := range urlts { |
| 85 | urls = append(urls, v) |
| 86 | } |
| 87 | |
| 88 | return RemoveRepeatElement(urls), RemoveRepeatElement(other) |
| 89 | } |
| 90 | |
| 91 | // 处理Headers配置 |
| 92 | func SetHeadersConfig(header *http.Header) *http.Header { |
no test coverage detected