swift-字符串中的子字符串

Question

我有一些網址字符串，例如：

讓url1 =“ https：// abc / help？user_info_id = 5＃help-article03 ”

讓url2 =“ https：// cde？user_info_id = 455＆artical = 4 ”

讓url3 =“ https：// ghi / help？user_info_id = 5 ”

現在，我想從這些URL中獲取user_info_id。

url1返回5，url2返回455，url3返回5

這是我的代碼

func getID(string: String) -> String {
    let range = string.rangeOfString("user_info_id=")
    guard let endIndex = range?.endIndex else { return "" }
    let remain = string.substringFromIndex(endIndex)

    let num = "0123456789"
    var result = ""
    for c in remain.characters {
        if num.containsString(String(c)) {
            result.append(c)
        } else {
            break
        }
    }
    return result
}

請告訴我你的想法。 謝謝！

Answer 1

使用(NS)URLComponents並過濾查詢項：

let url1 = "https://cde?user_info_id=455&artical=4"
if let components = NSURLComponents(string: url1),
  let queryItems = components.queryItems,
  let userInfoId = queryItems.filter({$0.name == "user_info_id"}).first where userInfoId.value != nil {
  print(userInfoId.value!)
}