価格を表す数値を、0のとき「無料」にしたり3桁ごとにカンマで区切ったりした文字列にする

こんにちは

SwiftでiOSアプリを作っています。
Intとして持っている価格を表す変数 price を、

• 0なら 無料 と表示
• それ以外なら3桁ごとにカンマで区切って
• ¥ マークを先頭につけてStringとして返す

関数を作りましたが、これが複数のモデルにおいて必要になったので
今回はこれをプロトコル指向でまとめた PriceFormattable を紹介します

TL; DR

プロトコル側

PriceFormattable.swift

import Foundation

protocol PriceFormattable {

    var price: Int? { get set }

    func priceString() -> String?

}

extension PriceFormattable {

    func priceString() -> String? {
        guard let price = price else {
            return nil
        }

        guard price != 0 else {
            return "無料"
        }

        let numberFormatter = NumberFormatter()
        numberFormatter.numberStyle = NumberFormatter.Style.decimal

        guard let formattedPriceString = numberFormatter.string(from: NSNumber(value: price)) else {
            return ""
        }

        return "¥\(formattedPriceString)"
    }

}

構造体側

Item.swift

import Foundation

struct Item: PriceFormattable {
    var name: String?
    var price: Int?
}

やってること

プロトコル側

PriceFormattable.swift

protocol PriceFormattable {
    var price: Int? { get set }
    func priceString() -> String?
}

PriceFormattable を採用するクラスや構造体は、 price を持っていることを必須とします。
priceString() 関数を生やします。

PriceFormattable.swift

extension PriceFormattable {
    func priceString() -> String? {
        guard let price = price else {
            return nil
        }
        guard price != 0 else {
            return "無料"
        }
        let numberFormatter = NumberFormatter()
        numberFormatter.numberStyle = NumberFormatter.Style.decimal
        guard let formattedPriceString = numberFormatter.string(from: NSNumber(value: price)) else {
            return ""
        }
        return "¥\(formattedPriceString)"
    }
}

すぐ下に Protocol Extension を書きます。中身は地味なので割愛します
これで price 変数を持って PriceFormattable を採用するだけで priceString() 関数が使えるようになります。

実装側

Item.swift

struct Item: PriceFormattable {
    ...
}

PriceFormattable プロトコルを採用します。これで

なんとかViewController.swift

var item = Item()
item.price = 100

print(item.priceString()) // => "¥ 100"

とかいうことができるようになります。