[PATCH]support virtual table statistics

src/maat_virtual.c

@@ -13,6 +13,7 @@
 #include "maat_kv.h"
 #include "maat_utils.h"
 #include "log/log.h"
+#include "alignment.h"
 #include "maat_rule.h"
 #include "maat_table.h"
 
@@ -24,6 +25,15 @@ struct virtual_schema {
     struct table_manager *ref_tbl_mgr;
 };
 
+struct virtual_runtime {
+    size_t n_worker_thread;
+    long long *scan_times;
+    long long *scan_bytes;
+    long long *scan_cpu_time;
+    long long *hit_times;
+    long long *hit_item_num;
+};
+
 void *virtual_schema_new(cJSON *json, struct table_manager *tbl_mgr, 
                          const char *table_name, struct log_handle *logger)
 {
@@ -66,6 +76,178 @@ void virtual_schema_free(void *virtual_schema)
     FREE(virtual_schema);
 }
 
+void *virtual_runtime_new(void *virtual_schema, size_t max_thread_num,
+                          struct maat_garbage_bin *garbage_bin,
+                          struct log_handle *logger)
+{
+    if (NULL == virtual_schema) {
+        return NULL;
+    }
+
+    struct virtual_runtime *virt_rt = ALLOC(struct virtual_runtime, 1);
+
+    virt_rt->n_worker_thread = max_thread_num;
+    virt_rt->scan_times = alignment_int64_array_alloc(max_thread_num);
+    virt_rt->scan_bytes = alignment_int64_array_alloc(max_thread_num);
+    virt_rt->scan_cpu_time = alignment_int64_array_alloc(max_thread_num);
+    virt_rt->hit_times = alignment_int64_array_alloc(max_thread_num);
+    virt_rt->hit_item_num = alignment_int64_array_alloc(max_thread_num);
+
+    return virt_rt;
+}
+
+void virtual_runtime_free(void *virtual_runtime)
+{
+    if (NULL == virtual_runtime) {
+        return;
+    }
+
+    struct virtual_runtime *virt_rt = (struct virtual_runtime *)virtual_runtime; 
+
+    if (virt_rt->scan_times != NULL) {
+        alignment_int64_array_free(virt_rt->scan_times);
+        virt_rt->scan_times = NULL;
+    }
+
+    if (virt_rt->scan_bytes != NULL) {
+        alignment_int64_array_free(virt_rt->scan_bytes);
+        virt_rt->scan_bytes = NULL;
+    }
+
+    if (virt_rt->scan_cpu_time != NULL) {
+        alignment_int64_array_free(virt_rt->scan_cpu_time);
+        virt_rt->scan_cpu_time = NULL;
+    }
+
+    if (virt_rt->hit_times != NULL) {
+        alignment_int64_array_free(virt_rt->hit_times);
+        virt_rt->hit_times = NULL;
+    }
+
+    if (virt_rt->hit_item_num != NULL) {
+        alignment_int64_array_free(virt_rt->hit_item_num);
+        virt_rt->hit_item_num = NULL;
+    }
+
+    FREE(virt_rt);
+}
+
+void virtual_runtime_scan_bytes_add(struct virtual_runtime *virt_rt,
+                                    int thread_id, long long val)
+{
+    if (NULL == virt_rt || thread_id < 0) {
+        return;
+    } 
+
+    alignment_int64_array_add(virt_rt->scan_bytes, thread_id, val);
+}
+
+long long virtual_runtime_scan_bytes(void *virtual_runtime)
+{
+    if (NULL == virtual_runtime) {
+        return 0;
+    }
+
+    struct virtual_runtime *virt_rt = (struct virtual_runtime *)virtual_runtime; 
+    long long sum = alignment_int64_array_sum(virt_rt->scan_bytes,
+                                              virt_rt->n_worker_thread);
+    alignment_int64_array_reset(virt_rt->scan_bytes,
+                                virt_rt->n_worker_thread);
+
+    return sum; 
+}
+
+void virtual_runtime_scan_times_inc(struct virtual_runtime *virt_rt,
+                                    int thread_id)
+{
+    if (NULL == virt_rt || thread_id < 0) {
+        return;
+    }
+
+    alignment_int64_array_add(virt_rt->scan_times, thread_id, 1);
+}
+
+long long virtual_runtime_scan_times(void *virtual_runtime)
+{
+    if (NULL == virtual_runtime) {
+        return 0;
+    }
+
+    struct virtual_runtime *virt_rt = (struct virtual_runtime *)virtual_runtime; 
+    long long sum = alignment_int64_array_sum(virt_rt->scan_times,
+                                              virt_rt->n_worker_thread);
+    alignment_int64_array_reset(virt_rt->scan_times,
+                                virt_rt->n_worker_thread);
+
+    return sum; 
+}
+
+long long virtual_runtime_scan_cpu_time(void *virtual_runtime)
+{
+    if (NULL == virtual_runtime) {
+        return 0;
+    }
+
+    struct virtual_runtime *virt_rt = (struct virtual_runtime *)virtual_runtime; 
+    long long sum = alignment_int64_array_sum(virt_rt->scan_cpu_time,
+                                              virt_rt->n_worker_thread);
+    alignment_int64_array_reset(virt_rt->scan_cpu_time,
+                                virt_rt->n_worker_thread);
+
+    return sum; 
+}
+
+void virtual_runtime_hit_times_inc(struct virtual_runtime *virt_rt,
+                                   int thread_id)
+{
+    if (NULL == virt_rt || thread_id < 0) {
+        return;
+    }
+
+    alignment_int64_array_add(virt_rt->hit_times, thread_id, 1);
+}
+
+long long virtual_runtime_hit_times(void *virtual_runtime)
+{
+    if (NULL == virtual_runtime) {
+        return 0;
+    }
+
+    struct virtual_runtime *virt_rt = (struct virtual_runtime *)virtual_runtime;
+    long long sum = alignment_int64_array_sum(virt_rt->hit_times,
+                                              virt_rt->n_worker_thread);
+    alignment_int64_array_reset(virt_rt->hit_times,
+                                virt_rt->n_worker_thread);
+
+    return sum;
+}
+
+void virtual_runtime_hit_item_num_add(void *virtual_runtime, int thread_id,
+                                      long long val)
+{
+    if (NULL == virtual_runtime) {
+        return;
+    }
+
+    struct virtual_runtime *virt_rt = (struct virtual_runtime *)virtual_runtime; 
+    alignment_int64_array_add(virt_rt->hit_item_num, thread_id, val);
+}
+
+long long virtual_runtime_hit_item_num(void *virtual_runtime)
+{
+    if (NULL == virtual_runtime) {
+        return 0;
+    } 
+
+    struct virtual_runtime *virt_rt = (struct virtual_runtime *)virtual_runtime; 
+    long long sum = alignment_int64_array_sum(virt_rt->hit_item_num,
+                                              virt_rt->n_worker_thread);
+    alignment_int64_array_reset(virt_rt->hit_item_num,
+                                virt_rt->n_worker_thread);
+    
+    return sum;
+}
+
 static int virtual_table_get_physical_table_id(void *virtual_schema)
 {
     if (NULL == virtual_schema) {