SwiftでRubyの宇宙船演算子を実装した

宇宙船演算子 <=>

「<=>」は、その形状から「宇宙船演算子」と呼ばれる演算子で、「左辺 < 右辺」であれば-1を、「左辺 > 右辺」であれば1を、等しければ0を返す演算子です。

100 <=> 500 # -1
100 <=> 100 # 0
500 <=> 100 # 1

例えば、sort処理に使う¹
JunkBox～主に個人的防備録～: RubyのUFO演算子を利用したソート処理について

swiftは演算子のオーバーライド、カスタム演算子を簡単に記述できるので、宇宙船演算子を実装してみた。

• 参考
  • Swift Operators - NSHipster

Operators.swift

infix operator <=> { associativity left }

public func <=> <T: Comparable>(lhs: T, rhs: T) -> Int {
    switch (lhs, rhs) {
    case let (l, r) where l < r:
        return -1
    case let (l, r) where l > r:
        return 1
    default: 
        return 0
    }
}

OperatorsTests.swift

func testSpacshipOperator() {
    XCTAssertEqual(1 <=> 2, -1)
    XCTAssertEqual(1 <=> 1, 0)
    XCTAssertEqual(2 <=> 1, 1)
}

1. Rubyのarray.sort関数は、要素の順番比較として<=>演算子が使われています sort, sort! (Array) - Rubyリファレンス ↩