Swift：Array+Extension（数组拓展类）
2022-06-14 本文已影响0人
心猿意码_
JKSwiftExtension，测试用例在 ArrayExtensionViewController.swift 里面
目录：
1、数组 的基本扩展
2、数组 有关索引 的扩展方法
3、遵守 Equatable 协议的数组 (增删改查) 扩展
4、遵守 NSObjectProtocol 协议对应数组的扩展方法
5、针对数组元素是 String 的扩展
一、数组 的基本扩展
// MARK:- 一、数组 的基本扩展
public extension Array {
// MARK: 1.1、安全的取某个索引的值
/// 安全的取某个索引的值
/// - Parameter index: 索引
/// - Returns: 对应 inde 的 value
func indexValue(safe index: Index) -> Element? {
return indices.contains(index) ? self[index] : nil
}
// MARK: 1.2、数组新增元素(可转入一个数组)
/// 数组新增元素(可转入一个数组)
/// - Parameter elements: 数组
mutating func append(_ elements: [Element]) {
for element in elements {
self.append(element)
}
}
}
二、数组 有关索引 的扩展方法
// MARK:- 二、数组 有关索引 的扩展方法
public extension Array where Element : Equatable {
// MARK: 2.1、获取数组中的指定元素的索引值
/// 获取数组中的指定元素的索引值
/// - Parameter item: 元素
/// - Returns: 索引值数组
func indexes(_ item: Element) -> [Int] {
var indexes = [Int]()
for index in 0..<count where self[index] == item {
indexes.append(index)
}
return indexes
}
// MARK: 2.2、获取元素首次出现的位置
/// 获取元素首次出现的位置
/// - Parameter item: 元素
/// - Returns: 索引值
func firstIndex(_ item: Element) -> Int? {
for (index, value) in lazy.enumerated() where value == item {
return index
}
return nil
}
// MARK: 2.3、获取元素最后出现的位置
/// 获取元素最后出现的位置
/// - Parameter item: 元素
/// - Returns: 索引值
func lastIndex(_ item: Element) -> Int? {
let indexs = indexes(item)
return indexs.last
}
}
三、遵守 Equatable 协议的数组 (增删改查) 扩展
// MARK:- 三、遵守 Equatable 协议的数组 (增删改查) 扩展
public extension Array where Element : Equatable {
// MARK: 3.1、删除数组的中的元素(可删除第一个出现的或者删除全部出现的)
/// 删除数组的中的元素(可删除第一个出现的或者删除全部出现的)
/// - Parameters:
/// - element: 要删除的元素
/// - isRepeat: 是否删除重复的元素
@discardableResult
mutating func remove(_ element: Element, isRepeat: Bool = true) -> Array {
var removeIndexs: [Int] = []
for i in 0 ..< count {
if self[i] == element {
removeIndexs.append(i)
if !isRepeat { break }
}
}
// 倒序删除
for index in removeIndexs.reversed() {
self.remove(at: index)
}
return self
}
// MARK: 3.2、从删除数组中删除一个数组中出现的元素，支持是否重复删除, 否则只删除第一次出现的元素
/// 从删除数组中删除一个数组中出现的元素，支持是否重复删除, 否则只删除第一次出现的元素
/// - Parameters:
/// - elements: 被删除的数组元素
/// - isRepeat: 是否删除重复的元素
@discardableResult
mutating func removeArray(_ elements: [Element], isRepeat: Bool = true) -> Array {
for element in elements {
if self.contains(element) {
self.remove(element, isRepeat: isRepeat)
}
}
return self
}
}
四、遵守 NSObjectProtocol 协议对应数组的扩展方法
// MARK:- 四、遵守 NSObjectProtocol 协议对应数组的扩展方法
public extension Array where Element : NSObjectProtocol {
// MARK: 4.1、删除数组中遵守NSObjectProtocol协议的元素，是否删除重复的元素
/// 删除数组中遵守NSObjectProtocol协议的元素
/// - Parameters:
/// - object: 元素
/// - isRepeat: 是否删除重复的元素
@discardableResult
mutating func remove(object: NSObjectProtocol, isRepeat: Bool = true) -> Array {
var removeIndexs: [Int] = []
for i in 0..<count {
if self[i].isEqual(object) {
removeIndexs.append(i)
if !isRepeat {
break
}
}
}
for index in removeIndexs.reversed() {
self.remove(at: index)
}
return self
}
// MARK: 4.2、删除一个遵守NSObjectProtocol的数组中的元素，支持重复删除
/// 删除一个遵守NSObjectProtocol的数组中的元素，支持重复删除
/// - Parameters:
/// - objects: 遵守NSObjectProtocol的数组
/// - isRepeat: 是否删除重复的元素
@discardableResult
mutating func removeArray(objects: [NSObjectProtocol], isRepeat: Bool = true) -> Array {
for object in objects {
if self.contains(where: {$0.isEqual(object)} ){
self.remove(object: object, isRepeat: isRepeat)
}
}
return self
}
}
五、针对数组元素是 String 的扩展
// MARK:- 五、针对数组元素是 String 的扩展
public extension Array where Self.Element == String {
// MARK: 5.1、数组转字符转（数组的元素是 字符串），如：["1", "2", "3"] 连接器为 - ，那么转化后为 "1-2-3"
/// 数组转字符转（数组的元素是 字符串），如：["1", "2", "3"] 连接器为 - ，那么转化后为 "1-2-3"
/// - Parameter separator: 连接器
/// - Returns: 转化后的字符串
func toStrinig(separator: String = "") -> String {
return self.joined(separator: separator)
}
}
六、最后是使用演示
import UIKit
class ViewController: UIViewController {
override func viewDidLoad() {
super.viewDidLoad()
view.backgroundColor = UIColor.red
self.arraySafeUse()
}
// MARK: - 数组篇
func arraySafeUse() {
/// 安全的取某个索引的值
let arr : Array = ["1"]
let indexStr = arr.indexValue(safe: 2)
let str = arr[safe: 2]
print("indexStr->下标越界: \(String(describing: indexStr))")
print("str->下标越界: \(String(describing: str))")
/// 数组新增元素(可转入一个数组)
var addArr : Array<Any> = []
addArr.append("1")
let nullStr : String? = nil
addArr.append(nullStr as Any)
addArr.append(["2","3"])
print("addArr->新增元素：\(String(describing: addArr))")
///获取数组中的指定元素的索引值
let indexArr : Array = ["1","2"]
let idxstr = indexArr.indexes("2")
print("idxstr->获取索引下标：\(String(describing: idxstr))")
///获取元素首次出现的位置
var firstArr : Array = ["3","1","2","2"]
let firstStr = firstArr.firstIndex("2")
print("firstStr->元素首次出现的位置：\(String(describing: firstStr))")
///获取元素最后出现的位置
let lastStr = firstArr.lastIndex("2")
print("lastStr->元素最后出现的位置：\(String(describing: lastStr))")
///删除数组的中的元素(可删除第一个出现的或者删除全部出现的)
firstArr.remove("2", isRepeat: true)
print("firstArr->删除全部出现的：\(String(describing: firstArr))")
var removeArr : Array = ["3","1","2","2"]
removeArr.remove("2", isRepeat: false)
print("removeArr->删除第一个出现的：\(String(describing: removeArr))")
///从删除数组中删除一个数组中出现的元素，支持是否重复删除, 否则只删除第一次出现的元素
var bigArr : Array = ["1","2","3","4"]
bigArr.removeArray(removeArr, isRepeat: true)
print("bigArr->从bigArr中删除removeArr中出现的元素：\(String(describing: bigArr))")
///数组转字符转（数组的元素是 字符串），如：["1", "2", "3"] 连接器为 - ，那么转化后为 "1-2-3"
let changeArr : Array = ["1","2","3","4"]
let changeStr = changeArr.toStrinig(separator: "-")
print("数组转字符：\(String(describing: changeStr))")
}
}
- 打印结果
indexStr->下标越界: nil
str->下标越界: nil
addArr->新增元素：["1", nil, ["2", "3"]]
idxstr->获取索引下标：[1]
firstStr->元素首次出现的位置：Optional(2)
lastStr->元素最后出现的位置：Optional(3)
firstArr->删除全部出现的：["3", "1"]
removeArr->删除第一个出现的：["3", "1", "2"]
bigArr->从bigArr中删除removeArr中出现的元素：["4"]
数组转字符：1-2-3-4
