优化时间检查代码

Hu Chunming
1 parent 1e5961fe
Showing 4 changed files with 34 additions and 28 deletions
src/ai_platform/mvpt.cpp
src/helpers/time_helper.hpp
src/util/util_tools.cpp
src/util/util_tools.h
@@ -6,6 +6,7 @@
 #include <mutex>
 #include <chrono>
 #include <thread>
+#include <ctime>
 #include "../decoder/interface/DecoderManager.h"
 #include "../decoder/interface/utiltools.hpp"
@@ -615,17 +616,13 @@ bool CMultiSourceProcess::save_obj_pic(ObjectInfo&amp; res_obj)
 }
 bool CMultiSourceProcess::CheckTime() {
-    struct tm* info;
-    int nYear, nMonth, nDay;
-    time_t raw;
-    time(&raw);
-    info = localtime(&raw);
-    nYear = info->tm_year + 1900;
-    nMonth = info->tm_mon + 1;
-    nDay = info->tm_mday;
-    if (nYear == 2026 && nMonth <= 3)
+ 
+    long long deadline_ts = helpers::timer::get_ts_by_day_ms(2026, 4, 1);
+    long long cur_ts = helpers::timer::get_cur_time_ms();
+    
+    if (cur_ts < deadline_ts)
     {
-        return true;
+      return true;
     }
     else
     {
@@ -17,6 +17,8 @@
 #include <thread>
 #include <unordered_map>
 #include <spdlog/spdlog.h>
+#include <ctime>
+
 using namespace std;
@@ -57,6 +59,31 @@ namespace helpers
             return std::string(buf);
         }
+        static long long get_ts_by_day_ms(int year, int month, int day)
+        {
+            // 初始化tm结构体
+            std::tm t = {};
+            t.tm_year = year - 1900; // 年从1900年开始计数
+            t.tm_mon = month - 1;            // 月份从0开始计数，3月是第4个月
+            t.tm_mday = day;          // 日
+            t.tm_hour = 0;          // 小时
+            t.tm_min = 0;           // 分钟
+            t.tm_sec = 0;            // 秒
+        
+            // 使用mktime转换为time_t（如果需要）
+            std::time_t timestamp = std::mktime(&t); // 或者直接使用下面的chrono方法跳过这一步
+            if (timestamp == -1) {
+                return -1; // 或者抛出异常等错误处理方式
+            }
+        
+            // 使用chrono转换（如果不需要先转换为time_t）
+            auto timepoint = std::chrono::system_clock::from_time_t(timestamp);
+            auto duration = timepoint.time_since_epoch();
+            auto ms = std::chrono::duration_cast<std::chrono::milliseconds>(duration).count();
+    
+            return ms;
+        }
+
     }  // namespace timer
 }  // namespace helpers
-#include "util_tools.h"
-
-using namespace std;
-
-namespace UtilTools{
-    long long get_cur_time_ms(){
-        chrono::time_point<chrono::system_clock, chrono::milliseconds> tpMicro
-            = chrono::time_point_cast<chrono::milliseconds>(chrono::system_clock::now());
-
-        return tpMicro.time_since_epoch().count();
-    }
-}
 \ No newline at end of file
-#include <chrono>
-
-
-namespace UtilTools{
-    long long get_cur_time_ms();
-}