chore(query): add log if does not have workload group (#18374)

* chore(query): add log if does not have workload group

* chore(query): add log if does not have workload group

Author: zhang2014

src/query/management/src/workload/workload_resource.rs

@@ -162,15 +162,9 @@ impl WorkloadGroupResourceManagerInner {
                 workload_resource.meta.quotas.get(MEMORY_QUOTA_KEY)
             {
                 self.percent_normalizer.update(*v);
-                self.update_mem_usage(&online_workload_group);
-            } else if let Some(QuotaValue::Bytes(v)) =
-                workload_resource.meta.quotas.get(MEMORY_QUOTA_KEY)
-            {
-                workload_resource
-                    .max_memory_usage
-                    .store(*v, Ordering::Relaxed);
             }
 
+            self.update_mem_usage(&online_workload_group);
             return Ok(workload_resource);
         };
 
@@ -182,46 +176,26 @@ impl WorkloadGroupResourceManagerInner {
             let old_percentage = old_workload_group.meta.quotas.get(MEMORY_QUOTA_KEY);
 
             match (old_percentage, new_percentage) {
-                (None, None) => {
-                    return Ok(workload_resource);
-                }
-                (None, Some(QuotaValue::Bytes(_v))) => {
-                    self.update_mem_usage(&online_workload_group);
-                    return Ok(workload_resource);
-                }
                 (None, Some(QuotaValue::Percentage(v))) => {
                     self.percent_normalizer.update(*v);
-                    self.update_mem_usage(&online_workload_group);
                 }
                 (Some(QuotaValue::Percentage(v)), None) => {
                     self.percent_normalizer.remove(*v);
-                    self.update_mem_usage(&online_workload_group);
                 }
                 (Some(QuotaValue::Percentage(v)), Some(QuotaValue::Bytes(_))) => {
                     self.percent_normalizer.remove(*v);
-                    self.update_mem_usage(&online_workload_group);
                 }
                 (Some(QuotaValue::Percentage(old)), Some(QuotaValue::Percentage(new))) => {
                     self.percent_normalizer.remove(*old);
                     self.percent_normalizer.update(*new);
-                    self.update_mem_usage(&online_workload_group);
-                }
-                (Some(QuotaValue::Bytes(_)), None) => {
-                    workload_resource
-                        .max_memory_usage
-                        .store(0, Ordering::Relaxed);
-                    return Ok(workload_resource);
                 }
                 (Some(QuotaValue::Bytes(_)), Some(QuotaValue::Percentage(v))) => {
                     self.percent_normalizer.update(*v);
-                    self.update_mem_usage(&online_workload_group);
-                }
-                (Some(QuotaValue::Bytes(_old)), Some(QuotaValue::Bytes(_new))) => {
-                    self.update_mem_usage(&online_workload_group);
-                    return Ok(workload_resource);
                 }
                 _ => {}
             }
+
+            self.update_mem_usage(&online_workload_group);
         }
 
         Ok(workload_resource)
@@ -259,7 +233,17 @@ impl WorkloadGroupResourceManagerInner {
                         .max_memory_usage
                         .store(memory_usage, Ordering::Relaxed);
                 } else {
-                    workload_group.max_memory_usage.store(0, Ordering::Relaxed)
+                    let limit = self.global_mem_stat.get_limit();
+                    let usage_ratio = workload_group.meta.get_max_memory_usage_ratio();
+                    let mut memory_usage = 0;
+
+                    if limit > 0 {
+                        memory_usage = limit as usize / 100 * usage_ratio;
+                    }
+
+                    workload_group
+                        .max_memory_usage
+                        .store(memory_usage, Ordering::Relaxed)
                 }
             }
         }

src/query/service/src/servers/http/v1/http_query_handlers.rs

@@ -541,6 +541,16 @@ pub(crate) async fn query_handler(
 
             parent_mem_stat = ParentMemStat::Normal(workload_group.mem_stat.clone());
             tracking_workload_group = Some(workload_group);
+        } else if let Ok(user) = session.get_current_user() {
+            log::info!(
+                "[Workload-Group] The user {} does not have a workload group.",
+                user.name
+            );
+        } else {
+            log::info!(
+                "[Workload-Group] The query {:?} does not have a workload group.",
+                session.get_current_query_id()
+            );
         }
 
         let name = Some(ctx.query_id.clone());