传递随机类型作为参数swift

Question

有这段代码：

self.window = UIWindow(frame: UIScreen.main.bounds)
let mainStoryboard: UIStoryboard = UIStoryboard(name: "Main", bundle: nil)

var exampleViewController: CustomClass1 = mainStoryboard.instantiateViewController(withIdentifier: "custom") as! CustomClass1

self.window?.rootViewController = exampleViewController        
self.window?.makeKeyAndVisible()

我想创建一个在其中包含此代码的函数，但问题是我想将类作为参数传递，这样如果我传递CustomClass2，代码的中间行应该像

一样工作

var exampleViewController: CustomClass2 = mainStoryboard.instantiateViewController(withIdentifier: "custom") as! CustomClass2

应该像这样调用函数：foo(class: CustomClass3)

我该如何写这个函数？

PS：我不确定这是否有所不同，但所有这些CustomClasses都是Custom ViewControllers。