Add force update of UserStats.

2 files changed, 14 insertions, 3 deletions

diff --git a/editor/user_stats.cpp b/editor/user_stats.cpp
index 629f4543a9..4b0a9290e7 100644
--- a/editor/user_stats.cpp
+++ b/editor/user_stats.cpp
@@ -127,12 +127,14 @@ bool UserStatsLoader::Update(string const & userName)
   return true;
 }
 
-void UserStatsLoader::Update(string const & userName, TOnUpdateCallback fn)
+void UserStatsLoader::Update(string const & userName, UpdatePolicy const policy,
+                             TOnUpdateCallback fn)
 {
   auto nothingToUpdate = false;
+  if (policy == UpdatePolicy::Lazy)
   {
     lock_guard<mutex> g(m_mutex);
-    nothingToUpdate = m_userStats && m_userName == userName && m_userStats &&
+    nothingToUpdate = m_userStats && m_userName == userName &&
                       difftime(m_lastUpdate, time(nullptr)) < kSecondsInHour;
   }
 
@@ -148,6 +150,11 @@ void UserStatsLoader::Update(string const & userName, TOnUpdateCallback fn)
   }).detach();
 }
 
+void UserStatsLoader::Update(string const & userName, TOnUpdateCallback fn)
+{
+  Update(userName, UpdatePolicy::Lazy, fn);
+}
+
 void UserStatsLoader::DropStats(string const & userName)
 {
   lock_guard<mutex> g(m_mutex);
diff --git a/editor/user_stats.hpp b/editor/user_stats.hpp
index 200330ec23..4b71882736 100644
--- a/editor/user_stats.hpp
+++ b/editor/user_stats.hpp
@@ -39,13 +39,17 @@ class UserStatsLoader
 public:
   using TOnUpdateCallback = function<void()>;
 
+  enum class UpdatePolicy { Lazy, Force };
+
   UserStatsLoader();
 
   /// Synchronously sends request to the server. Updates stats and returns true on success.
   bool Update(string const & userName);
 
-  /// Launch the update process if stats are too old.
+  /// Launches the update process if stats are too old or if policy is UpdatePolicy::Force.
   /// The process posts fn to a gui thread on success.
+  void Update(string const & userName, UpdatePolicy policy, TOnUpdateCallback fn);
+  /// Calls Update with UpdatePolicy::Lazy.
   void Update(string const & userName, TOnUpdateCallback fn);
 
   /// Resets internal state and removes records from settings.