registered new Offsets and predictors

Signed-off-by: Lehmann_Fabian <[email protected]>
CommonWorkflowScheduler · Mar 20, 2024 · 37e4fa6 · 37e4fa6
1 parent daa51c9
commit 37e4fa6
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/main/java/cws/k8s/scheduler/prediction/MemoryScaler.java b/src/main/java/cws/k8s/scheduler/prediction/MemoryScaler.java
@@ -7,7 +7,12 @@
 import cws.k8s.scheduler.prediction.extractor.MemoryExtractor;
 import cws.k8s.scheduler.prediction.offset.MaxOffset;
 import cws.k8s.scheduler.prediction.offset.PercentileOffset;
+import cws.k8s.scheduler.prediction.offset.StandardDeviationOffset;
+import cws.k8s.scheduler.prediction.offset.VarianceOffset;
+import cws.k8s.scheduler.prediction.predictor.ConstantNumberPredictor;
 import cws.k8s.scheduler.prediction.predictor.LinearPredictor;
+import cws.k8s.scheduler.prediction.predictor.PolynomialPredictor;
+import cws.k8s.scheduler.prediction.predictor.PonderingPredictor;
 import io.fabric8.kubernetes.api.builder.Builder;
 import lombok.extern.slf4j.Slf4j;
 
@@ -74,6 +79,10 @@ private Builder<Predictor> applyOffset( final Builder<Predictor> builder, final
                 return () -> new MaxOffset( builder.build() );
             } else if ( offsetValue.equals( "none" ) ) {
                 return builder;
+            } else if ( offsetValue.equals( "var" ) ) {
+                return () -> new VarianceOffset( builder.build() );
+            } else if ( offsetValue.equals( "std" ) ) {
+                return () -> new StandardDeviationOffset( builder.build() );
             } else {
                 throw new IllegalArgumentException("unrecognized offset parameter: " + offsetValue );
             }
@@ -88,6 +97,19 @@ private Builder<Predictor> applyPredictor( String predictorString ) {
         if ( predictorString.equalsIgnoreCase( "linear" )) {
             log.debug( "using LinearPredictor" );
             return () -> new LinearPredictor( inputExtractor, outputExtractor );
+        } else if ( predictorString.equalsIgnoreCase( "ponder" )) {
+            log.debug( "using PonderingPredictor" );
+            return () -> new PonderingPredictor( inputExtractor, outputExtractor );
+        } else if ( predictorString.toLowerCase().startsWith( "const" ) ) {
+            final String substring = predictorString.substring( "const".length() );
+            final long value = substring.length() == 0 ? 0 : Long.parseLong( substring );
+            log.debug( "using ConstantPredictor with value: {}", value );
+            return () -> new ConstantNumberPredictor( outputExtractor, value );
+        } else if ( predictorString.toLowerCase().startsWith( "poly" ) ) {
+            final String substring = predictorString.substring( "poly".length() );
+            final int value = substring.length() == 0 ? 0 : Integer.parseInt( substring );
+            log.debug( "using PolyPredictor with value: {}", value );
+            return () -> new PolynomialPredictor( inputExtractor, outputExtractor, value );
         } else {
             throw new IllegalArgumentException("unrecognized memoryPredictorString: " + predictorString);
         }