PowerShell/src/Microsoft.PowerShell.Commands.Management/cimSupport/cmdletization/cim/QueryJob.cs at master · WebGameLinux/PowerShell · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
/********************************************************************++
Copyright (c) Microsoft Corporation.  All rights reserved.
--********************************************************************/

using System;
using System.Globalization;
using System.Text;
using Microsoft.Management.Infrastructure;
using Dbg = System.Management.Automation.Diagnostics;

namespace Microsoft.PowerShell.Cmdletization.Cim
{
    /// <summary>
    /// Job that handles executing a WQL (in the future CQL?) query on a remote CIM server
    /// </summary>
    internal class QueryInstancesJob : QueryJobBase
    {
        private readonly string _wqlQuery;
        private readonly bool _useEnumerateInstances;
        internal QueryInstancesJob(CimJobContext jobContext, CimQuery cimQuery, string wqlCondition)
                : base(jobContext, cimQuery)
        {
            Dbg.Assert(wqlCondition != null, "Caller should verify that wqlCondition is not null");

            var wqlQueryBuilder = new StringBuilder();
            wqlQueryBuilder.Append("SELECT * FROM ");
            wqlQueryBuilder.Append(this.JobContext.ClassName);
            wqlQueryBuilder.Append(" ");
            wqlQueryBuilder.Append(wqlCondition);
            _wqlQuery = wqlQueryBuilder.ToString();

            if (string.IsNullOrWhiteSpace(wqlCondition))
            {
                _useEnumerateInstances = true;
            }
            else
            {
                if (jobContext.CmdletInvocationContext.CmdletDefinitionContext.UseEnumerateInstancesInsteadOfWql)
                {
                    _useEnumerateInstances = true;
                }
            }
        }

        internal override IObservable<CimInstance> GetCimOperation()
        {
            this.WriteVerboseStartOfCimOperation();

            IObservable<CimInstance> observable;
            if (_useEnumerateInstances)
            {
                observable = this.JobContext.Session.EnumerateInstancesAsync(
                    this.JobContext.Namespace,
                    this.JobContext.ClassNameOrNullIfResourceUriIsUsed,
                    this.CreateOperationOptions());
            }
            else
            {
                observable = this.JobContext.Session.QueryInstancesAsync(
                    this.JobContext.Namespace,
                    "WQL",
                    _wqlQuery,
                    this.CreateOperationOptions());
            }

            return observable;
        }

        internal override string Description
        {
            get
            {
                return this.FailSafeDescription;
            }
        }

        internal override string FailSafeDescription
        {
            get
            {
                return string.Format(
                    CultureInfo.InvariantCulture,
                    CmdletizationResources.CimJob_SafeQueryDescription,
                    this.JobContext.CmdletizationClassName,
                    this.JobContext.Session.ComputerName,
                    _wqlQuery);
            }
        }
    }
}