tableViewCellをリロードした後にもセルの選択状態を維持したい

Question

tableViewを部分的にリロードした後、選択状態にあったセルの状態を維持したいのですがうまくいきません。どうやったらできますか？

以前はいろんな方法でできていたような気がするのですが。

performBatchUpdate / UIViewAnimate.completion / beginUpdate / dispatchQueueなど

func hoge(reloadIndexPaths: [IndexPath]) {
    // 選択済みセルのインデックスパスをここで保存
    let indexPath: IndexPath? = tableView.indexPathForSelectedRow 
    tableView.reloadRows(at: reloadIndexPaths, with: .none)
  
    // reload後にここでセルを選択状態に戻す
    // いろいろ試したのですが、うまくいきません
    // 例えばメインスレッドに直列する
    DispatchQueue.main.async {
        tableView.selectRow(at: indexPath, animate: true, scrollPosition: .none)
    }

Answer 1

ごめんなさい。自分で選択できないようにしてました。

func tableView(_ tableView: UITableView, willSelectRowAt indexPath: IndexPath) -> IndexPath? {
        switch hogeStatus {
        case .a: return indexPath
        case .b: return nil
        }
    }