mirror of
https://github.com/Relintai/pandemonium_engine.git
synced 2025-01-14 14:41:15 +01:00
Refactored FileCache so it actually works with this setup.
This commit is contained in:
parent
0bbff6ae3b
commit
ead4b0594b
@ -2,12 +2,18 @@
|
|||||||
|
|
||||||
#include "core/os/dir_access.h"
|
#include "core/os/dir_access.h"
|
||||||
#include "core/os/file_access.h"
|
#include "core/os/file_access.h"
|
||||||
|
#include "core/os/os.h"
|
||||||
|
#include "core/print_string.h"
|
||||||
|
|
||||||
String FileCache::get_wwwroot() {
|
String FileCache::get_wwwroot() {
|
||||||
return wwwroot;
|
return _wwwroot_orig;
|
||||||
}
|
}
|
||||||
void FileCache::set_wwwroot(const String &val) {
|
void FileCache::set_wwwroot(const String &val) {
|
||||||
wwwroot = val;
|
_wwwroot_orig = val;
|
||||||
|
}
|
||||||
|
|
||||||
|
String FileCache::get_wwwroot_abs() {
|
||||||
|
return _wwwroot_orig;
|
||||||
}
|
}
|
||||||
|
|
||||||
int FileCache::get_cache_invalidation_time() {
|
int FileCache::get_cache_invalidation_time() {
|
||||||
@ -18,25 +24,67 @@ void FileCache::set_cache_invalidation_time(const int &val) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void FileCache::wwwroot_register_file(const String &file_path) {
|
void FileCache::wwwroot_register_file(const String &file_path) {
|
||||||
registered_files.insert(file_path);
|
RegisteredFileEntry e;
|
||||||
|
e.orig_path = file_path;
|
||||||
|
e.lowercase_path = file_path.to_lower();
|
||||||
|
|
||||||
|
_registered_files.push_back(e);
|
||||||
}
|
}
|
||||||
|
|
||||||
void FileCache::wwwroot_deregister_file(const String &file_path) {
|
void FileCache::wwwroot_deregister_file(const String &file_path) {
|
||||||
registered_files.erase(file_path);
|
for (int i = 0; i < _registered_files.size(); ++i) {
|
||||||
|
const RegisteredFileEntry &e = _registered_files[i];
|
||||||
|
|
||||||
|
if (file_path == e.orig_path) {
|
||||||
|
_registered_files.remove(i);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool FileCache::wwwroot_has_file(const String &file_path) {
|
bool FileCache::wwwroot_has_file(const String &file_path) {
|
||||||
return registered_files.has(file_path);
|
//return registered_files.has(file_path);
|
||||||
|
|
||||||
|
for (int i = 0; i < _registered_files.size(); ++i) {
|
||||||
|
const RegisteredFileEntry &e = _registered_files[i];
|
||||||
|
|
||||||
|
if (file_path == e.lowercase_path) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
int FileCache::wwwroot_get_file_index(const String &file_path) {
|
||||||
|
for (int i = 0; i < _registered_files.size(); ++i) {
|
||||||
|
const RegisteredFileEntry &e = _registered_files[i];
|
||||||
|
|
||||||
|
if (file_path == e.lowercase_path) {
|
||||||
|
return i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
String FileCache::wwwroot_get_file_orig_path(const int index) {
|
||||||
|
ERR_FAIL_INDEX_V(index, _registered_files.size(), "");
|
||||||
|
|
||||||
|
return _registered_files[index].orig_path;
|
||||||
}
|
}
|
||||||
|
|
||||||
void FileCache::wwwroot_refresh_cache() {
|
void FileCache::wwwroot_refresh_cache() {
|
||||||
_lock.write_lock();
|
_lock.write_lock();
|
||||||
|
|
||||||
registered_files.clear();
|
_registered_files.clear();
|
||||||
|
|
||||||
wwwroot.path_clean_end_slash();
|
if (_wwwroot_orig != "") {
|
||||||
|
_wwwroot = DirAccess::get_full_path(_wwwroot_orig, DirAccess::ACCESS_FILESYSTEM);
|
||||||
|
|
||||||
wwwroot_evaluate_dir(wwwroot);
|
_wwwroot = _wwwroot.path_clean_end_slash();
|
||||||
|
wwwroot_evaluate_dir(_wwwroot);
|
||||||
|
}
|
||||||
|
|
||||||
_lock.write_unlock();
|
_lock.write_unlock();
|
||||||
}
|
}
|
||||||
@ -57,8 +105,8 @@ void FileCache::wwwroot_evaluate_dir(const String &path, const bool should_exist
|
|||||||
|
|
||||||
if (!da->current_is_dir()) {
|
if (!da->current_is_dir()) {
|
||||||
String np = path + "/" + f;
|
String np = path + "/" + f;
|
||||||
np = np.substr(wwwroot.size(), np.size() - wwwroot.size());
|
np = np.substr(_wwwroot.size() - 1, np.size() - _wwwroot.size());
|
||||||
registered_files.insert(np);
|
wwwroot_register_file(np);
|
||||||
} else {
|
} else {
|
||||||
wwwroot_evaluate_dir(path + "/" + f);
|
wwwroot_evaluate_dir(path + "/" + f);
|
||||||
}
|
}
|
||||||
@ -162,7 +210,7 @@ void FileCache::set_cached_body(const String &path, const String &body) {
|
|||||||
void FileCache::clear() {
|
void FileCache::clear() {
|
||||||
_lock.write_lock();
|
_lock.write_lock();
|
||||||
|
|
||||||
registered_files.clear();
|
_registered_files.clear();
|
||||||
|
|
||||||
for (Map<String, CacheEntry *>::Element *E = cache_map.front(); E; E++) {
|
for (Map<String, CacheEntry *>::Element *E = cache_map.front(); E; E++) {
|
||||||
CacheEntry *ce = E->get();
|
CacheEntry *ce = E->get();
|
||||||
@ -182,7 +230,7 @@ FileCache::FileCache() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
FileCache::~FileCache() {
|
FileCache::~FileCache() {
|
||||||
registered_files.clear();
|
_registered_files.clear();
|
||||||
}
|
}
|
||||||
|
|
||||||
void FileCache::_bind_methods() {
|
void FileCache::_bind_methods() {
|
||||||
|
@ -1,11 +1,12 @@
|
|||||||
#ifndef FILE_CACHE_H
|
#ifndef FILE_CACHE_H
|
||||||
#define FILE_CACHE_H
|
#define FILE_CACHE_H
|
||||||
|
|
||||||
|
#include "core/hash_map.h"
|
||||||
#include "core/map.h"
|
#include "core/map.h"
|
||||||
#include "core/os/os.h"
|
#include "core/os/os.h"
|
||||||
#include "core/os/rw_lock.h"
|
#include "core/os/rw_lock.h"
|
||||||
#include "core/set.h"
|
|
||||||
#include "core/ustring.h"
|
#include "core/ustring.h"
|
||||||
|
#include "core/vector.h"
|
||||||
|
|
||||||
#include "core/reference.h"
|
#include "core/reference.h"
|
||||||
|
|
||||||
@ -16,6 +17,8 @@ public:
|
|||||||
String get_wwwroot();
|
String get_wwwroot();
|
||||||
void set_wwwroot(const String &val);
|
void set_wwwroot(const String &val);
|
||||||
|
|
||||||
|
String get_wwwroot_abs();
|
||||||
|
|
||||||
int get_cache_invalidation_time();
|
int get_cache_invalidation_time();
|
||||||
void set_cache_invalidation_time(const int &val);
|
void set_cache_invalidation_time(const int &val);
|
||||||
|
|
||||||
@ -24,6 +27,9 @@ public:
|
|||||||
void wwwroot_register_file(const String &file_path);
|
void wwwroot_register_file(const String &file_path);
|
||||||
void wwwroot_deregister_file(const String &file_path);
|
void wwwroot_deregister_file(const String &file_path);
|
||||||
bool wwwroot_has_file(const String &file_path);
|
bool wwwroot_has_file(const String &file_path);
|
||||||
|
//return -1 if does not exists
|
||||||
|
int wwwroot_get_file_index(const String &file_path);
|
||||||
|
String wwwroot_get_file_orig_path(const int index);
|
||||||
void wwwroot_refresh_cache();
|
void wwwroot_refresh_cache();
|
||||||
void wwwroot_evaluate_dir(const String &path, const bool should_exist = true);
|
void wwwroot_evaluate_dir(const String &path, const bool should_exist = true);
|
||||||
|
|
||||||
@ -37,11 +43,8 @@ public:
|
|||||||
FileCache();
|
FileCache();
|
||||||
~FileCache();
|
~FileCache();
|
||||||
|
|
||||||
String wwwroot;
|
|
||||||
uint64_t cache_invalidation_time;
|
uint64_t cache_invalidation_time;
|
||||||
|
|
||||||
Set<String> registered_files;
|
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
static void _bind_methods();
|
static void _bind_methods();
|
||||||
|
|
||||||
@ -56,6 +59,16 @@ protected:
|
|||||||
|
|
||||||
RWLock _lock;
|
RWLock _lock;
|
||||||
Map<String, CacheEntry *> cache_map;
|
Map<String, CacheEntry *> cache_map;
|
||||||
|
|
||||||
|
String _wwwroot_orig;
|
||||||
|
String _wwwroot;
|
||||||
|
|
||||||
|
struct RegisteredFileEntry {
|
||||||
|
String orig_path;
|
||||||
|
String lowercase_path;
|
||||||
|
};
|
||||||
|
|
||||||
|
Vector<RegisteredFileEntry> _registered_files;
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
@ -15,9 +15,11 @@ void WebRoot::set_www_root_path(const String &val) {
|
|||||||
_www_root_path = val;
|
_www_root_path = val;
|
||||||
|
|
||||||
if (val == "") {
|
if (val == "") {
|
||||||
|
_www_root_file_cache->set_wwwroot(val);
|
||||||
_www_root_file_cache->clear();
|
_www_root_file_cache->clear();
|
||||||
} else {
|
} else {
|
||||||
_www_root_file_cache->wwwroot_evaluate_dir(val);
|
_www_root_file_cache->set_wwwroot(val);
|
||||||
|
_www_root_file_cache->wwwroot_refresh_cache();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -134,9 +136,12 @@ bool WebRoot::process_middlewares(Ref<WebServerRequest> request) {
|
|||||||
|
|
||||||
bool WebRoot::try_send_wwwroot_file(Ref<WebServerRequest> request) {
|
bool WebRoot::try_send_wwwroot_file(Ref<WebServerRequest> request) {
|
||||||
String path = request->get_path_full();
|
String path = request->get_path_full();
|
||||||
|
path = path.to_lower();
|
||||||
|
|
||||||
if (_www_root_file_cache->wwwroot_has_file(path)) {
|
int file_indx = _www_root_file_cache->wwwroot_get_file_index(path);
|
||||||
send_file(path, request);
|
|
||||||
|
if (file_indx != -1) {
|
||||||
|
send_file(_www_root_file_cache->wwwroot_get_file_orig_path(file_indx), request);
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@ -145,7 +150,7 @@ bool WebRoot::try_send_wwwroot_file(Ref<WebServerRequest> request) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void WebRoot::send_file(const String &path, Ref<WebServerRequest> request) {
|
void WebRoot::send_file(const String &path, Ref<WebServerRequest> request) {
|
||||||
String fp = _www_root_file_cache->wwwroot + path;
|
String fp = _www_root_file_cache->get_wwwroot_abs() + path;
|
||||||
|
|
||||||
request->send_file(fp);
|
request->send_file(fp);
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user