io.radicalbit.flink.pmml.scala.api.functions.EvaluationFunction.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of flink-jpmml-scala_2.10 Show documentation

flink-jpmml-scala

The newest version!

/*
 * Copyright (C) 2017  Radicalbit
 *
 * This file is part of flink-JPMML
 *
 * flink-JPMML is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * flink-JPMML is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with flink-JPMML.  If not, see .
 */

package io.radicalbit.flink.pmml.scala.api.functions

import io.radicalbit.flink.pmml.scala.api.PmmlModel
import io.radicalbit.flink.pmml.scala.api.exceptions.ModelLoadingException
import io.radicalbit.flink.pmml.scala.api.reader.ModelReader
import io.radicalbit.flink.pmml.scala.logging.LazyLogging
import org.apache.flink.api.common.functions.RichFlatMapFunction
import org.apache.flink.configuration.Configuration

import scala.util.{Failure, Success, Try}

/** Abstract class extending a [[RichFlatMapFunction]]; it provides:
  * the `evaluator` lazy evaluated object as instance of [[PmmlModel]]
  * the open method as a builder of the evaluator instance at operator initialization time
  *
  * @param reader The model reader instance coupled with the model source path
  * @tparam IN The input Type
  * @tparam OUT The output Type
  */
private[scala] abstract class EvaluationFunction[IN, OUT](reader: ModelReader)
    extends RichFlatMapFunction[IN, OUT]
    with LazyLogging {

  protected lazy val evaluator: PmmlModel = PmmlModel.fromReader(reader)

  override def open(parameters: Configuration): Unit = Try(evaluator.evaluator.model.getModel) match {
    case Success(model) => logger.info(s"Model has been read successfully, model name: {}", model.getModelName)
    case Failure(e) => throw new ModelLoadingException(e.getMessage, e)
  }

}