Remove use of util::opts from resource_thread

This commit is contained in:
Anthony Ramine 2016-07-27 09:44:47 +02:00
parent c1409af543
commit 1662e292d8
4 changed files with 43 additions and 21 deletions

View file

@ -42,13 +42,13 @@ use std::collections::HashMap;
use std::error::Error; use std::error::Error;
use std::fs::File; use std::fs::File;
use std::io::prelude::*; use std::io::prelude::*;
use std::path::Path; use std::ops::Deref;
use std::path::{Path, PathBuf};
use std::rc::Rc; use std::rc::Rc;
use std::sync::mpsc::{Receiver, Sender, channel}; use std::sync::mpsc::{Receiver, Sender, channel};
use std::sync::{Arc, RwLock}; use std::sync::{Arc, RwLock};
use storage_thread::StorageThreadFactory; use storage_thread::StorageThreadFactory;
use url::Url; use url::Url;
use util::opts;
use util::prefs::PREFS; use util::prefs::PREFS;
use util::thread::spawn_named; use util::thread::spawn_named;
use websocket_loader; use websocket_loader;
@ -165,10 +165,16 @@ fn start_sending_opt(start_chan: LoadConsumer, metadata: Metadata,
/// Returns a tuple of (public, private) senders to the new threads. /// Returns a tuple of (public, private) senders to the new threads.
pub fn new_resource_threads(user_agent: String, pub fn new_resource_threads(user_agent: String,
devtools_chan: Option<Sender<DevtoolsControlMsg>>, devtools_chan: Option<Sender<DevtoolsControlMsg>>,
profiler_chan: ProfilerChan) -> (ResourceThreads, ResourceThreads) { profiler_chan: ProfilerChan,
config_dir: Option<PathBuf>)
-> (ResourceThreads, ResourceThreads) {
let filemanager_chan: IpcSender<FileManagerThreadMsg> = FileManagerThreadFactory::new(TFD_PROVIDER); let filemanager_chan: IpcSender<FileManagerThreadMsg> = FileManagerThreadFactory::new(TFD_PROVIDER);
let (public_core, private_core) = new_core_resource_thread(user_agent, devtools_chan, let (public_core, private_core) = new_core_resource_thread(
profiler_chan, filemanager_chan.clone()); user_agent,
devtools_chan,
profiler_chan,
filemanager_chan.clone(),
config_dir);
let storage: IpcSender<StorageThreadMsg> = StorageThreadFactory::new(); let storage: IpcSender<StorageThreadMsg> = StorageThreadFactory::new();
(ResourceThreads::new(public_core, storage.clone(), filemanager_chan.clone()), (ResourceThreads::new(public_core, storage.clone(), filemanager_chan.clone()),
ResourceThreads::new(private_core, storage, filemanager_chan)) ResourceThreads::new(private_core, storage, filemanager_chan))
@ -179,7 +185,8 @@ pub fn new_resource_threads(user_agent: String,
pub fn new_core_resource_thread(user_agent: String, pub fn new_core_resource_thread(user_agent: String,
devtools_chan: Option<Sender<DevtoolsControlMsg>>, devtools_chan: Option<Sender<DevtoolsControlMsg>>,
profiler_chan: ProfilerChan, profiler_chan: ProfilerChan,
filemanager_chan: IpcSender<FileManagerThreadMsg>) filemanager_chan: IpcSender<FileManagerThreadMsg>,
config_dir: Option<PathBuf>)
-> (CoreResourceThread, CoreResourceThread) { -> (CoreResourceThread, CoreResourceThread) {
let (public_setup_chan, public_setup_port) = ipc::channel().unwrap(); let (public_setup_chan, public_setup_port) = ipc::channel().unwrap();
let (private_setup_chan, private_setup_port) = ipc::channel().unwrap(); let (private_setup_chan, private_setup_port) = ipc::channel().unwrap();
@ -192,6 +199,7 @@ pub fn new_core_resource_thread(user_agent: String,
let mut channel_manager = ResourceChannelManager { let mut channel_manager = ResourceChannelManager {
resource_manager: resource_manager, resource_manager: resource_manager,
config_dir: config_dir,
}; };
channel_manager.start(public_setup_chan_clone, channel_manager.start(public_setup_chan_clone,
private_setup_chan_clone, private_setup_chan_clone,
@ -202,14 +210,16 @@ pub fn new_core_resource_thread(user_agent: String,
} }
struct ResourceChannelManager { struct ResourceChannelManager {
resource_manager: CoreResourceManager resource_manager: CoreResourceManager,
config_dir: Option<PathBuf>,
} }
fn create_resource_groups() -> (ResourceGroup, ResourceGroup) { fn create_resource_groups(config_dir: Option<&Path>)
-> (ResourceGroup, ResourceGroup) {
let mut hsts_list = HstsList::from_servo_preload(); let mut hsts_list = HstsList::from_servo_preload();
let mut auth_cache = AuthCache::new(); let mut auth_cache = AuthCache::new();
let mut cookie_jar = CookieStorage::new(); let mut cookie_jar = CookieStorage::new();
if let Some(ref config_dir) = opts::get().config_dir { if let Some(config_dir) = config_dir {
read_json_from_file(&mut auth_cache, config_dir, "auth_cache.json"); read_json_from_file(&mut auth_cache, config_dir, "auth_cache.json");
read_json_from_file(&mut hsts_list, config_dir, "hsts_list.json"); read_json_from_file(&mut hsts_list, config_dir, "hsts_list.json");
read_json_from_file(&mut cookie_jar, config_dir, "cookie_jar.json"); read_json_from_file(&mut cookie_jar, config_dir, "cookie_jar.json");
@ -236,7 +246,8 @@ impl ResourceChannelManager {
private_control_sender: CoreResourceThread, private_control_sender: CoreResourceThread,
public_receiver: IpcReceiver<CoreResourceMsg>, public_receiver: IpcReceiver<CoreResourceMsg>,
private_receiver: IpcReceiver<CoreResourceMsg>) { private_receiver: IpcReceiver<CoreResourceMsg>) {
let (public_resource_group, private_resource_group) = create_resource_groups(); let (public_resource_group, private_resource_group) =
create_resource_groups(self.config_dir.as_ref().map(Deref::deref));
let mut rx_set = IpcReceiverSet::new().unwrap(); let mut rx_set = IpcReceiverSet::new().unwrap();
let private_id = rx_set.add(private_receiver).unwrap(); let private_id = rx_set.add(private_receiver).unwrap();
@ -297,7 +308,7 @@ impl ResourceChannelManager {
let _ = sender.send(()); let _ = sender.send(());
} }
CoreResourceMsg::Exit(sender) => { CoreResourceMsg::Exit(sender) => {
if let Some(ref config_dir) = opts::get().config_dir { if let Some(ref config_dir) = self.config_dir {
match group.auth_cache.read() { match group.auth_cache.read() {
Ok(auth_cache) => write_json_to_file(&*auth_cache, config_dir, "auth_cache.json"), Ok(auth_cache) => write_json_to_file(&*auth_cache, config_dir, "auth_cache.json"),
Err(_) => warn!("Error writing auth cache to disk"), Err(_) => warn!("Error writing auth cache to disk"),
@ -319,8 +330,10 @@ impl ResourceChannelManager {
} }
} }
pub fn read_json_from_file<T: Decodable>(data: &mut T, config_dir: &str, filename: &str) { pub fn read_json_from_file<T>(data: &mut T, config_dir: &Path, filename: &str)
let path = Path::new(config_dir).join(filename); where T: Decodable
{
let path = config_dir.join(filename);
let display = path.display(); let display = path.display();
let mut file = match File::open(&path) { let mut file = match File::open(&path) {
@ -346,13 +359,15 @@ pub fn read_json_from_file<T: Decodable>(data: &mut T, config_dir: &str, filenam
} }
} }
pub fn write_json_to_file<T: Encodable>(data: &T, config_dir: &str, filename: &str) { pub fn write_json_to_file<T>(data: &T, config_dir: &Path, filename: &str)
where T: Encodable
{
let json_encoded: String; let json_encoded: String;
match json::encode(&data) { match json::encode(&data) {
Ok(d) => json_encoded = d, Ok(d) => json_encoded = d,
Err(_) => return, Err(_) => return,
} }
let path = Path::new(config_dir).join(filename); let path = config_dir.join(filename);
let display = path.display(); let display = path.display();
let mut file = match File::create(&path) { let mut file = match File::create(&path) {

View file

@ -8,6 +8,7 @@ use resource_thread;
use std::borrow::ToOwned; use std::borrow::ToOwned;
use std::collections::BTreeMap; use std::collections::BTreeMap;
use std::collections::HashMap; use std::collections::HashMap;
use std::path::Path;
use url::Url; use url::Url;
use util::opts; use util::opts;
use util::thread::spawn_named; use util::thread::spawn_named;
@ -39,7 +40,8 @@ impl StorageManager {
fn new(port: IpcReceiver<StorageThreadMsg>) -> StorageManager { fn new(port: IpcReceiver<StorageThreadMsg>) -> StorageManager {
let mut local_data = HashMap::new(); let mut local_data = HashMap::new();
if let Some(ref config_dir) = opts::get().config_dir { if let Some(ref config_dir) = opts::get().config_dir {
resource_thread::read_json_from_file(&mut local_data, config_dir, "local_data.json"); resource_thread::read_json_from_file(
&mut local_data, Path::new(config_dir), "local_data.json");
} }
StorageManager { StorageManager {
port: port, port: port,
@ -76,7 +78,8 @@ impl StorageManager {
} }
StorageThreadMsg::Exit(sender) => { StorageThreadMsg::Exit(sender) => {
if let Some(ref config_dir) = opts::get().config_dir { if let Some(ref config_dir) = opts::get().config_dir {
resource_thread::write_json_to_file(&self.local_data, config_dir, "local_data.json"); resource_thread::write_json_to_file(
&self.local_data, Path::new(config_dir), "local_data.json");
} }
let _ = sender.send(()); let _ = sender.send(());
break break

View file

@ -239,7 +239,8 @@ fn create_constellation(opts: opts::Opts,
let (public_resource_threads, private_resource_threads) = let (public_resource_threads, private_resource_threads) =
new_resource_threads(opts.user_agent.clone(), new_resource_threads(opts.user_agent.clone(),
devtools_chan.clone(), devtools_chan.clone(),
time_profiler_chan.clone()); time_profiler_chan.clone(),
opts.config_dir.map(Into::into));
let image_cache_thread = new_image_cache_thread(public_resource_threads.sender(), let image_cache_thread = new_image_cache_thread(public_resource_threads.sender(),
webrender_api_sender.as_ref().map(|wr| wr.create_api())); webrender_api_sender.as_ref().map(|wr| wr.create_api()));
let font_cache_thread = FontCacheThread::new(public_resource_threads.sender(), let font_cache_thread = FontCacheThread::new(public_resource_threads.sender(),

View file

@ -41,7 +41,8 @@ fn test_exit() {
let (tx, _rx) = ipc::channel().unwrap(); let (tx, _rx) = ipc::channel().unwrap();
let (sender, receiver) = ipc::channel().unwrap(); let (sender, receiver) = ipc::channel().unwrap();
let filemanager_chan = FileManagerThreadFactory::new(TFD_PROVIDER); let filemanager_chan = FileManagerThreadFactory::new(TFD_PROVIDER);
let (resource_thread, _) = new_core_resource_thread("".to_owned(), None, ProfilerChan(tx), filemanager_chan); let (resource_thread, _) = new_core_resource_thread(
"".to_owned(), None, ProfilerChan(tx), filemanager_chan, None);
resource_thread.send(CoreResourceMsg::Exit(sender)).unwrap(); resource_thread.send(CoreResourceMsg::Exit(sender)).unwrap();
receiver.recv().unwrap(); receiver.recv().unwrap();
} }
@ -51,7 +52,8 @@ fn test_bad_scheme() {
let (tx, _rx) = ipc::channel().unwrap(); let (tx, _rx) = ipc::channel().unwrap();
let (sender, receiver) = ipc::channel().unwrap(); let (sender, receiver) = ipc::channel().unwrap();
let filemanager_chan = FileManagerThreadFactory::new(TFD_PROVIDER); let filemanager_chan = FileManagerThreadFactory::new(TFD_PROVIDER);
let (resource_thread, _) = new_core_resource_thread("".to_owned(), None, ProfilerChan(tx), filemanager_chan); let (resource_thread, _) = new_core_resource_thread(
"".to_owned(), None, ProfilerChan(tx), filemanager_chan, None);
let (start_chan, start) = ipc::channel().unwrap(); let (start_chan, start) = ipc::channel().unwrap();
let url = Url::parse("bogus://whatever").unwrap(); let url = Url::parse("bogus://whatever").unwrap();
resource_thread.send(CoreResourceMsg::Load(LoadData::new(LoadContext::Browsing, url, &ResourceTest), resource_thread.send(CoreResourceMsg::Load(LoadData::new(LoadContext::Browsing, url, &ResourceTest),
@ -231,7 +233,8 @@ fn test_cancelled_listener() {
let (tx, _rx) = ipc::channel().unwrap(); let (tx, _rx) = ipc::channel().unwrap();
let (exit_sender, exit_receiver) = ipc::channel().unwrap(); let (exit_sender, exit_receiver) = ipc::channel().unwrap();
let filemanager_chan = FileManagerThreadFactory::new(TFD_PROVIDER); let filemanager_chan = FileManagerThreadFactory::new(TFD_PROVIDER);
let (resource_thread, _) = new_core_resource_thread("".to_owned(), None, ProfilerChan(tx), filemanager_chan); let (resource_thread, _) = new_core_resource_thread(
"".to_owned(), None, ProfilerChan(tx), filemanager_chan, None);
let (sender, receiver) = ipc::channel().unwrap(); let (sender, receiver) = ipc::channel().unwrap();
let (id_sender, id_receiver) = ipc::channel().unwrap(); let (id_sender, id_receiver) = ipc::channel().unwrap();
let (sync_sender, sync_receiver) = ipc::channel().unwrap(); let (sync_sender, sync_receiver) = ipc::channel().unwrap();