net/socket_stream/socket_stream_job_manager.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "net/socket_stream/socket_stream_job_manager.h"
   6
   7 #include "base/memory/singleton.h"
   8
   9 namespace net {
  10
  11 SocketStreamJobManager::SocketStreamJobManager() {
  12 }
  13
  14 SocketStreamJobManager::~SocketStreamJobManager() {
  15 }
  16
  17 // static
  18 SocketStreamJobManager* SocketStreamJobManager::GetInstance() {
  19   return Singleton<SocketStreamJobManager>::get();
  20 }
  21
  22 SocketStreamJob* SocketStreamJobManager::CreateJob(
  23     const GURL& url, SocketStream::Delegate* delegate,
  24     URLRequestContext* context, CookieStore* cookie_store) const {
  25   // If url is invalid, create plain SocketStreamJob, which will close
  26   // the socket immediately.
  27   if (!url.is_valid()) {
  28     SocketStreamJob* job = new SocketStreamJob();
  29     job->InitSocketStream(new SocketStream(url, delegate, context,
  30                                            cookie_store));
  31     return job;
  32   }
  33
  34   const std::string& scheme = url.scheme();  // already lowercase
  35
  36   base::AutoLock locked(lock_);
  37   FactoryMap::const_iterator found = factories_.find(scheme);
  38   if (found != factories_.end()) {
  39     SocketStreamJob* job = found->second(url, delegate, context, cookie_store);
  40     if (job)
  41       return job;
  42   }
  43   SocketStreamJob* job = new SocketStreamJob();
  44   job->InitSocketStream(new SocketStream(url, delegate, context, cookie_store));
  45   return job;
  46 }
  47
  48 SocketStreamJob::ProtocolFactory*
  49 SocketStreamJobManager::RegisterProtocolFactory(
  50     const std::string& scheme, SocketStreamJob::ProtocolFactory* factory) {
  51   base::AutoLock locked(lock_);
  52
  53   SocketStreamJob::ProtocolFactory* old_factory;
  54   FactoryMap::iterator found = factories_.find(scheme);
  55   if (found != factories_.end()) {
  56     old_factory = found->second;
  57   } else {
  58     old_factory = NULL;
  59   }
  60   if (factory) {
  61     factories_[scheme] = factory;
  62   } else if (found != factories_.end()) {
  63     factories_.erase(found);
  64   }
  65   return old_factory;
  66 }
  67
  68 }  // namespace net