Browse Source

refactor: use base's map utills in WebFrameMain (#45759)

refactor: use base::FindPtrOrNull() in WebFrameMain::FromFrameTreeNodeId()

refactor: use base::FindPtrOrNull() in WebFrameMain::FromFrameToken()

Co-authored-by: trop[bot] <37223003+trop[bot]@users.noreply.github.com>
Co-authored-by: Charles Kerr <[email protected]>
trop[bot] 1 month ago
parent
commit
31a95f7857
1 changed files with 3 additions and 8 deletions
  1. 3 8
      shell/browser/api/electron_api_web_frame_main.cc

+ 3 - 8
shell/browser/api/electron_api_web_frame_main.cc

@@ -9,6 +9,7 @@
 #include <utility>
 #include <vector>
 
+#include "base/containers/map_util.h"
 #include "base/feature_list.h"
 #include "base/logging.h"
 #include "base/no_destructor.h"
@@ -122,19 +123,13 @@ FrameTokenMap& GetFrameTokenMap() {
 // static
 WebFrameMain* WebFrameMain::FromFrameTreeNodeId(
     content::FrameTreeNodeId frame_tree_node_id) {
-  FrameTreeNodeIdMap& frame_map = GetFrameTreeNodeIdMap();
-  auto iter = frame_map.find(frame_tree_node_id);
-  auto* web_frame = iter == frame_map.end() ? nullptr : iter->second;
-  return web_frame;
+  return base::FindPtrOrNull(GetFrameTreeNodeIdMap(), frame_tree_node_id);
 }
 
 // static
 WebFrameMain* WebFrameMain::FromFrameToken(
     content::GlobalRenderFrameHostToken frame_token) {
-  FrameTokenMap& frame_map = GetFrameTokenMap();
-  auto iter = frame_map.find(frame_token);
-  auto* web_frame = iter == frame_map.end() ? nullptr : iter->second;
-  return web_frame;
+  return base::FindPtrOrNull(GetFrameTokenMap(), frame_token);
 }
 
 // static