這個問題是一個朋友問我怎么寫腺劣,一開始我是拒絕的。我想這種東西網(wǎng)上隨便 google 下不就有了嗎因块。他說橘原,查了,但沒大看明白贮聂。于是我就查了下靠柑,沒想到這個寫法確實有點詭異寨辩,我第一反應(yīng)也沒看明白吓懈。所以隨便水一篇文章,強行完成本周的博客任務(wù)靡狞,順便給朋友一個交代耻警。
本文分為兩部分,第一部分是 Swift 怎么調(diào)用 Objective-C 的可變參數(shù)函數(shù),第二部分是 Objective-C 怎么調(diào)用 Swift 的可變參數(shù)函數(shù)甘穿。
Swift 調(diào)用 Objective-C 的可變參數(shù)函數(shù)
先寫一個例子
隨便寫一個 Objective-C 的可變參數(shù)函數(shù):接受 n 個 String 類型的參數(shù)腮恩,把它們一個一個地打印出來,然后返回參數(shù)一共有多少個温兼。這個方法毫無意義秸滴,只是為了強行有個返回值做例子編出來的而已……
- (NSInteger)foo:(NSString *)value,...
{
va_list list;
va_start(list, value);
NSInteger count = 0;
while (YES)
{
NSString *string = va_arg(list, NSString*);
if (!string) {
break;
}
NSLog(@"%@",string);
count++;
}
va_end(list);
return count;
}
這個方法直接在 swift 里調(diào)是調(diào)不了的。為了想要在 swift 里調(diào)用募判,需要把它稍微改造下荡含。
怎么改造一下
- 把方法簽名里的
,...
改成一個參數(shù)args:(va_list)list
-
va_list list;
和va_start(list, value);
這兩句需要去掉,因為我們的va_list
是傳進來的届垫。va_end
應(yīng)該也可以去掉了释液,不去掉也不會報錯,也許也可以保留著作為一個 good practice 吧装处。
改完之后的 Objective-C 方法:
- (NSInteger)foo:(va_list)list
{
NSInteger count = 0;
while (YES)
{
NSString *string = va_arg(list, NSString*);
if (!string) {
break;
}
NSLog(@"%@",string);
count++;
}
return count;
}
在 Swift 里怎么調(diào)用
既然 va_list
是作為一個參數(shù)傳進去的误债,關(guān)鍵是要用特殊方法構(gòu)造一個 va_list
。就跟在 Objective-C 里可以用 malloc 來強行構(gòu)造 va_list
一樣妄迁,Swift 里也有辦法寝蹈,有一個函數(shù)可以用:
public func withVaList<R>(_ args: [CVarArg], _ body: (CVaListPointer) -> R) -> R
這個函數(shù)的形式看起來不大常見,其實也很簡單登淘,它就是接受一個數(shù)組作為第一個參數(shù)躺盛,第二個參數(shù)是個閉包,閉包的參數(shù)就是生成好的va_list
形帮,而返回值你隨便返回什么都可以槽惫,閉包的返回值就是整個函數(shù)的返回值。
換句話說辩撑,就是你先傳給它一個數(shù)組界斜,讓它根據(jù)這個數(shù)組構(gòu)造 va_list
;然后它把構(gòu)造好的 va_list
用閉包的參數(shù)傳回來給你合冀,那么在閉包里這個 va_list
就隨你怎么用了各薇;如果閉包里你有什么結(jié)果想傳出去的,可以作為閉包的返回值返回君躺,它就會作為這個函數(shù)的返回值傳出去峭判,接受了這個返回值,后面就隨你怎么用了棕叫。
let testClass = TestClass()
let count = withVaList(["hello", "hamster", "good", "morning"]) { args -> Int in
return testClass.foo(args)
}
print(count)
輸出:
hello
hamster
good
morning
4
文檔里說了林螃,這個生成的 va_list
只許你在閉包里用,你不許把它傳出去在外面用俺泣,不然不保證 valid疗认。讓我們皮一下試試……
let testClass = TestClass()
let args = withVaList(["hello", "hamster", "good", "morning"]) { args -> CVaListPointer in
return args
}
print(testClass.foo(args))
結(jié)果是 crash完残,EXC_BAD_ACCESS,估計是到了閉包外面那塊空間已經(jīng)被釋放掉了横漏。這也從側(cè)面證明了不需要再寫 va_end
了吧……
還有另一個類似的函數(shù) getVaList
谨设,把 va_list
作為返回值返回出來的,寫法更簡潔缎浇,把上面的寫法改改就是這樣:
let count = testClass.foo(getVaList(["hello", "hamster", "good", "morning"]))
print(count)
但是文檔明確說了兩點:
- 能用
withVaList
就不要用getVaList
扎拣。具體原因沒說。 - 那為啥還要提供給你這個方法呢素跺?是因為有些情況語言規(guī)則不讓用
withVaList
鹏秋,比如在 class initializer 里。這時候就只好用getVaList
了亡笑。
包裝成 Swift 的可變參數(shù)方法
上面這語法侣夷,如果要用得很多,每次都這么寫怪煩的仑乌。我們可以給它包裝成一個 Swift 的可變參數(shù)方法……
extension TestClass {
func foo(_ strings: String...) -> Int {
return withVaList(strings) { args -> Int in
return foo(args)
}
}
}
然后調(diào)用的時候就一勞永逸了:
let testClass = TestClass()
let count = testClass.foo("hello", "hamster", "good", "morning")
print(count)
感慨下 Swift 的語法簡潔太多了百拓,不是嗎?
Objective-C 調(diào)用 Swift 的可變參數(shù)函數(shù)
既然 Swift 的語法這么簡潔晰甚,我們干脆把可變參數(shù)方法都在 Swift 里實現(xiàn)衙传,然后讓 Objective-C 來調(diào)唄?
然而 Swift 無情地拒絕了:
真的要調(diào)怎么辦厕九?只好另寫一個接受數(shù)組為參數(shù)的方法蓖捶,在 Objective-C 里調(diào)這個方法,或者再寫一個 Objective-C 的可變參數(shù)方法把它 wrap 一層了……