iosarraysswiftappend

Can somebody give a snippet of "append if not exists" method in swift array?


Because I use this routine a lot, can somebody create an extension method of Swift array which will detect whether if the data that is going to be appended already exists, then it's not appended? I know that it's only a matter of few code like this:

var arr = [Int]()
for element in inputArr {
    if !arr.contains(element) { arr.append(element); }
}

Becomes:

var arr = [Int]()
for element in inputArr { arr.appendUnique(element); }

Or:

var arr = [String]()
for element in inputArr {
    if !arr.contains(element) { arr.append(element); }
}

Becomes:

var arr = [String]()
for element in inputArr { arr.appendUnique(element); }

Same method for different element types. Frankly, from this simple code, I also want to learn on how to extend the Collection with variable types. It fascinates me how Array's methods can have different parameter types whenever the object was initialized with different parameter types. Array and Dictionary are two things that I still don't get how to extend them properly. Thanks.


Solution

  • You can extend RangeReplaceableCollection, constrain its elements to Equatable and declare your method as mutating. If you want to return Bool in case the appends succeeds you can also make the result discardable. Your extension should look like this:

    extension RangeReplaceableCollection where Element: Equatable {
        @discardableResult
        mutating func appendIfNotContains(_ element: Element) -> (appended: Bool, memberAfterAppend: Element) {
            if let index = firstIndex(of: element) {
                return (false, self[index])
            } else {
                append(element)
                return (true, element)
            }
        }
    }