@Inject public TaskSystemTable(TaskManager taskManager, NodeInfo nodeInfo) { this.taskManager = taskManager; this.nodeId = nodeInfo.getNodeId(); }
@GET @Produces(APPLICATION_JSON) public NodeStatus getStatus() { return new NodeStatus( nodeInfo.getNodeId(), version, environment, coordinator, nanosSince(startTime), nodeInfo.getExternalAddress(), nodeInfo.getInternalAddress(), memoryManager.getInfo(), logicalCores, operatingSystemMXBean == null ? 0 : operatingSystemMXBean.getProcessCpuLoad(), operatingSystemMXBean == null ? 0 : operatingSystemMXBean.getSystemCpuLoad(), memoryMXBean.getHeapMemoryUsage().getUsed(), memoryMXBean.getHeapMemoryUsage().getMax(), memoryMXBean.getNonHeapMemoryUsage().getUsed()); } }
.filter(descriptor -> !nodeInfo.getNodeId().equals(descriptor.getNodeId())) .collect(toImmutableSet());
@Inject public DiscoveryNodeManager( @ServiceType("presto") ServiceSelector serviceSelector, NodeInfo nodeInfo, FailureDetector failureDetector, NodeVersion expectedNodeVersion, @ForNodeManager HttpClient httpClient, InternalCommunicationConfig internalCommunicationConfig) { this.serviceSelector = requireNonNull(serviceSelector, "serviceSelector is null"); this.failureDetector = requireNonNull(failureDetector, "failureDetector is null"); this.expectedNodeVersion = requireNonNull(expectedNodeVersion, "expectedNodeVersion is null"); this.httpClient = requireNonNull(httpClient, "httpClient is null"); this.nodeStateUpdateExecutor = newSingleThreadScheduledExecutor(threadsNamed("node-state-poller-%s")); this.nodeStateEventExecutor = newCachedThreadPool(threadsNamed("node-state-events-%s")); this.httpsRequired = internalCommunicationConfig.isHttpsRequired(); this.currentNode = findCurrentNode( serviceSelector.selectAllServices(), requireNonNull(nodeInfo, "nodeInfo is null").getNodeId(), expectedNodeVersion, httpsRequired); refreshNodesInternal(); }
taskId, locationFactory.createLocalTaskLocation(taskId), nodeInfo.getNodeId(), queryContexts.getUnchecked(taskId.getQueryId()), sqlTaskExecutionFactory,
@BeforeMethod public void setup() { testHttpClient = new TestingHttpClient(input -> new TestingResponse(OK, ArrayListMultimap.create(), ACTIVE.name().getBytes())); expectedVersion = new NodeVersion("1"); coordinator = new PrestoNode(UUID.randomUUID().toString(), URI.create("https://192.0.2.8"), expectedVersion, false); activeNodes = ImmutableList.of( new PrestoNode(nodeInfo.getNodeId(), URI.create("http://192.0.1.1"), expectedVersion, false), new PrestoNode(UUID.randomUUID().toString(), URI.create("http://192.0.2.1:8080"), expectedVersion, false), new PrestoNode(UUID.randomUUID().toString(), URI.create("http://192.0.2.3"), expectedVersion, false), coordinator); inactiveNodes = ImmutableList.of( new PrestoNode(UUID.randomUUID().toString(), URI.create("https://192.0.3.9"), NodeVersion.UNKNOWN, false), new PrestoNode(UUID.randomUUID().toString(), URI.create("https://192.0.4.9"), new NodeVersion("2"), false)); List<ServiceDescriptor> descriptors = new ArrayList<>(); for (PrestoNode node : Iterables.concat(activeNodes, inactiveNodes)) { descriptors.add(serviceDescriptor("presto") .setNodeId(node.getNodeIdentifier()) .addProperty("http", node.getHttpUri().toString()) .addProperty("node_version", node.getNodeVersion().toString()) .addProperty("coordinator", String.valueOf(node.equals(coordinator))) .build()); } selector = new StaticServiceSelector(descriptors); }
@Inject public TaskSystemTable(TaskManager taskManager, NodeInfo nodeInfo) { this.taskManager = taskManager; this.nodeId = nodeInfo.getNodeId(); }
@Inject public QuerySystemTable(QueryManager queryManager, NodeInfo nodeInfo) { this.queryManager = queryManager; this.nodeId = nodeInfo.getNodeId(); }
@Inject public TaskSystemTable(TaskManager taskManager, NodeInfo nodeInfo) { this.taskManager = taskManager; this.nodeId = nodeInfo.getNodeId(); }
@Inject public TaskSystemTable(TaskManager taskManager, NodeInfo nodeInfo) { this.taskManager = taskManager; this.nodeId = nodeInfo.getNodeId(); }
@Inject public NodeIdUserAgentRequestFilter(NodeInfo nodeInfo) { requireNonNull(nodeInfo, "nodeInfo is null"); this.nodeId = requireNonNull(nodeInfo.getNodeId(), "nodeId is null"); }
public ServiceDescriptorBuilder setNodeInfo(NodeInfo nodeInfo) { requireNonNull(nodeInfo, "nodeInfo is null"); this.nodeId = nodeInfo.getNodeId(); this.pool = nodeInfo.getPool(); return this; }
public ServiceDescriptorBuilder setNodeInfo(NodeInfo nodeInfo) { Preconditions.checkNotNull(nodeInfo, "nodeInfo is null"); this.nodeId = nodeInfo.getNodeId(); this.pool = nodeInfo.getPool(); return this; }
public ServiceDescriptorBuilder setNodeInfo(NodeInfo nodeInfo) { Preconditions.checkNotNull(nodeInfo, "nodeInfo is null"); this.nodeId = nodeInfo.getNodeId(); this.pool = nodeInfo.getPool(); return this; }
@Override public ListenableFuture<Void> unannounce() { URI uri = discoveryServiceURI.get(); if (uri == null) { return immediateFuture(null); } Request request = prepareDelete() .setUri(URI.create(uri + "/v1/announcement/" + nodeInfo.getNodeId())) .setHeader("User-Agent", nodeInfo.getNodeId()) .build(); return httpClient.executeAsync(request, new DiscoveryResponseHandler<>("Unannouncement", uri)); }
@Override public CheckedFuture<Duration, DiscoveryException> announce(Set<ServiceAnnouncement> services) { Preconditions.checkNotNull(services, "services is null"); ImmutableSet.Builder<ServiceDescriptor> builder = ImmutableSet.builder(); for (ServiceAnnouncement service : services) { builder.add(new ServiceDescriptor(service.getId(), nodeInfo.getNodeId(), service.getType(), nodeInfo.getPool(), null, ServiceState.RUNNING, service.getProperties())); } announcements.set(builder.build()); return Futures.immediateCheckedFuture(maxAge); }
@Override public ListenableFuture<Void> unannounce() { URI uri = discoveryServiceURI.get(); if (uri == null) { return Futures.immediateCheckedFuture(null); } Request request = prepareDelete() .setUri(URI.create(uri + "/v1/announcement/" + nodeInfo.getNodeId())) .setHeader("User-Agent", nodeInfo.getNodeId()) .build(); return httpClient.executeAsync(request, new DiscoveryResponseHandler<>("Unannouncement", uri)); }
@Override public CheckedFuture<Void, DiscoveryException> unannounce() { URI uri = discoveryServiceURI.get(); if (uri == null) { return Futures.immediateFailedCheckedFuture(new DiscoveryException("No discovery servers are available")); } Request request = prepareDelete() .setUri(URI.create(uri + "/v1/announcement/" + nodeInfo.getNodeId())) .setHeader("User-Agent", nodeInfo.getNodeId()) .build(); return httpClient.executeAsync(request, new DiscoveryResponseHandler<Void>("Unannouncement")); }
@Test public void test() { assertEquals(agent.getAgentId(), nodeInfo.getNodeId()); assertEquals(agent.getLocation(), agent.getLocation()); assertEquals(agent.getResources(), TEST_RESOURCES); }
@Test public void test() { assertEquals(agent.getAgentId(), nodeInfo.getNodeId()); assertEquals(agent.getLocation(), agent.getLocation()); assertEquals(agent.getResources(), TEST_RESOURCES); } }