sparkreaddb

@throws(classOf[ReadDBFeaturesException])
  def getRequirementFeaturesFromDBByProject(project: String): Option[DataFrame] ={
    var result: Option[DataFrame] = None
    try{
      val storedRFsDF = spark
        .sqlContext
        .read
        .format("com.microsoft.azure.cosmosdb.spark")
        .cosmosDB(featuresStorageMap)
        .toDF()
      result = Some(storedRFsDF.filter($"project" === project))
    }catch{
      case(e: Exception) => {
        logger.error(s"Error reading from database. reason: ${e}")
        throw new ReadDBFeaturesException(s"Error reading from database. reason: ${e}")
      }
    }
    result
  }
  
  
  @throws(classOf[ReadDBFeaturesException])
  def getRequirementFeatureFromDBByID(feature_id: String): RequirementFeaturePersistence = {
    try{
      val fId: Int = feature_id.toInt
      val rfp: RequirementFeaturePersistence = this.datastore
        .createQuery(classOf[RequirementFeaturePersistence])
        .filter("feature_id =" , feature_id)
        .get()
      rfp
    }catch{
      case e: Exception => throw new ReadDBFeaturesException(s"Error getting RFP from DB. Reason: ${e}")
    }
  }