Merge pull request #46 from layer5io/kumarabd/feature/fixces

updated imagehub filter

Author: kumarabd

api/main.go

@@ -33,7 +33,7 @@ func CORSMiddleware(next http.Handler) http.Handler {
 
 func hello(w http.ResponseWriter, req *http.Request) {
 	token := req.Header.Get("Authorization")
-	w.Write([]byte("Hello, " + token + ", welcome to dockercon 2020!"))
+	w.Write([]byte("Hello, " + token + ", welcome to Istiocon 2021!"))
 }
 
 func auth(w http.ResponseWriter, req *http.Request) {

rate-limit-filter/src/lib.rs

@@ -76,7 +76,7 @@ impl HttpContext for UpstreamCall {
     }
 
     fn on_http_response_headers(&mut self, _num_headers: usize) -> Action {
-        self.set_http_response_header("x-app-response", Some("reply from the filter"));
+        self.set_http_response_header("x-app-serving", Some("rate-limit-filter"));
         proxy_wasm::hostcalls::log(LogLevel::Debug, format!("RESPONDING").as_str()).ok();
         Action::Continue
     }

rate-limit-filter/src/rate_limiter.rs

@@ -14,8 +14,9 @@ pub struct RateLimiter {
 impl RateLimiter {
     fn new(key: &String, plan: &String) -> Self {
         let limit = match plan.as_str() {
-            "group" => Some(100),
-            "user" => Some(10),
+            "Enterprise" => Some(100),
+            "Team" => Some(50),
+            "Personal" => Some(10),
             _ => None,
         };
         Self {
@@ -31,8 +32,9 @@ impl RateLimiter {
                 let data: Option<Self> = bincode::deserialize(&data).unwrap_or(None);
                 if let Some(mut obj) = data {
                     let limit = match plan.as_str() {
-                        "group" => Some(100),
-                        "user" => Some(10),
+                        "Enterprise" => Some(100),
+                        "Team" => Some(50),
+                        "Personal" => Some(10),
                         _ => None,
                     };
                     obj.rpm = limit;