genericsを使って異なる型を戻り値とする関数の書き方を教えて下さい。

Question

genericsを使って異なる型を戻り値として受け取る関数の書き方を教えて下さい。

現状playgroundで↓なかんじでかいてまして、戻り値をオプショナルでないA型もしくはB型で受け取りたいと思っています。

import UIKit

enum Type {
    case a
    case b
}

class A {
    var name: String
    
    init(name: String) {
        self.name = name
    }
}

class B {
    var name: String
    
    init(name: String) {
        self.name = name
    }
}

class D {
    
    var name: String
    var arrayOfA: [A]
    var arrayOfB: [B]
    
    init(name: String) {
        self.name = name
        arrayOfA = [A]()
        arrayOfB = [B]()
    }
}


class Test {
    
    var data: D
    
    init(dataName: String) {
        data = D(name: dataName)
    }
    
    func getContents<T>(type: Type, index: Int) -> T {
        switch type {
        case .a :
            if data.arrayOfA.isEmpty { fatalError() } // ここは通らないのでクラッシュさせる
            return data.arrayOfA[index]  // <= Cannot convert return expression of type 'A' to return type 'T'
        case .b :
            if data.arrayOfA.isEmpty { fatalError() }　// ここは通らないのでクラッシュさせる
            return data.arrayOfB[index]  // <= Cannot convert return expression of type 'B' to return type 'T'
        }
    }
}


let test = Test(dataName: "test")

test.data.arrayOfA.append(A(name: "1"))
test.data.arrayOfB.append(B(name: "い"))

let sample = test.getContents(type: .a, index: 0)  // <= Generic parameter 'T' could not be inferred

print(type(of: sample), sample)

Answer 1

まず僕はAとBに何らかの類似性がない場合はジェネリクスを使うべき場面ではないと考えます。
そのありを無視して、回答としては

関数の方はエラーの出ている両方の行に as! Tを追加してください。

さらに

let sample = test.getContents(type: .a, index: 0)

この行はコンパイラがsampleの型を確定することが不可能なので

let sample: A = test.getContents(type: .a, index: 0)

のように型を明示してください。

Comments

[退会済みユーザー]

回答ありがとうございます。
もう一度同一の関数で処理するべきか考えてみます。
sampleの型は自明だと思ってたのですが、コンパイラが確定できないからエラーが出ているわけですね。

[MasakiHori]

人間が見ると自明であってもコンパイラはあくまでも任意の型を返す関数としてしか認識しません。

型パラメータは前後のヒントから解決されますので例えば
func hoge(_ a: A)
という関数があった場合は
hoge(test.getContents(type: .a, index: 0))
と利用されると型パラメータがAとして処理されます。
また、
let i: Int = test.getContents(type: .a, index: 0)
としても型パラメータがIntとして処理されコンパイルは通ります。
もちろんクラッシュしますがコンパイラはそこまではわかりません。