Merge branch 'develop'
diff --git a/README.md b/README.md
index 7b29862..1ca76f4 100644
--- a/README.md
+++ b/README.md
@@ -6,13 +6,21 @@
## Versions
+### v0.3.0
+
+- update for PredictionIO 0.9.2, including:
+
+ - use new PEventStore API
+ - use appName in DataSource parameter
+
+
### v0.2.0
- update for PredictionIO 0.9.2
### v0.1.1
-- update for PredictionIO 0.9.0
+- update build.sbt and template.json for PredictionIO 0.9.2
### v0.1.0
diff --git a/engine.json b/engine.json
index 9ae7254..162c967 100644
--- a/engine.json
+++ b/engine.json
@@ -4,7 +4,7 @@
"engineFactory": "org.template.vanilla.VanillaEngine",
"datasource": {
"params" : {
- "appId": 1
+ "appName": "INVALID_APP_NAME"
}
},
"algorithms": [
diff --git a/src/main/scala/DataSource.scala b/src/main/scala/DataSource.scala
index 6116ee7..210387e 100644
--- a/src/main/scala/DataSource.scala
+++ b/src/main/scala/DataSource.scala
@@ -5,7 +5,7 @@
import io.prediction.controller.EmptyActualResult
import io.prediction.controller.Params
import io.prediction.data.storage.Event
-import io.prediction.data.storage.Storage
+import io.prediction.data.store.PEventStore
import org.apache.spark.SparkContext
import org.apache.spark.SparkContext._
@@ -13,7 +13,7 @@
import grizzled.slf4j.Logger
-case class DataSourceParams(appId: Int) extends Params
+case class DataSourceParams(appName: String) extends Params
class DataSource(val dsp: DataSourceParams)
extends PDataSource[TrainingData,
@@ -23,10 +23,10 @@
override
def readTraining(sc: SparkContext): TrainingData = {
- val eventsDb = Storage.getPEvents()
+
// read all events of EVENT involving ENTITY_TYPE and TARGET_ENTITY_TYPE
- val eventsRDD: RDD[Event] = eventsDb.find(
- appId = dsp.appId,
+ val eventsRDD: RDD[Event] = PEventStore.find(
+ appName = dsp.appName,
entityType = Some("ENTITY_TYPE"),
eventNames = Some(List("EVENT")),
targetEntityType = Some(Some("TARGET_ENTITY_TYPE")))(sc)