| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/metrics/stats_table.h" | 5 #include "base/metrics/stats_table.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/process_util.h" | 8 #include "base/process_util.h" |
| 9 #include "base/scoped_ptr.h" | 9 #include "base/scoped_ptr.h" |
| 10 #include "base/shared_memory.h" | 10 #include "base/shared_memory.h" |
| (...skipping 260 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 271 tls_index_.Free(); | 271 tls_index_.Free(); |
| 272 | 272 |
| 273 // Cleanup our shared memory. | 273 // Cleanup our shared memory. |
| 274 delete impl_; | 274 delete impl_; |
| 275 | 275 |
| 276 // If we are the global table, unregister ourselves. | 276 // If we are the global table, unregister ourselves. |
| 277 if (global_table_ == this) | 277 if (global_table_ == this) |
| 278 global_table_ = NULL; | 278 global_table_ = NULL; |
| 279 } | 279 } |
| 280 | 280 |
| 281 int StatsTable::GetSlot() const { |
| 282 TLSData* data = GetTLSData(); |
| 283 if (!data) |
| 284 return 0; |
| 285 return data->slot; |
| 286 } |
| 287 |
| 281 int StatsTable::RegisterThread(const std::string& name) { | 288 int StatsTable::RegisterThread(const std::string& name) { |
| 282 int slot = 0; | 289 int slot = 0; |
| 283 if (!impl_) | 290 if (!impl_) |
| 284 return 0; | 291 return 0; |
| 285 | 292 |
| 286 // Registering a thread requires that we lock the shared memory | 293 // Registering a thread requires that we lock the shared memory |
| 287 // so that two threads don't grab the same slot. Fortunately, | 294 // so that two threads don't grab the same slot. Fortunately, |
| 288 // thread creation shouldn't happen in inner loops. | 295 // thread creation shouldn't happen in inner loops. |
| 289 { | 296 { |
| 290 SharedMemoryAutoLock lock(impl_->shared_memory()); | 297 SharedMemoryAutoLock lock(impl_->shared_memory()); |
| (...skipping 13 matching lines...) Expand all Loading... |
| 304 } | 311 } |
| 305 | 312 |
| 306 // Set our thread local storage. | 313 // Set our thread local storage. |
| 307 TLSData* data = new TLSData; | 314 TLSData* data = new TLSData; |
| 308 data->table = this; | 315 data->table = this; |
| 309 data->slot = slot; | 316 data->slot = slot; |
| 310 tls_index_.Set(data); | 317 tls_index_.Set(data); |
| 311 return slot; | 318 return slot; |
| 312 } | 319 } |
| 313 | 320 |
| 314 StatsTable::TLSData* StatsTable::GetTLSData() const { | 321 int StatsTable::CountThreadsRegistered() const { |
| 315 TLSData* data = | 322 if (!impl_) |
| 316 static_cast<TLSData*>(tls_index_.Get()); | 323 return 0; |
| 317 if (!data) | 324 |
| 325 // Loop through the shared memory and count the threads that are active. |
| 326 // We intentionally do not lock the table during the operation. |
| 327 int count = 0; |
| 328 for (int index = 1; index <= impl_->max_threads(); index++) { |
| 329 char* name = impl_->thread_name(index); |
| 330 if (*name != '\0') |
| 331 count++; |
| 332 } |
| 333 return count; |
| 334 } |
| 335 |
| 336 int StatsTable::FindCounter(const std::string& name) { |
| 337 // Note: the API returns counters numbered from 1..N, although |
| 338 // internally, the array is 0..N-1. This is so that we can return |
| 339 // zero as "not found". |
| 340 if (!impl_) |
| 341 return 0; |
| 342 |
| 343 // Create a scope for our auto-lock. |
| 344 { |
| 345 AutoLock scoped_lock(counters_lock_); |
| 346 |
| 347 // Attempt to find the counter. |
| 348 CountersMap::const_iterator iter; |
| 349 iter = counters_.find(name); |
| 350 if (iter != counters_.end()) |
| 351 return iter->second; |
| 352 } |
| 353 |
| 354 // Counter does not exist, so add it. |
| 355 return AddCounter(name); |
| 356 } |
| 357 |
| 358 int* StatsTable::GetLocation(int counter_id, int slot_id) const { |
| 359 if (!impl_) |
| 360 return NULL; |
| 361 if (slot_id > impl_->max_threads()) |
| 318 return NULL; | 362 return NULL; |
| 319 | 363 |
| 320 DCHECK(data->slot); | 364 int* row = impl_->row(counter_id); |
| 321 DCHECK_EQ(data->table, this); | 365 return &(row[slot_id-1]); |
| 322 return data; | 366 } |
| 367 |
| 368 const char* StatsTable::GetRowName(int index) const { |
| 369 if (!impl_) |
| 370 return NULL; |
| 371 |
| 372 return impl_->counter_name(index); |
| 373 } |
| 374 |
| 375 int StatsTable::GetRowValue(int index) const { |
| 376 return GetRowValue(index, 0); |
| 377 } |
| 378 |
| 379 int StatsTable::GetRowValue(int index, int pid) const { |
| 380 if (!impl_) |
| 381 return 0; |
| 382 |
| 383 int rv = 0; |
| 384 int* row = impl_->row(index); |
| 385 for (int slot_id = 0; slot_id < impl_->max_threads(); slot_id++) { |
| 386 if (pid == 0 || *impl_->thread_pid(slot_id) == pid) |
| 387 rv += row[slot_id]; |
| 388 } |
| 389 return rv; |
| 390 } |
| 391 |
| 392 int StatsTable::GetCounterValue(const std::string& name) { |
| 393 return GetCounterValue(name, 0); |
| 394 } |
| 395 |
| 396 int StatsTable::GetCounterValue(const std::string& name, int pid) { |
| 397 if (!impl_) |
| 398 return 0; |
| 399 |
| 400 int row = FindCounter(name); |
| 401 if (!row) |
| 402 return 0; |
| 403 return GetRowValue(row, pid); |
| 404 } |
| 405 |
| 406 int StatsTable::GetMaxCounters() const { |
| 407 if (!impl_) |
| 408 return 0; |
| 409 return impl_->max_counters(); |
| 410 } |
| 411 |
| 412 int StatsTable::GetMaxThreads() const { |
| 413 if (!impl_) |
| 414 return 0; |
| 415 return impl_->max_threads(); |
| 416 } |
| 417 |
| 418 int* StatsTable::FindLocation(const char* name) { |
| 419 // Get the static StatsTable |
| 420 StatsTable *table = StatsTable::current(); |
| 421 if (!table) |
| 422 return NULL; |
| 423 |
| 424 // Get the slot for this thread. Try to register |
| 425 // it if none exists. |
| 426 int slot = table->GetSlot(); |
| 427 if (!slot && !(slot = table->RegisterThread(""))) |
| 428 return NULL; |
| 429 |
| 430 // Find the counter id for the counter. |
| 431 std::string str_name(name); |
| 432 int counter = table->FindCounter(str_name); |
| 433 |
| 434 // Now we can find the location in the table. |
| 435 return table->GetLocation(counter, slot); |
| 323 } | 436 } |
| 324 | 437 |
| 325 void StatsTable::UnregisterThread() { | 438 void StatsTable::UnregisterThread() { |
| 326 UnregisterThread(GetTLSData()); | 439 UnregisterThread(GetTLSData()); |
| 327 } | 440 } |
| 328 | 441 |
| 329 void StatsTable::UnregisterThread(TLSData* data) { | 442 void StatsTable::UnregisterThread(TLSData* data) { |
| 330 if (!data) | 443 if (!data) |
| 331 return; | 444 return; |
| 332 DCHECK(impl_); | 445 DCHECK(impl_); |
| (...skipping 11 matching lines...) Expand all Loading... |
| 344 // This is called by the TLS destructor, which on some platforms has | 457 // This is called by the TLS destructor, which on some platforms has |
| 345 // already cleared the TLS info, so use the tls_data argument | 458 // already cleared the TLS info, so use the tls_data argument |
| 346 // rather than trying to fetch it ourselves. | 459 // rather than trying to fetch it ourselves. |
| 347 TLSData* tls_data = static_cast<TLSData*>(data); | 460 TLSData* tls_data = static_cast<TLSData*>(data); |
| 348 if (tls_data) { | 461 if (tls_data) { |
| 349 DCHECK(tls_data->table); | 462 DCHECK(tls_data->table); |
| 350 tls_data->table->UnregisterThread(tls_data); | 463 tls_data->table->UnregisterThread(tls_data); |
| 351 } | 464 } |
| 352 } | 465 } |
| 353 | 466 |
| 354 int StatsTable::CountThreadsRegistered() const { | |
| 355 if (!impl_) | |
| 356 return 0; | |
| 357 | |
| 358 // Loop through the shared memory and count the threads that are active. | |
| 359 // We intentionally do not lock the table during the operation. | |
| 360 int count = 0; | |
| 361 for (int index = 1; index <= impl_->max_threads(); index++) { | |
| 362 char* name = impl_->thread_name(index); | |
| 363 if (*name != '\0') | |
| 364 count++; | |
| 365 } | |
| 366 return count; | |
| 367 } | |
| 368 | |
| 369 int StatsTable::GetSlot() const { | |
| 370 TLSData* data = GetTLSData(); | |
| 371 if (!data) | |
| 372 return 0; | |
| 373 return data->slot; | |
| 374 } | |
| 375 | |
| 376 int StatsTable::FindEmptyThread() const { | 467 int StatsTable::FindEmptyThread() const { |
| 377 // Note: the API returns slots numbered from 1..N, although | 468 // Note: the API returns slots numbered from 1..N, although |
| 378 // internally, the array is 0..N-1. This is so that we can return | 469 // internally, the array is 0..N-1. This is so that we can return |
| 379 // zero as "not found". | 470 // zero as "not found". |
| 380 // | 471 // |
| 381 // The reason for doing this is because the thread 'slot' is stored | 472 // The reason for doing this is because the thread 'slot' is stored |
| 382 // in TLS, which is always initialized to zero, not -1. If 0 were | 473 // in TLS, which is always initialized to zero, not -1. If 0 were |
| 383 // returned as a valid slot number, it would be confused with the | 474 // returned as a valid slot number, it would be confused with the |
| 384 // uninitialized state. | 475 // uninitialized state. |
| 385 if (!impl_) | 476 if (!impl_) |
| (...skipping 25 matching lines...) Expand all Loading... |
| 411 for (int index = 1; index <= impl_->max_counters(); index++) { | 502 for (int index = 1; index <= impl_->max_counters(); index++) { |
| 412 char* row_name = impl_->counter_name(index); | 503 char* row_name = impl_->counter_name(index); |
| 413 if (!*row_name && !free_slot) | 504 if (!*row_name && !free_slot) |
| 414 free_slot = index; // save that we found a free slot | 505 free_slot = index; // save that we found a free slot |
| 415 else if (!strncmp(row_name, name.c_str(), kMaxCounterNameLength)) | 506 else if (!strncmp(row_name, name.c_str(), kMaxCounterNameLength)) |
| 416 return index; | 507 return index; |
| 417 } | 508 } |
| 418 return free_slot; | 509 return free_slot; |
| 419 } | 510 } |
| 420 | 511 |
| 421 int StatsTable::FindCounter(const std::string& name) { | |
| 422 // Note: the API returns counters numbered from 1..N, although | |
| 423 // internally, the array is 0..N-1. This is so that we can return | |
| 424 // zero as "not found". | |
| 425 if (!impl_) | |
| 426 return 0; | |
| 427 | |
| 428 // Create a scope for our auto-lock. | |
| 429 { | |
| 430 AutoLock scoped_lock(counters_lock_); | |
| 431 | |
| 432 // Attempt to find the counter. | |
| 433 CountersMap::const_iterator iter; | |
| 434 iter = counters_.find(name); | |
| 435 if (iter != counters_.end()) | |
| 436 return iter->second; | |
| 437 } | |
| 438 | |
| 439 // Counter does not exist, so add it. | |
| 440 return AddCounter(name); | |
| 441 } | |
| 442 | |
| 443 int StatsTable::AddCounter(const std::string& name) { | 512 int StatsTable::AddCounter(const std::string& name) { |
| 444 if (!impl_) | 513 if (!impl_) |
| 445 return 0; | 514 return 0; |
| 446 | 515 |
| 447 int counter_id = 0; | 516 int counter_id = 0; |
| 448 { | 517 { |
| 449 // To add a counter to the shared memory, we need the | 518 // To add a counter to the shared memory, we need the |
| 450 // shared memory lock. | 519 // shared memory lock. |
| 451 SharedMemoryAutoLock lock(impl_->shared_memory()); | 520 SharedMemoryAutoLock lock(impl_->shared_memory()); |
| 452 | 521 |
| (...skipping 10 matching lines...) Expand all Loading... |
| 463 } | 532 } |
| 464 | 533 |
| 465 // now add to our in-memory cache | 534 // now add to our in-memory cache |
| 466 { | 535 { |
| 467 AutoLock lock(counters_lock_); | 536 AutoLock lock(counters_lock_); |
| 468 counters_[name] = counter_id; | 537 counters_[name] = counter_id; |
| 469 } | 538 } |
| 470 return counter_id; | 539 return counter_id; |
| 471 } | 540 } |
| 472 | 541 |
| 473 int* StatsTable::GetLocation(int counter_id, int slot_id) const { | 542 StatsTable::TLSData* StatsTable::GetTLSData() const { |
| 474 if (!impl_) | 543 TLSData* data = |
| 475 return NULL; | 544 static_cast<TLSData*>(tls_index_.Get()); |
| 476 if (slot_id > impl_->max_threads()) | 545 if (!data) |
| 477 return NULL; | 546 return NULL; |
| 478 | 547 |
| 479 int* row = impl_->row(counter_id); | 548 DCHECK(data->slot); |
| 480 return &(row[slot_id-1]); | 549 DCHECK_EQ(data->table, this); |
| 481 } | 550 return data; |
| 482 | |
| 483 const char* StatsTable::GetRowName(int index) const { | |
| 484 if (!impl_) | |
| 485 return NULL; | |
| 486 | |
| 487 return impl_->counter_name(index); | |
| 488 } | |
| 489 | |
| 490 int StatsTable::GetRowValue(int index, int pid) const { | |
| 491 if (!impl_) | |
| 492 return 0; | |
| 493 | |
| 494 int rv = 0; | |
| 495 int* row = impl_->row(index); | |
| 496 for (int slot_id = 0; slot_id < impl_->max_threads(); slot_id++) { | |
| 497 if (pid == 0 || *impl_->thread_pid(slot_id) == pid) | |
| 498 rv += row[slot_id]; | |
| 499 } | |
| 500 return rv; | |
| 501 } | |
| 502 | |
| 503 int StatsTable::GetRowValue(int index) const { | |
| 504 return GetRowValue(index, 0); | |
| 505 } | |
| 506 | |
| 507 int StatsTable::GetCounterValue(const std::string& name, int pid) { | |
| 508 if (!impl_) | |
| 509 return 0; | |
| 510 | |
| 511 int row = FindCounter(name); | |
| 512 if (!row) | |
| 513 return 0; | |
| 514 return GetRowValue(row, pid); | |
| 515 } | |
| 516 | |
| 517 int StatsTable::GetCounterValue(const std::string& name) { | |
| 518 return GetCounterValue(name, 0); | |
| 519 } | |
| 520 | |
| 521 int StatsTable::GetMaxCounters() const { | |
| 522 if (!impl_) | |
| 523 return 0; | |
| 524 return impl_->max_counters(); | |
| 525 } | |
| 526 | |
| 527 int StatsTable::GetMaxThreads() const { | |
| 528 if (!impl_) | |
| 529 return 0; | |
| 530 return impl_->max_threads(); | |
| 531 } | |
| 532 | |
| 533 int* StatsTable::FindLocation(const char* name) { | |
| 534 // Get the static StatsTable | |
| 535 StatsTable *table = StatsTable::current(); | |
| 536 if (!table) | |
| 537 return NULL; | |
| 538 | |
| 539 // Get the slot for this thread. Try to register | |
| 540 // it if none exists. | |
| 541 int slot = table->GetSlot(); | |
| 542 if (!slot && !(slot = table->RegisterThread(""))) | |
| 543 return NULL; | |
| 544 | |
| 545 // Find the counter id for the counter. | |
| 546 std::string str_name(name); | |
| 547 int counter = table->FindCounter(str_name); | |
| 548 | |
| 549 // Now we can find the location in the table. | |
| 550 return table->GetLocation(counter, slot); | |
| 551 } | 551 } |
| 552 | 552 |
| 553 } // namespace base | 553 } // namespace base |
| OLD | NEW |