如何使Win32_Process查询更快?

时间:2018-12-12 11:04:08

标签: c# windows wmi

我具有以下功能,可将过程信息提取到数据表中:

public DataTable getProcesses()
        {
            DataTable dt = new DataTable();

            dt.Columns.Add("ID");
            dt.Columns.Add("Name");
            dt.Columns.Add("Path");
            dt.Columns.Add("User");
            dt.Columns.Add("Priority");

            string pid = "-";
            string name = "-";
            string path = "-";
            string priort = "-";
            string user = "-";

            string query = "Select * From Win32_Process";
            ManagementObjectSearcher searcher = new ManagementObjectSearcher(query);
            ManagementObjectCollection processList = searcher.Get();

            foreach (ManagementObject proc in processList)
            {
                pid = proc["ProcessID"].ToString();
                name = proc["Name"].ToString();

                if (proc["ExecutablePath"] != null)
                {
                    path = proc["ExecutablePath"].ToString();
                    priort = proc["Priority"].ToString();
                }
                string[] argList = new string[2];
                int returnVal = Convert.ToInt32(proc.InvokeMethod("GetOwner", argList));
                if (returnVal == 0)
                {
                    // return DOMAIN\user
                    user = argList[1] + "\\" + argList[0];
                }

                dt.Rows.Add(pid, name, path, user, priort);
            }

            return dt;
        }

这最终可以工作,并给我我想要的回报,但是执行需要20到30秒。

我想知道是否有任何方法可以优化功能,或者特别是查询,这很可能是“延迟”的来源。

修改

在完成注释中建议的某些操作后,它平均下降到15-20秒,但这仍然太长了。最多可以忍受4-5秒。我仍然没有更改查询,可以做些什么使它更快?

修改2

应用@NicoRiff建议的一些更改后,仍然获得相同的运行时,我进行了一些调试,以了解有效花费了这么长时间。原来它是特定的一行:int returnVal = Convert.ToInt32(proc.InvokeMethod("GetOwner", argList));,这将使我“拥有”每个进程的用户。

此行大约需要60毫秒,而其他所有行都需要1-2毫秒。超过200次迭代(这是我拥有的进程数,我只能想象使用一个更大的列表将花费的时间),总共花费大约12-13秒(仅用于那一行),得出15-20总计。

现在我已经“解决”了问题,如何优化该功能?

2 个答案:

答案 0 :(得分:1)

我个人一直想摆脱使用DataTable的麻烦。您可以使用List<>或可以解决许多潜在问题的集合。

话虽如此,您可能想签出ORMi库以获取有关List<>的WMI信息。您可以通过以下方式实现正在尝试的目标:

        WMIHelper helper = new WMIHelper("root\\CimV2");

        string pid = "-";
        string name = "-";
        string path = "-";
        string priort = "-";
        string user = "-";

        var processes = helper.Query("Select * From Win32_Process").ToList();

        foreach (var p in processes)
        {
            pid = p.ProcessID;
            name = p.Name;
            path = p.ExecutablePath ?? String.Empty;
            priort = p.Priority ?? String.Empty;
        }

以上代码可用于dynamic对象,不需要您编写任何模型。只是那一点代码。如果需要使用方法,则可以声明模型并使用强类型对象:

1)定义模型:

[WMIClass(Name = "Win32_Process", Namespace = "root\\CimV2")]
public class Process
{
    public int Handle { get; set; }
    public string Name { get; set; }
    public int ProcessID { get; set; }
    public string ExecutablePath { get; set; }
    public int Priority { get; set; }

    /// <summary>
    /// Date the process begins executing.
    /// </summary>
    public DateTime CreationDate { get; set; }

    public dynamic GetOwnerSid()
    {
        return WMIMethod.ExecuteMethod(this);
    }

    public ProcessOwner GetOwner()
    {
        return WMIMethod.ExecuteMethod<ProcessOwner>(this);
    }

    public int AttachDebugger()
    {
        return WMIMethod.ExecuteMethod<int>(this);
    }
}

public class ProcessOwner
{
    public string Domain { get; set; }
    public int ReturnValue { get; set; }
    public string User { get; set; }
}

2)查询WMI

        List<Process> processes = helper.Query<Process>().ToList();

        foreach (Process p in processes)
        {
            pid = p.ProcessID;
            name = p.Name;
            path = p.ExecutablePath ?? String.Empty;
            priort = p.Priority ?? String.Empty;

            dynamic d = p.GetOwnerSid();
            ProcessOwner po = p.GetOwner();
        }

即使对于第二种方法而言,第二种方法似乎工作量过多,您也将获得更加简洁易懂的代码。

注意:我已经在ORMi上尝试过您的代码,并且在1-2秒内得到了结果。正如其他人所说,这可能取决于您的环境。

注2:始终仅使用SELECT语句上所需的属性。 WMI SELECT *的价格非常昂贵。始终指定属性。在您的情况下,它将是:

Select ProcessID, Name, ExecutablePath, Priority From Win32_Process

(ORMi还会为您解决这个问题,因为它总是查询模型上设置的属性。)

答案 1 :(得分:1)

从技术上讲,这不会回答您的问题,因为它没有利用Win32_Process查询。 但是,使用Powershell可以在短时间内(〜1.5 vs〜25秒)产生相同的结果。

您需要在X64模式下运行此命令来询问64位进程,并且需要Powershell提升权限才能返回UserName。

请注意,这是我第一次从C#调用Powershell脚本,可能会有更好的方法。 它还需要一些错误陷阱以使其更强大(不值得按原样生产)。

using System.Management.Automation;
using System.Management.Automation.Runspaces;

private DataTable getProcesses() 
{
    // Create the datatable and columns
    DataTable dt = new DataTable();
    dt.Columns.Add("ID");
    dt.Columns.Add("Name");
    dt.Columns.Add("Path");
    dt.Columns.Add("User");
    dt.Columns.Add("Priority");
    dt.Columns.Add("BasePriority"); 

    string script = $"get-process -IncludeUserName | select id, processname, path, username, priorityclass";

    List<string[]> psOutput = new List<string[]>();

    // Invoke Powershell cmdlet and get output
    using (PowerShell ps = PowerShell.Create())
    {
        ps.AddScript(script);
        var output = ps.Invoke();
        if(ps.Streams.Error.Count > 0)
        {
            throw new Exception($"Error running script:{Environment.NewLine}{string.Join(Environment.NewLine, ps.Streams.Error.Select(e => e.ToString()))}");
        }

        // clean and split the output
        psOutput.AddRange(output.Select(i => i.ToString().Replace("; ", ";").TrimStart("@{".ToCharArray()).TrimEnd('}').Split(';')));
    }

    // populate the DataTable
    psOutput
        .AsParallel()           // this does not really help when not calling Process.GetProcessById
        .Select(p => p.Select(f => f.Split("=")).ToDictionary(k => k.FirstOrDefault(), v => v.LastOrDefault()))
        .Select(kv => new object[] {    // "flatten" the dictionaries into object[]'s that will become the datarows
                        kv["Id"]
                        , kv["ProcessName"]
                        , kv["Path"]
                        , kv["UserName"]
                        , kv["PriorityClass"]
                        , Process.GetProcessById(int.Parse(kv["Id"])).BasePriority  // if you need the numerical base priority - takes quite a bit longer though (Not sure how to get this via Powershell)
                    }
            )
        .ToList()
        .ForEach(r => dt.Rows.Add(r));  // add each object[] to the datatable

    // return the datatable 
    return dt;
}