Проверьте, содержит ли столбец пропущенные значения

Я пишу функцию проверки, которая должна проверить, что:

1. Dataset<Row> ds имеет столбцы col1 и col2. Он может содержать и другие столбцы.
2. Убедитесь, что в col1 и col2 нет пропущенных значений (NaN, NULL, null и т. д.).

Код должен быть кодом Java. Вот что у меня есть до сих пор

private static boolean isSleepDataValid(SparkSession spark, Dataset<Row> ds) {

    ds.createOrReplaceTempView("validityData");
    if (spark.sql("SELECT col1 FROM validityData WHERE col1 IS NULL").count() > 0) {
        return false;
    }
    if (spark.sql("SELECT col2 FROM validityData WHERE col2 IS NULL").count() > 0) {
        return false;
    }
    return true;
}

Однако это уродливо, и я ищу более приятный и чистый способ. Я хотел бы избежать SparkSession spark в подписи и SQL в теле функции.