Stop SparkSession after certain time - similar to SparkStreamingContext.awaitTerminationorTimeout()

Question

Is there a way to stop a SparkSession after a certain time has expired, like StreamingContext. awaitTerminationOrTimeout(1000L) ?