Add PSI full memory pressure soft/hard limit ratio to LoadShedConfiguration

Summary: Add PSI full memory pressure soft/hard limit ratio to LoadShedConfiguration

Differential Revision: D95131570

fbshipit-source-id: ade88f8b91527625a230170d98ad3cb084b3e319

Author: Rongxin Xia

wangle/acceptor/LoadShedConfiguration.cpp

@@ -95,6 +95,14 @@ void LoadShedConfiguration::checkIsSane(const SysParams& sysParams) const {
     CHECK_GE(udpMemSoftLimitRatio_, 0.0);
     CHECK_LE(udpMemSoftLimitRatio_, udpMemHardLimitRatio_);
 
+    // PSI full memory pressure soft/hard limit ratios must have values in
+    // the range of [0-1], inclusive, where the hard limit must be greater
+    // than or equal to the soft limit.
+    CHECK_GE(memPressureFullHardLimitRatio_, 0.0);
+    CHECK_LE(memPressureFullHardLimitRatio_, 1.0);
+    CHECK_GE(memPressureFullSoftLimitRatio_, 0.0);
+    CHECK_LE(memPressureFullSoftLimitRatio_, memPressureFullHardLimitRatio_);
+
     // Period must be greater than or equal to 0.
     CHECK_GE(period_.count(), std::chrono::milliseconds(0).count());
   }

wangle/acceptor/LoadShedConfiguration.h

@@ -269,6 +269,32 @@ class LoadShedConfiguration {
     return nicHardLimitRatio_;
   }
 
+  /**
+   * Set/get the PSI full memory pressure avg10 soft limit ratio.
+   * Various shedding protections should engage when above this limit.
+   */
+  void setMemPressureFullSoftLimitRatio(double limit) {
+    CHECK_GE(limit, 0.0);
+    CHECK_LE(limit, 1.0);
+    memPressureFullSoftLimitRatio_ = limit;
+  }
+  double getMemPressureFullSoftLimitRatio() const {
+    return memPressureFullSoftLimitRatio_;
+  }
+
+  /**
+   * Set/get the PSI full memory pressure avg10 hard limit ratio.
+   * More extreme shedding protections should engage when above this limit.
+   */
+  void setMemPressureFullHardLimitRatio(double limit) {
+    CHECK_GE(limit, 0.0);
+    CHECK_LE(limit, 1.0);
+    memPressureFullHardLimitRatio_ = limit;
+  }
+  double getMemPressureFullHardLimitRatio() const {
+    return memPressureFullHardLimitRatio_;
+  }
+
   void setLoadUpdatePeriod(std::chrono::milliseconds period) {
     period_ = period;
   }
@@ -325,6 +351,9 @@ class LoadShedConfiguration {
   double nicSoftLimitRatio_{1.0};
   double nicHardLimitRatio_{1.0};
 
+  double memPressureFullSoftLimitRatio_{1.0};
+  double memPressureFullHardLimitRatio_{1.0};
+
   std::chrono::milliseconds period_;
 
   bool loadSheddingEnabled_{true};

wangle/acceptor/test/LoadShedConfigurationTest.cpp

@@ -35,6 +35,12 @@ TEST(LoadShedConfigurationTest, TestSettersAndGetters) {
   lsc.setCpuHardLimitRatio(0.4);
   EXPECT_EQ(0.4, lsc.getCpuHardLimitRatio());
 
+  lsc.setMemPressureFullSoftLimitRatio(0.5);
+  EXPECT_EQ(0.5, lsc.getMemPressureFullSoftLimitRatio());
+
+  lsc.setMemPressureFullHardLimitRatio(0.6);
+  EXPECT_EQ(0.6, lsc.getMemPressureFullHardLimitRatio());
+
   EXPECT_EQ(0, lsc.getCpuUsageExceedWindowSize());
   lsc.setCpuUsageExceedWindowSize(12);
   EXPECT_EQ(12, lsc.getCpuUsageExceedWindowSize());