今天无意中看到项目中有下面一段代码,怎么看都感觉有点不舒服,顺手就给优化了下,下面是之前的代码,这是一个支付宝返回状态码的判断处理:

 AlipaySDK.defaultService().processOrder(withPaymentResult: url, standbyCallback: { (resultDic) in
                let returnCode = resultDic?["resultStatus"] as! String
                switch returnCode{
                case "6001":
                    ToastUtility.showTip("用户中途取消")
                case "8000":
                    ToastUtility.showTip("正在处理中....")
                case "4000":
                    ToastUtility.showTip("订单支付失败")
                case "9000":
                    ToastUtility.showTip("订单支付成功")
                default:
                    break
                }
            })

上面代码的不好之处:
1 单纯的状态码我们并不知道它代表什么意思
2 如果项目中还有其他的状态判断也这样写,太散了,没有一个集中管理的地方

下面是优化后的代码:

// PayStatus.swift 

enum PayStatus:String{
    case cancel = "6001"
    case paying = "8000"
    case payFail = "4000"
    case paySucess = "9000"
    var rawValue:String {
         get {
          switch self {
                 case .cancel:
                     return "用户中途取消"
                 case .paying:
                     return "正在处理中...."
                 case .payFail:
                     return "订单支付失败"
                 case .paySucess:
                     return "订单支付成功"
                 }
         }
     }
}

AlipaySDK.defaultService().processOrder(withPaymentResult: url, standbyCallback: { (resultDic) in
                let returnCode = resultDic?["resultStatus"] as! String
                let status = PayStatus.init(rawValue: returnCode)
                ToastUtility.showTip(status!.rawValue)
            })

这里我通过返回的状态码初始化枚举,重写了计算属性rawValue,支付宝回调里面非常简单,就是初始化枚举,并打印枚举的rawValue,这样通过枚举的case我们能清楚的知道每个状态码代表什么意思,另外枚举我们放到一个单独文件,方便复用,以后有类似的状态吗返回我们也可以定义成枚举.

下面我们详细讲解一下为什么我们通过init(rawValue)初始化枚举,而打印出来的status!.rawValue,却不是我们传入的值呢?
我把上面的代码抽象成以下代码方便调试:

//
//  main.swift
//  test
//
//  Copyright © 2021 mac. All rights reserved.
//

import Foundation

enum PayStatus:String{
    case cancel = "6001"
    case paying = "8000"
    case payFail = "4000"
    case paySucess = "9000"
    var rawValue:String {
         get {
          switch self {
                 case .cancel:
                     return "用户中途取消"
                 case .paying:
                     return "正在处理中...."
                 case .payFail:
                     return "订单支付失败"
                 case .paySucess:
                     return "订单支付成功"
                 }
         }
     }
}
let returnCode = "8000"
let status = PayStatus.init(rawValue: returnCode)
//print(status!.rawValue)

我们在let status = PayStatus.init(rawValue: returnCode)打上断点,并勾选Xcode上的Always Show Disassemly

let returnCode = "4000"
let status = PayStatus.init(rawValue: returnCode)

let returnCode = "4000"
let status = PayStatus.init(rawValue: returnCode)

status内存里的值会依次是02,03.这说明了通过let status = PayStatus.init(rawValue: returnCode)初始化它会通过returnCode去匹配PayStatus枚举的case,并赋值case的flag,就是标记这个值的位置,我们知道枚举底层在原始值里只记录原始值的位置,并不会记录值的内容好通过returnCode,我们能确定一个枚举值了,那么print(status!.rawValue)这么怎么会是我们自定义的呢.

对于rawValue 如果我们不提供实现,swift会帮我们默认实现,如果我们实现了swift就不会帮我们实现了.
OK 这就是为什么我们通过init(rawValue)初始化枚举,而打印出来的status!.rawValue,却不是我们传入的值呢?