Получить значение параметров URL

Я пытаюсь получить параметры из URL-адреса с помощью Swift. Допустим, у меня есть следующий URL:

http://mysite3994.com?test1=blah&test2=blahblah

Как я могу получить значения test1 и test2?

Вы можете использовать приведенный ниже код, чтобы получить параметр

func getQueryStringParameter(url: String, param: String) -> String? {
  guard let url = URLComponents(string: url) else { return nil }
  return url.queryItems?.first(where: { $0.name == param })?.value
}

Вызовите метод как let test1 = getQueryStringParameter(url, param: "test1")

Другой способ с расширением:

extension URL {
    public var queryParameters: [String: String]? {
        guard
            let components = URLComponents(url: self, resolvingAgainstBaseURL: true),
            let queryItems = components.queryItems else { return nil }
        return queryItems.reduce(into: [String: String]()) { (result, item) in
            result[item.name] = item.value
        }
    }
}

Шаг 1. Создайте расширение URL

extension URL {
    func valueOf(_ queryParamaterName: String) -> String? {
        guard let url = URLComponents(string: self.absoluteString) else { return nil }
        return url.queryItems?.first(where: { $0.name == queryParamaterName })?.value
    }
}

Шаг 2. Как использовать расширение

let newURL = URL(string: "http://mysite3994.com?test1=blah&test2=blahblah")!

newURL.valueOf("test1") // Output i.e "blah"
newURL.valueOf("test2") // Output i.e "blahblah"

Я также сделал расширение URL-адреса, но поместил поиск параметров запроса в нижний индекс.

extension URL {
    subscript(queryParam:String) -> String? {
        guard let url = URLComponents(string: self.absoluteString) else { return nil }
        return url.queryItems?.first(where: { $0.name == queryParam })?.value
    }
}

Применение:

let url = URL(string: "http://some-website.com/documents/127/?referrer=147&mode=open")!

let referrer = url["referrer"]  // "147"
let mode     = url["mode"]      // "open"

Похоже, что ни один из существующих ответов не работает, когда ссылка ведет на веб-сайт, созданный на Angular. Это связано с тем, что пути Angular часто включают в себя #символ (хэш) во всех ссылках, что url.queryItemsвсегда приводит к возврату nil.

Если ссылка выглядит так: http://example.com/path/#/morepath/aaa?test1=blah&test2=blahblah

Тогда параметры можно будет получить только из url.fragment. При добавлении некоторой дополнительной логики синтаксического анализа к расширению @Matt более универсальный код будет выглядеть так:

extension URL {
    subscript(queryParam: String) -> String? {
        guard let url = URLComponents(string: self.absoluteString) else { return nil }
        if let parameters = url.queryItems {
            return parameters.first(where: { $0.name == queryParam })?.value
        } else if let paramPairs = url.fragment?.components(separatedBy: "?").last?.components(separatedBy: "&") {
            for pair in paramPairs where pair.contains(queryParam) {
                return pair.components(separatedBy: "=").last
            }
            return nil
        } else {
            return nil
        }
    }
}

Использование остается прежним:

let url = URL(string: "http://example.com/path/#/morepath/aaa?test1=blah&test2=blahblah")!

let referrer = url["test1"]  // "blah"
let mode     = url["test2"]  // "blahblah"

Другой способ сделать это - создать расширение для URL-адреса для возврата компонентов, а затем создать расширение для [URLQueryItem] для извлечения значения из queryItems.

extension URL {
    var components: URLComponents? {
        return URLComponents(url: self, resolvingAgainstBaseURL: false)
    }
}

extension Array where Iterator.Element == URLQueryItem {
    subscript(_ key: String) -> String? {
        return first(where: { $0.name == key })?.value
    }
}

И это пример того, как это можно использовать:

if let urlComponents = URL(string: "http://mysite3994.com?test1=blah&test2=blahblah")?.components,
    let test1Value = urlComponents.queryItems?["test1"] {
    print(test1Value)
}