canonicalize.go 1.26 KB
Newer Older
ale's avatar
ale committed
1
2
3
4
5
6
// Copyright 2012-present Oliver Eilhard. All rights reserved.
// Use of this source code is governed by a MIT-license.
// See http://olivere.mit-license.org/license.txt for details.

package elastic

ale's avatar
ale committed
7
8
9
10
import (
	"net/url"
	"strings"
)
ale's avatar
ale committed
11
12
13
14
15
16
17
18
19

// canonicalize takes a list of URLs and returns its canonicalized form, i.e.
// remove anything but scheme, userinfo, host, path, and port.
// It also removes all trailing slashes. Invalid URLs or URLs that do not
// use protocol http or https are skipped.
//
// Example:
// http://127.0.0.1:9200/?query=1 -> http://127.0.0.1:9200
// http://127.0.0.1:9200/db1/ -> http://127.0.0.1:9200/db1
ale's avatar
ale committed
20
// http://127.0.0.1:9200/db1/// -> http://127.0.0.1:9200/db1
ale's avatar
ale committed
21
22
23
24
25
26
func canonicalize(rawurls ...string) []string {
	var canonicalized []string
	for _, rawurl := range rawurls {
		u, err := url.Parse(rawurl)
		if err == nil {
			if u.Scheme == "http" || u.Scheme == "https" {
ale's avatar
ale committed
27
28
29
30
				// Trim trailing slashes. Notice that strings.TrimSuffix will only remove the last slash,
				// not all slashes from the suffix, so we'll loop over the path to remove all slashes.
				for strings.HasSuffix(u.Path, "/") {
					u.Path = u.Path[:len(u.Path)-1]
ale's avatar
ale committed
31
32
33
34
35
36
37
38
39
				}
				u.Fragment = ""
				u.RawQuery = ""
				canonicalized = append(canonicalized, u.String())
			}
		}
	}
	return canonicalized
}