Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file |
| 2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
| 3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
| 4 | 4 |
| 5 /** | 5 /** |
| 6 * A load-balancing runner pool. | 6 * A load-balancing runner pool. |
| 7 */ | 7 */ |
| 8 library dart.pkg.isolate.loadbalancer; | 8 library dart.pkg.isolate.loadbalancer; |
| 9 | 9 |
| 10 import "runner.dart"; | 10 import "runner.dart"; |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 49 | 49 |
| 50 /** | 50 /** |
| 51 * Asynchronously create [size] runners and create a `LoadBalancer` of those. | 51 * Asynchronously create [size] runners and create a `LoadBalancer` of those. |
| 52 * | 52 * |
| 53 * This is a helper function that makes it easy to create a `LoadBalancer` | 53 * This is a helper function that makes it easy to create a `LoadBalancer` |
| 54 * with asynchronously created runners, for example: | 54 * with asynchronously created runners, for example: |
| 55 * | 55 * |
| 56 * var isolatePool = LoadBalancer.create(10, IsolateRunner.spawn); | 56 * var isolatePool = LoadBalancer.create(10, IsolateRunner.spawn); |
| 57 */ | 57 */ |
| 58 static Future<LoadBalancer> create(int size, Future<Runner> createRunner()) { | 58 static Future<LoadBalancer> create(int size, Future<Runner> createRunner()) { |
| 59 return Future.wait(new Iterable.generate(size, (_) => createRunner()), | 59 return Future |
| 60 cleanUp: (Runner runner) { runner.close(); }) | 60 .wait(new Iterable.generate(size, (_) => createRunner()), |
| 61 .then((runners) => new LoadBalancer(runners)); | 61 cleanUp: (Runner runner) { |
|
Lasse Reichstein Nielsen
2015/02/26 10:59:14
indent cleanUp to after the '('. Putting the body
| |
| 62 runner.close(); | |
| 63 }).then((runners) => new LoadBalancer(runners)); | |
| 62 } | 64 } |
| 63 | 65 |
| 64 static List<_LoadBalancerEntry> _createEntries(Iterable<Runner> runners) { | 66 static List<_LoadBalancerEntry> _createEntries(Iterable<Runner> runners) { |
| 65 var entries = runners.map((runner) => new _LoadBalancerEntry(runner)); | 67 var entries = runners.map((runner) => new _LoadBalancerEntry(runner)); |
| 66 return new List<_LoadBalancerEntry>.from(entries, growable: false); | 68 return new List<_LoadBalancerEntry>.from(entries, growable: false); |
| 67 } | 69 } |
| 68 | 70 |
| 69 /** | 71 /** |
| 70 * Execute the command in the currently least loaded isolate. | 72 * Execute the command in the currently least loaded isolate. |
| 71 * | 73 * |
| 72 * The optional [load] parameter represents the load that the command | 74 * The optional [load] parameter represents the load that the command |
| 73 * is causing on the isolate where it runs. | 75 * is causing on the isolate where it runs. |
| 74 * The number has no fixed meaning, but should be seen as relative to | 76 * The number has no fixed meaning, but should be seen as relative to |
| 75 * other commands run in the same load balancer. | 77 * other commands run in the same load balancer. |
| 76 * The `load` must not be negative. | 78 * The `load` must not be negative. |
| 77 * | 79 * |
| 78 * If [timeout] and [onTimeout] are provided, they are forwarded to | 80 * If [timeout] and [onTimeout] are provided, they are forwarded to |
| 79 * the runner running the function, which will handle a timeout | 81 * the runner running the function, which will handle a timeout |
| 80 * as normal. | 82 * as normal. |
| 81 */ | 83 */ |
| 82 Future run(function(argument), argument, {Duration timeout, | 84 Future run(function(argument), argument, |
| 83 onTimeout(), | 85 {Duration timeout, onTimeout(), int load: 100}) { |
|
Lasse Reichstein Nielsen
2015/02/26 10:59:14
Indent "{" to after then "(". That is:
Future run(
| |
| 84 int load: 100}) { | |
| 85 RangeError.checkNotNegative(load, "load"); | 86 RangeError.checkNotNegative(load, "load"); |
| 86 _LoadBalancerEntry entry = _first; | 87 _LoadBalancerEntry entry = _first; |
| 87 _increaseLoad(entry, load); | 88 _increaseLoad(entry, load); |
| 88 return entry.run(this, load, function, argument, timeout, onTimeout); | 89 return entry.run(this, load, function, argument, timeout, onTimeout); |
| 89 } | 90 } |
| 90 | 91 |
| 91 /** | 92 /** |
| 92 * Execute the same function in the least loaded [count] isolates. | 93 * Execute the same function in the least loaded [count] isolates. |
| 93 * | 94 * |
| 94 * This guarantees that the function isn't run twice in the same isolate, | 95 * This guarantees that the function isn't run twice in the same isolate, |
| 95 * so `count` is not allowed to exceed [length]. | 96 * so `count` is not allowed to exceed [length]. |
| 96 * | 97 * |
| 97 * The optional [load] parameter represents the load that the command | 98 * The optional [load] parameter represents the load that the command |
| 98 * is causing on the isolate where it runs. | 99 * is causing on the isolate where it runs. |
| 99 * The number has no fixed meaning, but should be seen as relative to | 100 * The number has no fixed meaning, but should be seen as relative to |
| 100 * other commands run in the same load balancer. | 101 * other commands run in the same load balancer. |
| 101 * The `load` must not be negative. | 102 * The `load` must not be negative. |
| 102 * | 103 * |
| 103 * If [timeout] and [onTimeout] are provided, they are forwarded to | 104 * If [timeout] and [onTimeout] are provided, they are forwarded to |
| 104 * the runners running the function, which will handle any timeouts | 105 * the runners running the function, which will handle any timeouts |
| 105 * as normal. | 106 * as normal. |
| 106 */ | 107 */ |
| 107 List<Future> runMultiple(int count, function(argument), argument, | 108 List<Future> runMultiple(int count, function(argument), argument, |
| 108 {Duration timeout, | 109 {Duration timeout, onTimeout(), int load: 100}) { |
|
Lasse Reichstein Nielsen
2015/02/26 10:59:14
Indent '{ ' to after '('.
| |
| 109 onTimeout(), | |
| 110 int load: 100}) { | |
| 111 RangeError.checkValueInInterval(count, 1, _length, "count"); | 110 RangeError.checkValueInInterval(count, 1, _length, "count"); |
| 112 RangeError.checkNotNegative(load, "load"); | 111 RangeError.checkNotNegative(load, "load"); |
| 113 if (count == 1) { | 112 if (count == 1) { |
| 114 return list1(run(function, argument, load: load, | 113 return list1(run(function, argument, |
| 115 timeout: timeout, onTimeout: onTimeout)); | 114 load: load, timeout: timeout, onTimeout: onTimeout)); |
|
Lasse Reichstein Nielsen
2015/02/26 10:59:14
Indent arguments to after '(' or put all arguments
| |
| 116 } | 115 } |
| 117 List result = new List<Future>(count); | 116 List result = new List<Future>(count); |
| 118 if (count == _length) { | 117 if (count == _length) { |
| 119 // No need to change the order of entries in the queue. | 118 // No need to change the order of entries in the queue. |
| 120 for (int i = 0; i < count; i++) { | 119 for (int i = 0; i < count; i++) { |
| 121 _LoadBalancerEntry entry = _queue[i]; | 120 _LoadBalancerEntry entry = _queue[i]; |
| 122 entry.load += load; | 121 entry.load += load; |
| 123 result[i] = entry.run(this, load, function, argument, | 122 result[i] = |
| 124 timeout, onTimeout); | 123 entry.run(this, load, function, argument, timeout, onTimeout); |
| 125 } | 124 } |
| 126 } else { | 125 } else { |
| 127 // Remove the [count] least loaded services and run the | 126 // Remove the [count] least loaded services and run the |
| 128 // command on each, then add them back to the queue. | 127 // command on each, then add them back to the queue. |
| 129 // This avoids running the same command twice in the same | 128 // This avoids running the same command twice in the same |
| 130 // isolate. | 129 // isolate. |
| 131 List entries = new List(count); | 130 List entries = new List(count); |
| 132 for (int i = 0; i < count; i++) { | 131 for (int i = 0; i < count; i++) { |
| 133 entries[i] = _removeFirst(); | 132 entries[i] = _removeFirst(); |
| 134 } | 133 } |
| 135 for (int i = 0; i < count; i++) { | 134 for (int i = 0; i < count; i++) { |
| 136 _LoadBalancerEntry entry = entries[i]; | 135 _LoadBalancerEntry entry = entries[i]; |
| 137 entry.load += load; | 136 entry.load += load; |
| 138 _add(entry); | 137 _add(entry); |
| 139 result[i] = entry.run(this, load, function, argument, | 138 result[i] = |
| 140 timeout, onTimeout); | 139 entry.run(this, load, function, argument, timeout, onTimeout); |
| 141 } | 140 } |
| 142 } | 141 } |
| 143 return result; | 142 return result; |
| 144 } | 143 } |
| 145 | 144 |
| 146 Future close() { | 145 Future close() { |
| 147 if (_stopFuture != null) return _stopFuture; | 146 if (_stopFuture != null) return _stopFuture; |
| 148 _stopFuture = MultiError.waitUnordered(_queue.take(_length) | 147 _stopFuture = |
| 149 .map((e) => e.close())); | 148 MultiError.waitUnordered(_queue.take(_length).map((e) => e.close())); |
| 150 // Remove all entries. | 149 // Remove all entries. |
| 151 for (int i = 0; i < _length; i++) _queue[i].queueIndex = -1; | 150 for (int i = 0; i < _length; i++) _queue[i].queueIndex = -1; |
| 152 _queue = null; | 151 _queue = null; |
| 153 _length = 0; | 152 _length = 0; |
| 154 return _stopFuture; | 153 return _stopFuture; |
| 155 } | 154 } |
| 156 | 155 |
| 157 /** | 156 /** |
| 158 * Place [element] in heap at [index] or above. | 157 * Place [element] in heap at [index] or above. |
| 159 * | 158 * |
| (...skipping 16 matching lines...) Expand all Loading... | |
| 176 | 175 |
| 177 /** | 176 /** |
| 178 * Place [element] in heap at [index] or above. | 177 * Place [element] in heap at [index] or above. |
| 179 * | 178 * |
| 180 * Put element into the empty cell at `index`. | 179 * Put element into the empty cell at `index`. |
| 181 * While the `element` has lower priority than either child, | 180 * While the `element` has lower priority than either child, |
| 182 * swap it with the highest priority child. | 181 * swap it with the highest priority child. |
| 183 */ | 182 */ |
| 184 void _bubbleDown(_LoadBalancerEntry element, int index) { | 183 void _bubbleDown(_LoadBalancerEntry element, int index) { |
| 185 while (true) { | 184 while (true) { |
| 186 int childIndex = index * 2 + 1; // Left child index. | 185 int childIndex = index * 2 + 1; // Left child index. |
|
Lasse Reichstein Nielsen
2015/02/26 10:59:14
There should always be two spaces before an end-of
| |
| 187 if (childIndex >= _length) break; | 186 if (childIndex >= _length) break; |
| 188 _LoadBalancerEntry child = _queue[childIndex]; | 187 _LoadBalancerEntry child = _queue[childIndex]; |
| 189 int rightChildIndex = childIndex + 1; | 188 int rightChildIndex = childIndex + 1; |
| 190 if (rightChildIndex < _length) { | 189 if (rightChildIndex < _length) { |
| 191 _LoadBalancerEntry rightChild = _queue[rightChildIndex]; | 190 _LoadBalancerEntry rightChild = _queue[rightChildIndex]; |
| 192 if (rightChild.compareTo(child) < 0) { | 191 if (rightChild.compareTo(child) < 0) { |
| 193 childIndex = rightChildIndex; | 192 childIndex = rightChildIndex; |
| 194 child = rightChild; | 193 child = rightChild; |
| 195 } | 194 } |
| 196 } | 195 } |
| (...skipping 79 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 276 class _LoadBalancerEntry implements Comparable<_LoadBalancerEntry> { | 275 class _LoadBalancerEntry implements Comparable<_LoadBalancerEntry> { |
| 277 // The current load on the isolate. | 276 // The current load on the isolate. |
| 278 int load = 0; | 277 int load = 0; |
| 279 // The current index in the heap-queue. | 278 // The current index in the heap-queue. |
| 280 // Negative when the entry is not part of the queue. | 279 // Negative when the entry is not part of the queue. |
| 281 int queueIndex = -1; | 280 int queueIndex = -1; |
| 282 | 281 |
| 283 // The service used to send commands to the other isolate. | 282 // The service used to send commands to the other isolate. |
| 284 Runner runner; | 283 Runner runner; |
| 285 | 284 |
| 286 _LoadBalancerEntry(Runner runner) | 285 _LoadBalancerEntry(Runner runner) : runner = runner; |
| 287 : runner = runner; | |
| 288 | 286 |
| 289 /** Whether the entry is still in the queue. */ | 287 /** Whether the entry is still in the queue. */ |
| 290 bool get inQueue => queueIndex >= 0; | 288 bool get inQueue => queueIndex >= 0; |
| 291 | 289 |
| 292 Future run(LoadBalancer balancer, int load, function(argumen), argument, | 290 Future run(LoadBalancer balancer, int load, function(argumen), argument, |
| 293 Duration timeout, onTimeout()) { | 291 Duration timeout, onTimeout()) { |
| 294 return runner.run(function, argument, | 292 return runner |
| 295 timeout: timeout, onTimeout: onTimeout).whenComplete(() { | 293 .run(function, argument, timeout: timeout, onTimeout: onTimeout) |
| 294 .whenComplete(() { | |
| 296 balancer._decreaseLoad(this, load); | 295 balancer._decreaseLoad(this, load); |
|
Lasse Reichstein Nielsen
2015/02/26 10:59:14
Indent body by four more.
| |
| 297 }); | 296 }); |
| 298 } | 297 } |
| 299 | 298 |
| 300 Future close() => runner.close(); | 299 Future close() => runner.close(); |
| 301 | 300 |
| 302 int compareTo(_LoadBalancerEntry other) => load - other.load; | 301 int compareTo(_LoadBalancerEntry other) => load - other.load; |
| 303 } | 302 } |
| 304 | |
| 305 | |
| OLD | NEW |