5 // Atsushi Enomoto (atsushi@xamarin.com)
7 // Copyright (C) 2013 Xamarin Inc. (http://www.xamarin.com)
9 // Permission is hereby granted, free of charge, to any person obtaining
10 // a copy of this software and associated documentation files (the
11 // "Software"), to deal in the Software without restriction, including
12 // without limitation the rights to use, copy, modify, merge, publish,
13 // distribute, sublicense, and/or sell copies of the Software, and to
14 // permit persons to whom the Software is furnished to do so, subject to
15 // the following conditions:
17 // The above copyright notice and this permission notice shall be
18 // included in all copies or substantial portions of the Software.
20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
29 using System.Collections.Generic;
31 using Microsoft.Build.Execution;
32 using Microsoft.Build.Framework;
33 using System.Threading.Tasks;
34 using System.Threading;
35 using System.Collections.Concurrent;
37 namespace Microsoft.Build.Internal
39 class BuildNodeManager
41 public BuildNodeManager (BuildManager buildManager)
43 BuildManager = buildManager;
44 new Thread (RunLoop).Start ();
50 queue_wait_handle.Set ();
53 public BuildManager BuildManager { get; private set; }
55 List<BuildNode> in_proc_nodes = new List<BuildNode> ();
56 List<BuildNode> out_proc_nodes = new List<BuildNode> ();
57 AutoResetEvent queue_wait_handle = new AutoResetEvent (false);
58 ConcurrentQueue<BuildSubmission> queued_builds = new ConcurrentQueue<BuildSubmission> ();
59 // FIXME: currently it is not in use but it should be stored somewhere for cancellation.
60 Dictionary<BuildSubmission,Task> ongoing_builds = new Dictionary<BuildSubmission, Task> ();
63 readonly TaskFactory<BuildResult> task_factory = new TaskFactory<BuildResult> ();
64 internal TaskFactory<BuildResult> ThreadTaskFactory {
65 get { return task_factory; }
72 if (queued_builds.Count == 0)
73 queue_wait_handle.WaitOne ();
76 BuildSubmission build;
77 if (!queued_builds.TryDequeue (out build))
79 StartOneBuild (build);
80 } catch (Exception ex) {
81 // FIXME: I guess INodeLogger should be used instead.
82 Console.Error.WriteLine ("Uncaught build node exception occured");
83 Console.Error.WriteLine (ex);
91 queue_wait_handle.Set ();
94 public void ResetCaches ()
96 in_proc_nodes.Clear ();
97 out_proc_nodes.Clear ();
100 public void Enqueue (BuildSubmission build)
102 queued_builds.Enqueue (build);
103 queue_wait_handle.Set ();
106 void StartOneBuild (BuildSubmission build)
108 var node = TakeNode (build);
109 // FIXME: Task (non-generic) here causes NotImplementedException in somewhere in Interlocked. It does not make sense.
110 ongoing_builds [build] = task_factory.StartNew (node.ExecuteBuild);
111 //new Thread (() => { node.ExecuteBuild (); }).Start ();
114 void EndOneBuild (BuildNode node)
116 var task = ongoing_builds [node.Build];
117 ongoing_builds [node.Build] = null;
121 // FIXME: take max nodes into account here, and get throttling working.
122 BuildNode TakeNode (BuildSubmission build)
124 var host = BuildManager.OngoingBuildParameters.HostServices;
125 NodeAffinity affinity;
127 affinity = NodeAffinity.Any;
129 affinity = host.GetNodeAffinity (build.BuildRequest.ProjectFullPath);
130 BuildNode n = GetReusableNode (affinity);
134 n = new BuildNode (this, affinity == NodeAffinity.Any ? NodeAffinity.InProc : affinity);
136 if (n.Affinity == NodeAffinity.InProc)
137 in_proc_nodes.Add (n);
139 out_proc_nodes.Add (n);
144 BuildNode GetReusableNode (NodeAffinity affinity)
146 if (!BuildManager.OngoingBuildParameters.EnableNodeReuse)
149 if (affinity != NodeAffinity.OutOfProc)
150 foreach (var n in in_proc_nodes)
151 if (n.IsAvailable && (n.Affinity & affinity) != 0)
153 if (affinity != NodeAffinity.InProc)
154 foreach (var n in out_proc_nodes)
155 if (n.IsAvailable && (n.Affinity & affinity) != 0)
160 internal class BuildNode
162 static Random rnd = new Random ();
164 public BuildNode (BuildNodeManager manager, NodeAffinity affinity)
171 public bool IsAvailable { get; private set; }
172 public int Id { get; private set; }
173 public BuildNodeManager Manager { get; set; }
174 public NodeAffinity Affinity { get; private set; }
175 public BuildSubmission Build { get; private set; }
177 public void Assign (BuildSubmission build)
183 public void Release ()
189 public BuildResult ExecuteBuild ()
193 // FIXME: depending on NodeAffinity, build it through another MSBuild process.
194 if (Affinity == NodeAffinity.OutOfProc)
195 throw new NotImplementedException ();
196 result = Build.InternalExecute ();
197 } catch (Exception ex) {
198 // FIXME: I guess INodeLogger should be used instead.
199 Console.Error.WriteLine ("Uncaught build node exception occured");
200 Console.Error.WriteLine (ex);
203 Manager.EndOneBuild (this);