mirror of
https://github.com/OrcaSlicer/OrcaSlicer.git
synced 2026-07-28 13:22:06 +00:00
fix: windows OTA profile update race on startup
This commit is contained in:
@@ -5,7 +5,6 @@
|
|||||||
#include <boost/nowide/fstream.hpp>
|
#include <boost/nowide/fstream.hpp>
|
||||||
#include <functional>
|
#include <functional>
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
#include <mutex>
|
|
||||||
#include <set>
|
#include <set>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
@@ -204,7 +203,6 @@ struct PresetUpdater::priv
|
|||||||
|
|
||||||
// Per-vendor update checking
|
// Per-vendor update checking
|
||||||
std::set<std::string> checked_vendors;
|
std::set<std::string> checked_vendors;
|
||||||
std::mutex vendor_check_mutex;
|
|
||||||
std::vector<std::thread> vendor_check_threads;
|
std::vector<std::thread> vendor_check_threads;
|
||||||
std::atomic<bool> vendor_check_cancel{false};
|
std::atomic<bool> vendor_check_cancel{false};
|
||||||
|
|
||||||
@@ -221,10 +219,10 @@ struct PresetUpdater::priv
|
|||||||
priv();
|
priv();
|
||||||
|
|
||||||
void set_download_prefs(AppConfig *app_config);
|
void set_download_prefs(AppConfig *app_config);
|
||||||
bool get_file(const std::string &url, const fs::path &target_path) const;
|
bool get_file(const std::string &url, const fs::path &target_path) const;
|
||||||
//BBS: refine preset update logic
|
//BBS: refine preset update logic
|
||||||
bool extract_file(const fs::path &source_path, const fs::path &dest_path = {});
|
bool extract_file(const fs::path &source_path, const fs::path &dest_path = {});
|
||||||
void prune_tmps() const;
|
void prune_tmp(const std::string& vendor_id) const;
|
||||||
void sync_version() const;
|
void sync_version() const;
|
||||||
void parse_version_string(const std::string& body) const;
|
void parse_version_string(const std::string& body) const;
|
||||||
void sync_resources(std::string http_url, std::map<std::string, Resource> &resources, bool check_patch = false, std::string current_version="", std::string changelog_file="");
|
void sync_resources(std::string http_url, std::map<std::string, Resource> &resources, bool check_patch = false, std::string current_version="", std::string changelog_file="");
|
||||||
@@ -371,22 +369,14 @@ bool PresetUpdater::priv::extract_file(const fs::path &source_path, const fs::pa
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Remove leftover paritally downloaded files, if any.
|
// Remove a leftover partial archive for the vendor about to be synchronized.
|
||||||
void PresetUpdater::priv::prune_tmps() const
|
void PresetUpdater::priv::prune_tmp(const std::string& vendor_id) const
|
||||||
{
|
{
|
||||||
boost::system::error_code ec;
|
boost::system::error_code ec;
|
||||||
auto dir_it = boost::filesystem::directory_iterator(cache_path, ec);
|
const fs::path tmp_path = cache_path / (vendor_id + TMP_EXTENSION);
|
||||||
if (ec) {
|
fs::remove(tmp_path, ec);
|
||||||
BOOST_LOG_TRIVIAL(warning) << "[Orca Updater]failed to list cache dir " << cache_path.string() << ": " << ec.message();
|
if (ec)
|
||||||
return;
|
BOOST_LOG_TRIVIAL(warning) << "[Orca Updater]failed to remove " << tmp_path.string() << ": " << ec.message();
|
||||||
}
|
|
||||||
for (auto &dir_entry : dir_it)
|
|
||||||
if (is_plain_file(dir_entry) && dir_entry.path().extension() == TMP_EXTENSION) {
|
|
||||||
BOOST_LOG_TRIVIAL(debug) << "[Orca Updater]remove old cached files: " << dir_entry.path().string();
|
|
||||||
fs::remove(dir_entry.path(), ec);
|
|
||||||
if (ec)
|
|
||||||
BOOST_LOG_TRIVIAL(warning) << "[Orca Updater]failed to remove " << dir_entry.path().string() << ": " << ec.message();
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
//BBS: refine the Preset Updater logic
|
//BBS: refine the Preset Updater logic
|
||||||
@@ -1346,49 +1336,29 @@ PresetUpdater::~PresetUpdater()
|
|||||||
|
|
||||||
//BBS: change directories by design
|
//BBS: change directories by design
|
||||||
//BBS: refine the preset updater logic
|
//BBS: refine the preset updater logic
|
||||||
void PresetUpdater::sync(std::string http_url, std::string language, std::string plugin_version, PresetBundle *preset_bundle)
|
void PresetUpdater::sync(std::string http_url, std::string language, std::string plugin_version, PresetBundle * /*preset_bundle*/)
|
||||||
{
|
{
|
||||||
//p->set_download_prefs(GUI::wxGetApp().app_config);
|
//p->set_download_prefs(GUI::wxGetApp().app_config);
|
||||||
if (!p->enabled_version_check && !p->enabled_config_update) { return; }
|
if (!p->enabled_version_check && !p->enabled_config_update) { return; }
|
||||||
|
|
||||||
// Copy the whole vendors data for use in the background thread
|
p->thread = std::thread([this, http_url, language, plugin_version]() {
|
||||||
// Unfortunatelly as of C++11, it needs to be copied again
|
try {
|
||||||
// into the closure (but perhaps the compiler can elide this).
|
this->p->sync_version();
|
||||||
VendorMap vendors = preset_bundle ? preset_bundle->vendors : VendorMap{};
|
if (p->cancel)
|
||||||
|
return;
|
||||||
// Determine active vendor before entering the thread
|
// Vendor profile updates are triggered by check_vendor_update()
|
||||||
std::string active_vendor;
|
// after the startup printer preset has been restored.
|
||||||
if (preset_bundle) {
|
this->p->sync_plugins(http_url, plugin_version);
|
||||||
const Preset& printer = preset_bundle->printers.get_edited_preset();
|
this->p->sync_printer_config(http_url);
|
||||||
if (printer.vendor)
|
//if (p->cancel)
|
||||||
active_vendor = printer.vendor->id;
|
// return;
|
||||||
}
|
//remove the tooltip currently
|
||||||
|
//this->p->sync_tooltip(http_url, language);
|
||||||
p->thread = std::thread([this, vendors, active_vendor, http_url, language, plugin_version]() {
|
} catch (const std::exception &e) {
|
||||||
this->p->prune_tmps();
|
BOOST_LOG_TRIVIAL(error) << "[Orca Updater] background sync failed: " << e.what();
|
||||||
if (p->cancel)
|
} catch (...) {
|
||||||
return;
|
BOOST_LOG_TRIVIAL(error) << "[Orca Updater] background sync failed with an unknown exception";
|
||||||
this->p->sync_version();
|
}
|
||||||
if (p->cancel)
|
|
||||||
return;
|
|
||||||
// Per-vendor config check for the active vendor at startup
|
|
||||||
if (!active_vendor.empty() && !vendors.empty()) {
|
|
||||||
this->p->sync_vendor_config(active_vendor);
|
|
||||||
if (p->cancel)
|
|
||||||
return;
|
|
||||||
{
|
|
||||||
std::lock_guard<std::mutex> lock(this->p->vendor_check_mutex);
|
|
||||||
this->p->checked_vendors.insert(active_vendor);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (p->cancel)
|
|
||||||
return;
|
|
||||||
this->p->sync_plugins(http_url, plugin_version);
|
|
||||||
this->p->sync_printer_config(http_url);
|
|
||||||
//if (p->cancel)
|
|
||||||
// return;
|
|
||||||
//remove the tooltip currently
|
|
||||||
//this->p->sync_tooltip(http_url, language);
|
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1397,16 +1367,17 @@ void PresetUpdater::check_vendor_update(const std::string& vendor_id)
|
|||||||
if (!p->enabled_config_update) return;
|
if (!p->enabled_config_update) return;
|
||||||
if (vendor_id.empty()) return;
|
if (vendor_id.empty()) return;
|
||||||
|
|
||||||
std::lock_guard<std::mutex> lock(p->vendor_check_mutex);
|
|
||||||
|
|
||||||
if (!p->checked_vendors.insert(vendor_id).second)
|
if (!p->checked_vendors.insert(vendor_id).second)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
p->vendor_check_threads.emplace_back([this, vendor_id]() {
|
p->vendor_check_threads.emplace_back([this, vendor_id]() {
|
||||||
try {
|
try {
|
||||||
|
this->p->prune_tmp(vendor_id);
|
||||||
this->p->sync_vendor_config(vendor_id);
|
this->p->sync_vendor_config(vendor_id);
|
||||||
} catch (const std::exception& e) {
|
} catch (const std::exception& e) {
|
||||||
BOOST_LOG_TRIVIAL(error) << "[Orca Updater] vendor update failed for " << vendor_id << ": " << e.what();
|
BOOST_LOG_TRIVIAL(error) << "[Orca Updater] vendor update failed for " << vendor_id << ": " << e.what();
|
||||||
|
} catch (...) {
|
||||||
|
BOOST_LOG_TRIVIAL(error) << "[Orca Updater] vendor update failed for " << vendor_id << " with an unknown exception";
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user