diff --git a/shell/browser/api/electron_api_web_frame_main.cc b/shell/browser/api/electron_api_web_frame_main.cc index d9ff7366aafb..40e3d811baf8 100644 --- a/shell/browser/api/electron_api_web_frame_main.cc +++ b/shell/browser/api/electron_api_web_frame_main.cc @@ -9,6 +9,7 @@ #include #include +#include "base/containers/map_util.h" #include "base/feature_list.h" #include "base/logging.h" #include "base/no_destructor.h" @@ -121,19 +122,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