✨ solve DegreeStatic algorithm config problem (#77)

fbpcchen · web-flow · commit 3fc59558a657 · 2023-08-04T14:55:24.000+08:00
diff --git a/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/Main.scala b/nebula-algorithm/src/main/scala/com/vesoft/nebula/algorithm/Main.scala
@@ -22,7 +22,8 @@ import com.vesoft.nebula.algorithm.config.{
   Node2vecConfig,
   PRConfig,
   ShortestPathConfig,
-  SparkConfig
+  SparkConfig,
+  DegreeStaticConfig
 }
 import com.vesoft.nebula.algorithm.lib.{
   BetweennessCentralityAlgo,
@@ -171,7 +172,8 @@ object Main {
           ShortestPathAlgo(spark, dataSet, spConfig, hasWeight)
         }
         case "degreestatic" => {
-          DegreeStaticAlgo(spark, dataSet)
+          val dsConfig = DegreeStaticConfig.getDegreeStaticConfig(configs)
+          DegreeStaticAlgo(spark, dataSet, dsConfig)
         }
         case "kcore" => {
           val kCoreConfig = KCoreConfig.getKCoreConfig(configs)

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,8 @@ import com.vesoft.nebula.algorithm.config.{`
`22`	`22`	`Node2vecConfig,`
`23`	`23`	`PRConfig,`
`24`	`24`	`ShortestPathConfig,`
`25`		`- SparkConfig`
	`25`	`+ SparkConfig,`
	`26`	`+ DegreeStaticConfig`
`26`	`27`	`}`
`27`	`28`	`import com.vesoft.nebula.algorithm.lib.{`
`28`	`29`	`BetweennessCentralityAlgo,`
`@@ -171,7 +172,8 @@ object Main {`
`171`	`172`	`ShortestPathAlgo(spark, dataSet, spConfig, hasWeight)`
`172`	`173`	`}`
`173`	`174`	`case "degreestatic" => {`
`174`		`- DegreeStaticAlgo(spark, dataSet)`
	`175`	`+ val dsConfig = DegreeStaticConfig.getDegreeStaticConfig(configs)`
	`176`	`+ DegreeStaticAlgo(spark, dataSet, dsConfig)`
`175`	`177`	`}`
`176`	`178`	`case "kcore" => {`
`177`	`179`	`val kCoreConfig = KCoreConfig.getKCoreConfig(configs)`