Блог

Я хотел бы удалить определенные символы из строки — вместо того, чтобы повторно вводить следующие символы в различные строки в моем приложении, я хотел бы создать расширение, на которое можно легко ссылаться в коде.

Как это можно превратить в расширение?

 var string = "11224B"
let removeCharacters: Set<Character> = [" ", ";", ".", "!", "/"]
string.removeAll(where: { removeCharacters.contains($0) })
print(string)
 

Что вам нужно, так это расширить протокол, который требует, чтобы вы реализовали метод removeAll(где:), в данном случае RangeReplaceableCollection и ограничили себя StringProtocol :

 extension RangeReplaceableCollection where Self: StringProtocol {
    mutating func remove(characters: Set<Element>) {
        removeAll(where: characters.contains)
    }
}
 

 var string = "1/1!2.2;4 B"
string.remove(characters: [" ", ";", ".", "!", "/"])
print(string)  // "11224Bn"
 

И метод без мутирования также, но вместо этого используется фильтр. Вам просто нужно вернуться Self :

 extension RangeReplaceableCollection where Self: StringProtocol {
    func removing(characters: Set<Element>) -> Self {
        filter { !characters.contains($0) }
    }
}
 

 let string = "1/1!2.2;4 B"
string.removing(characters: [" ", ";", ".", "!", "/"]) // "11224Bn"
 

Вы также можете сделать свой метод универсальным и разрешить любой тип последовательности, элемент которого равен типу элемента коллекции:

 extension RangeReplaceableCollection where Self: StringProtocol {
    mutating func remove<S: Sequence>(characters: S) where S.Element == Element {
        removeAll(where: characters.contains)
    }
    func removing<S: Sequence>(characters: S) -> Self where S.Element == Element {
        filter { !characters.contains($0) }
    }
}
 

 var string = "1/1!2.2;4 B"
let characters: Set<Character> = [" ", ";", ".", "!", "/"]
string.remove(characters: characters) 
string // "11224Bn"
 

 let string = "1/1!2.2;4 B"
let charactersString = " ;.!/"
string.removing(characters: charactersString) // "11224Bn"
 

Расширение:

 extension String {
    func removeCharacters(from characterSet: CharacterSet) -> String {
        let filteredString = self.unicodeScalars.filter { !characterSet.contains($0) }
        return String(String.UnicodeScalarView(filteredString))
    }
}
 

Использование:

 var string1 = "1122 ;4B"
print(string1.removeCharacters(from: [" ", ";", ".", "!", "/"]))
 

Использовать removeCharacters в качестве параметра по умолчанию:

 extension String {
    func removeCharacters(from characterSet: CharacterSet = [" ", ";", ".", "!", "/"]) -> String {
        let filteredString = self.unicodeScalars.filter { !characterSet.contains($0) }
        return String(String.UnicodeScalarView(filteredString))
    }
}

var string1 = "1122 ;4B"
print(string1.removeCharacters())