Beginnings of synchronizing notifications in the notification database.

content/browser/notifications/platform_notification_context_impl.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/browser/notifications/platform_notification_context_impl.h"
 
 #include "base/bind_helpers.h"
 #include "base/files/file_util.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "content/browser/notifications/notification_database.h"
 #include "content/browser/service_worker/service_worker_context_wrapper.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/notification_database_data.h"
+#include "content/public/browser/platform_notification_service.h"
 
 using base::DoNothing;
 
 namespace content {
 
 // Name of the directory in the user's profile directory where the notification
 // database files should be stored.
 const base::FilePath::CharType kPlatformNotificationsDirectory[] =
     FILE_PATH_LITERAL("Platform Notifications");
 
 PlatformNotificationContextImpl::PlatformNotificationContextImpl(
     const base::FilePath& path,
+    BrowserContext* browser_context,
     const scoped_refptr<ServiceWorkerContextWrapper>& service_worker_context)
     : path_(path),
+      browser_context_(browser_context),
       service_worker_context_(service_worker_context) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 }
 
 PlatformNotificationContextImpl::~PlatformNotificationContextImpl() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 
   // If the database has been initialized, it must be deleted on the task runner
   // thread as closing it may cause file I/O.
   if (database_) {
     DCHECK(task_runner_);
     task_runner_->DeleteSoon(FROM_HERE, database_.release());
   }
 }
 
 void PlatformNotificationContextImpl::Initialize() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  PlatformNotificationService* service =
+      GetContentClient()->browser()->GetPlatformNotificationService();
+  if (service) {
+    std::set<std::string> displayed_notifications;
+
+    bool notification_synchronization_supported =
+        service->GetDisplayedPersistentNotifications(browser_context_,
+                                                     &displayed_notifications);
+
+    // Synchronize the notifications stored in the database with the set of
+    // displaying notifications in |displayed_notifications|. This is necessary
+    // because flakiness may cause a platform to inform Chrome of a notification
+    // that has since been closed, or because the platform does not support
+    // notifications that exceed the lifetime of the browser process.
+
+    // TODO(peter): Synchronizing the actual notifications will be done when the
+    // persistent notification ids are stable. For M44 we need to support the
+    // case where there may be no notifications after a Chrome restart.
+    if (notification_synchronization_supported &&
+        !displayed_notifications.size()) {
+      prune_database_on_open_ = true;
+    }
+  }
+
   BrowserThread::PostTask(
       BrowserThread::IO,
       FROM_HERE,
       base::Bind(&PlatformNotificationContextImpl::InitializeOnIO, this));
 }
 
 void PlatformNotificationContextImpl::InitializeOnIO() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 
   // |service_worker_context_| may be NULL in tests.
@@ skipping 267 matching lines @@
     return;
   }
 
   database_.reset(new NotificationDatabase(GetDatabasePath()));
   NotificationDatabase::Status status =
       database_->Open(true /* create_if_missing */);
 
   UMA_HISTOGRAM_ENUMERATION("Notifications.Database.OpenResult",
                             status, NotificationDatabase::STATUS_COUNT);
 
+  // TODO(peter): Do finer-grained synchronization here.
+  if (prune_database_on_open_) {
+    prune_database_on_open_ = false;
+    DestroyDatabase();
+
+    database_.reset(new NotificationDatabase(GetDatabasePath()));
+    status = database_->Open(true /* create_if_missing */);
+
+    // TODO(peter): Find the appropriate UMA to cover in regards to
+    // synchronizing notifications after the implementation is complete.
+  }
+
   // When the database could not be opened due to corruption, destroy it, blow
   // away the contents of the directory and try re-opening the database.
   if (status == NotificationDatabase::STATUS_ERROR_CORRUPTED) {
     if (DestroyDatabase()) {
+      database_.reset(new NotificationDatabase(GetDatabasePath()));
       status = database_->Open(true /* create_if_missing */);
 
       UMA_HISTOGRAM_ENUMERATION(
           "Notifications.Database.OpenAfterCorruptionResult",
           status, NotificationDatabase::STATUS_COUNT);
     }
   }
 
   if (status == NotificationDatabase::STATUS_OK) {
     success_closure.Run();
@@ skipping 32 matching lines @@
 
   return path_.Append(kPlatformNotificationsDirectory);
 }
 
 void PlatformNotificationContextImpl::SetTaskRunnerForTesting(
     const scoped_refptr<base::SequencedTaskRunner>& task_runner) {
   task_runner_ = task_runner;
 }
 
 }  // namespace content