| // Copyright 2021 The Bazel Authors. All rights reserved. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #include <notify.h> |
| |
| #include "src/main/native/darwin/util.h" |
| #include "src/main/native/macros.h" |
| #include "src/main/native/unix_jni.h" |
| |
| namespace blaze_jni { |
| |
| void portable_start_memory_pressure_monitoring() { |
| // To test use: |
| // /usr/bin/log stream -level debug \ |
| // --predicate '(subsystem == "build.bazel")' |
| // sudo memory_pressure -S -l warn |
| // sudo memory_pressure -S -l critical |
| // or use the test notifications that we register. |
| static dispatch_once_t once_token; |
| dispatch_once(&once_token, ^{ |
| dispatch_queue_t queue = bazel::darwin::JniDispatchQueue(); |
| dispatch_source_t source = dispatch_source_create( |
| DISPATCH_SOURCE_TYPE_MEMORYPRESSURE, 0, |
| DISPATCH_MEMORYPRESSURE_WARN | DISPATCH_MEMORYPRESSURE_CRITICAL, queue); |
| CHECK(source != nullptr); |
| dispatch_source_set_event_handler(source, ^{ |
| dispatch_source_memorypressure_flags_t pressureLevel = |
| dispatch_source_get_data(source); |
| if (pressureLevel == DISPATCH_MEMORYPRESSURE_WARN) { |
| log_if_possible("memory pressure warning anomaly"); |
| memory_pressure_callback(MemoryPressureLevelWarning); |
| } else if (pressureLevel == DISPATCH_MEMORYPRESSURE_CRITICAL) { |
| log_if_possible("memory pressure critical anomaly"); |
| memory_pressure_callback(MemoryPressureLevelCritical); |
| } else { |
| log_if_possible("error: unknown memory pressure critical level: %d", |
| (int)pressureLevel); |
| } |
| }); |
| dispatch_resume(source); |
| // These are registered solely so we can test the system from end-to-end. |
| // Using the Apple memory_pressure requires admin access. |
| int testToken; |
| int32_t status = notify_register_dispatch( |
| "com.google.bazel.test.memorypressurelevel.warning", &testToken, queue, |
| ^(int state) { |
| log_if_possible("memory pressure test warning anomaly"); |
| memory_pressure_callback(MemoryPressureLevelWarning); |
| }); |
| CHECK(status == NOTIFY_STATUS_OK); |
| status = notify_register_dispatch( |
| "com.google.bazel.test.memorypressurelevel.critical", &testToken, queue, |
| ^(int state) { |
| log_if_possible("memory pressure test critical anomaly"); |
| memory_pressure_callback(MemoryPressureLevelCritical); |
| }); |
| CHECK(status == NOTIFY_STATUS_OK); |
| log_if_possible("memory pressure monitoring registered"); |
| }); |
| } |
| |
| } // namespace blaze_jni |