Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(246)

Unified Diff: tools/win/CreateTempFilesPerfEvaluation/CreateTempFilesPerfEval.cc

Issue 2810333008: A tool to evaluate methods in creating temp files (Closed)
Patch Set: Better printf to make the output more readable Created 3 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: tools/win/CreateTempFilesPerfEvaluation/CreateTempFilesPerfEval.cc
diff --git a/tools/win/CreateTempFilesPerfEvaluation/CreateTempFilesPerfEval.cc b/tools/win/CreateTempFilesPerfEvaluation/CreateTempFilesPerfEval.cc
new file mode 100644
index 0000000000000000000000000000000000000000..c427ff9f57e517d65a084d2c5abab5a647983daa
--- /dev/null
+++ b/tools/win/CreateTempFilesPerfEvaluation/CreateTempFilesPerfEval.cc
@@ -0,0 +1,272 @@
+// Copyright (c) 2017 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 <windows.h>
+
+#include <shlwapi.h>
+#include <stdio.h>
+#include <tchar.h>
+
+#include <algorithm>
+#include <iostream>
+#include <sstream>
+#include <string>
+
+#pragma warning(disable : 4996)
+
+// Create |count| number of temp files at |folder_path| using GUID based method.
+// The time cost in millisecond of creating every 500 temp files or all the
+// files if |count| < 500 is printed to the console.
+bool CreateFilesUsingGuid(UINT count, const char* folder_path);
+
+// Create |count| number of temp files at |folder_path| using GetTempFileName()
+// API. The time cost in millisecond of creating every 500 temp files or all the
+// files if |count| < 500 is printed to the console.
+bool CreateFilesUsingGetTempFileName(UINT count, const char* folder_path);
+
+// This method converts GUID to a string.
+char* ConvertGuidToString(const GUID* id, char* out);
+
+// If |folder_path| doesn't exist, creat it, otherwise check if it is empty.
+bool CreateOrValidateTempDirectory(const char* folder_path);
+
+// Deletes all the content in |folder_path|.
+void DeleteDirectoryContent(const char* folder_path);
+
+// Deletes |folder_path| including its contents and the raw directory.
+bool DeleteDirectory(const char* folder_path);
+
+// Deletes |folder_path| including its contents and the raw directory, and print
+// the delete status to the console.
+void DeleteDirectoryAndPrintMsg(const char* folder_path);
+
+// Prints the elapsed time at current step for the latest cycle.
+void FormatPrintElapsedTime(UINT cur_step,
+ UINT total_step,
+ const LARGE_INTEGER& elapsed_ms);
+
+// Maximum number of temp files allowed to create. This is limited by the
+// implementation of GetTempFileName().
+// "This limits GetTempFileName to a maximum of 65,535 unique file names if the
+// lpPathName and lpPrefixString parameters remain the same."
+// https://msdn.microsoft.com/en-us/library/windows/desktop/aa364991(v=vs.85).aspx
+UINT kMaxFileCreate = 65535;
+
+// Query the time cost each time when this amount of temp files are created.
+UINT kFileCountPerMetric = 500;
+
+int main() {
+ // Gets the temp path env string.
+ DWORD temp_path_ret = 0;
+ CHAR temp_folder_path[MAX_PATH];
+ temp_path_ret = ::GetTempPathA(MAX_PATH, temp_folder_path);
+ if (temp_path_ret > MAX_PATH || temp_path_ret == 0) {
+ std::cout << "GetTempPath failed" << std::endl;
+ return 0;
+ }
+
+ // A temporary directory where the new temp files created by GetTempFileName()
+ // are written.
+ std::string temp_dir_gettempfilename(
+ std::string(temp_folder_path).append("TempDirGetTempFileName\\"));
+
+ // A temporary directory where the new temp files created by Guid-based method
+ // are written.
+ std::string temp_dir_guid(
+ std::string(temp_folder_path).append("TempDirGuid\\"));
+
+ UINT file_create_count;
+ std::string user_input;
+
+ while (true) {
+ std::cout << "\nPlease enter # of files to create (maximum "
+ << kMaxFileCreate << "), or \"quit\" to end the program : ";
+ std::getline(std::cin, user_input);
+
+ std::transform(user_input.begin(), user_input.end(), user_input.begin(),
+ ::tolower);
+ if (user_input == "quit")
+ break;
+
+ std::cout << std::endl;
+ std::stringstream ss(user_input);
+
+ if (ss >> file_create_count && file_create_count <= kMaxFileCreate) {
+ std::cout << "\nPlease select method to create temp file names,\n"
+ << "\"t\" for GetTempFileName \n"
+ << "\"g\" for GUID-based \n"
+ << "\"b\" for both \n"
+ << "or \"quit\" to end the program : ";
+ std::getline(std::cin, user_input);
+
+ std::transform(user_input.begin(), user_input.end(), user_input.begin(),
+ ::tolower);
+ if (user_input == "quit")
+ break;
+
+ if (user_input == "t" || user_input == "b") {
+ std::cout << "\nGetTempFileName Performance:\n [start - end] / total "
+ "--- time "
+ "cost in ms"
+ << std::endl;
+ if (CreateFilesUsingGetTempFileName(file_create_count,
+ temp_dir_gettempfilename.c_str())) {
+ std::cout << "File creation succeeds at " << temp_dir_gettempfilename
+ << ", now clean all of them!" << std::endl;
+ }
+ DeleteDirectoryAndPrintMsg(temp_dir_gettempfilename.c_str());
+ }
+
+ if (user_input == "g" || user_input == "b") {
+ std::cout << "\nGUID-based Performance:\n [start - end] / total --- "
+ "time cost in ms"
+ << std::endl;
+ if (CreateFilesUsingGuid(file_create_count, temp_dir_guid.c_str())) {
+ std::cout << "File creation succeeds at " << temp_dir_guid
+ << ", now clean all of them!" << std::endl;
+ }
+ DeleteDirectoryAndPrintMsg(temp_dir_guid.c_str());
+ }
+ } else {
+ std::cout << "Input number is invalid, please enter # of files to create "
+ "(maximum "
+ << kMaxFileCreate << "), or \"quit\" to end the program : ";
+ }
+ std::cout << std::endl;
+ }
+ return 0;
+}
+
+bool CreateFilesUsingGuid(UINT count, const char* dir_path) {
+ if (!CreateOrValidateTempDirectory(dir_path))
+ return false;
+
+ LARGE_INTEGER starting_time, ending_time, elapsed_ms;
+ ::QueryPerformanceCounter(&starting_time);
+ LARGE_INTEGER frequency;
+ ::QueryPerformanceFrequency(&frequency);
+
+ for (UINT i = 1; i <= count; ++i) {
+ GUID guid;
+ ::CoCreateGuid(&guid);
+ char buffer[37];
+ ConvertGuidToString(&guid, buffer);
+ std::string temp_name = std::string(dir_path).append(buffer).append(".tmp");
+
+ HANDLE file_handle =
+ ::CreateFileA(temp_name.c_str(), GENERIC_READ, FILE_SHARE_READ, NULL,
+ CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
+ ::CloseHandle(file_handle);
+
+ if (i % kFileCountPerMetric == 0 || i == count) {
+ ::QueryPerformanceCounter(&ending_time);
+ // Convert the elapsed number of ticks to milliseconds.
+ elapsed_ms.QuadPart = (ending_time.QuadPart - starting_time.QuadPart) *
+ 1000 / frequency.QuadPart;
+
+ FormatPrintElapsedTime(i, count, elapsed_ms);
+
+ ::QueryPerformanceCounter(&starting_time);
+ }
+ }
+ return true;
+}
+
+bool CreateFilesUsingGetTempFileName(UINT count, const char* dir_path) {
+ if (!CreateOrValidateTempDirectory(dir_path))
+ return false;
+
+ CHAR temp_name[MAX_PATH];
+ LARGE_INTEGER starting_time, ending_time, elapsed_ms;
+ ::QueryPerformanceCounter(&starting_time);
+ LARGE_INTEGER frequency;
+ ::QueryPerformanceFrequency(&frequency);
+
+ for (UINT i = 1; i <= count; ++i) {
+ ::GetTempFileNameA(dir_path, "", 0, temp_name);
+ if (i % kFileCountPerMetric == 0 || i == count) {
+ ::QueryPerformanceCounter(&ending_time);
+ // Convert the elapsed number of ticks to milliseconds.
+ elapsed_ms.QuadPart = (ending_time.QuadPart - starting_time.QuadPart) *
+ 1000 / frequency.QuadPart;
+
+ FormatPrintElapsedTime(i, count, elapsed_ms);
+
+ ::QueryPerformanceCounter(&starting_time);
+ }
+ }
+ return true;
+}
+
+char* ConvertGuidToString(const GUID* id, char* out) {
+ int i;
+ char* ret = out;
+ out += sprintf(out, "%.8lX-%.4hX-%.4hX-", id->Data1, id->Data2, id->Data3);
+ for (i = 0; i < sizeof(id->Data4); ++i) {
+ out += sprintf(out, "%.2hhX", id->Data4[i]);
+ if (i == 1)
+ *(out++) = '-';
+ }
+ return ret;
+}
+
+bool CreateOrValidateTempDirectory(const char* folder_path) {
+ if (::PathFileExistsA(folder_path)) {
+ if (!::PathIsDirectoryEmptyA(folder_path)) {
+ std::cout << folder_path
+ << " directory is not empty, please remove all its content.";
+ return false;
+ }
+ return true;
+ } else if (::CreateDirectoryA(folder_path, NULL) == 0) {
+ std::cout << folder_path << "directory creation fails.";
+ return false;
+ } else {
+ return true;
+ }
+}
+
+void DeleteDirectoryContent(const char* folder_path) {
+ char file_found[MAX_PATH];
+ WIN32_FIND_DATAA info;
+ HANDLE hp;
+ sprintf(file_found, "%s\\*.*", folder_path);
+ hp = ::FindFirstFileA(file_found, &info);
+ do {
+ if ((strcmp(info.cFileName, ".") == 0) ||
+ (strcmp(info.cFileName, "..") == 0)) {
+ continue;
+ }
+ sprintf(file_found, "%s\\%s", folder_path, info.cFileName);
+ ::DeleteFileA(file_found);
+ } while (::FindNextFileA(hp, &info));
+ ::FindClose(hp);
+}
+
+bool DeleteDirectory(const char* folder_path) {
+ DeleteDirectoryContent(folder_path);
+ return ::RemoveDirectoryA(folder_path) != 0;
+}
+
+void DeleteDirectoryAndPrintMsg(const char* folder_path) {
+ if (DeleteDirectory(folder_path)) {
+ std::cout << folder_path << " directory is deleted!" << std::endl;
+ } else {
+ std::cout << "[Attention] " << folder_path
+ << " directory's deletion fails, please take a look by yourself!"
+ << std::endl;
+ }
+}
+
+void FormatPrintElapsedTime(UINT cur_step,
+ UINT total_step,
+ const LARGE_INTEGER& elapsed_ms) {
+ UINT count_prev = 0;
+ if (cur_step % kFileCountPerMetric == 0)
+ count_prev = cur_step + 1 - kFileCountPerMetric;
+ else if (cur_step > kFileCountPerMetric)
+ count_prev = cur_step / kFileCountPerMetric * kFileCountPerMetric + 1;
+ printf(" [%5d - %5d] / %d --- %lld\n", count_prev, cur_step, total_step,
+ elapsed_ms.QuadPart);
+}

Powered by Google App Engine
This is Rietveld 408576698