Removing Duplicates From Array of Custom Objects Swift

后端 未结 4 1400
孤独总比滥情好
孤独总比滥情好 2020-11-30 08:04

I have a custom class defined as follows :

class DisplayMessage : NSObject {
var id : String?
var partner_image : UIImage?
var partner_name : String?
var la         


        
4条回答
  •  有刺的猬
    2020-11-30 08:27

    Create a free duplicate version of an Array, using equality comparisons based on a given key

    public extension Sequence {
    
        public func uniq(by getIdentifier: (Iterator.Element) -> Id) -> [Iterator.Element] {
            var ids = Set()
            return self.reduce([]) { uniqueElements, element in
                if ids.insert(getIdentifier(element)).inserted {
                    return uniqueElements + CollectionOfOne(element)
                }
                return uniqueElements
            }
        }
    
    
       public func uniq(by keyPath: KeyPath) -> [Iterator.Element] {
          return self.uniq(by: { $0[keyPath: keyPath] })
       }
    }
    
    public extension Sequence where Iterator.Element: Hashable {
    
        var uniq: [Iterator.Element] {
            return self.uniq(by: { (element) -> Iterator.Element in
                return element
            })
        }
    
    }
    

    Usage

    let numbers =  [1,2,3,4,5,6,7,1,1,1,]
    let cars = [Car(id:1), Car(id:1), Car(id:2)]
    
    numbers.uniq
    cars.uniq(by: { $0.id})
    cars.uniq(by: \Car.id)
    cars.uniq(by: \.id)
    

提交回复
热议问题