Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright 2016 Google, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License")
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.netflix.spinnaker.clouddriver.google.provider.agent
import com.fasterxml.jackson.databind.ObjectMapper
import com.google.api.services.compute.ComputeRequest
import com.google.api.services.compute.model.*
import com.netflix.spectator.api.Registry
import com.netflix.spinnaker.cats.agent.AgentDataType
import com.netflix.spinnaker.cats.agent.CacheResult
import com.netflix.spinnaker.cats.provider.ProviderCache
import com.netflix.spinnaker.clouddriver.google.cache.CacheResultBuilder
import com.netflix.spinnaker.clouddriver.google.cache.Keys
import com.netflix.spinnaker.clouddriver.google.model.GoogleHealthCheck
import com.netflix.spinnaker.clouddriver.google.provider.agent.util.PaginatedRequest
import com.netflix.spinnaker.clouddriver.google.security.GoogleNamedAccountCredentials
import groovy.util.logging.Slf4j
import static com.netflix.spinnaker.cats.agent.AgentDataType.Authority.AUTHORITATIVE
import static com.netflix.spinnaker.clouddriver.google.cache.Keys.Namespace.HEALTH_CHECKS
@Slf4j
class GoogleHealthCheckCachingAgent extends AbstractGoogleCachingAgent {
final Set providedDataTypes = [
AUTHORITATIVE.forType(HEALTH_CHECKS.ns)
] as Set
String agentType = "$accountName/$GoogleHealthCheckCachingAgent.simpleName"
GoogleHealthCheckCachingAgent(String clouddriverUserAgentApplicationName,
GoogleNamedAccountCredentials credentials,
ObjectMapper objectMapper,
Registry registry) {
super(clouddriverUserAgentApplicationName,
credentials,
objectMapper,
registry)
}
@Override
CacheResult loadData(ProviderCache providerCache) {
List httpHealthCheckList = loadHealthChecks()
buildCacheResult(providerCache, httpHealthCheckList)
}
/**
* There are currently three different sets/families of health check endpoints that we cache:
* 1. /{project}/global/httpHealthChecks/{healthCheckname}
* 2. /{project}/global/httpsHealthChecks/{healthCheckname}
* 3. /{project}/global/healthChecks/{healthCheckname}
*
* Endpoint (3) can return HTTP and HTTPS endpoints, similar to endpoints (1) and (2).
* Hence, we have to differentiate which health checks came from which families. We do that with
* 'kind', which denotes the health check family. Check the GoogleHealthCheck class for the actual values for 'kind'.
*/
List loadHealthChecks() {
List ret = []
List httpHealthChecks = new PaginatedRequest(this) {
@Override
protected ComputeRequest request (String pageToken) {
return compute.httpHealthChecks().list(project).setPageToken(pageToken)
}
@Override
String getNextPageToken(HttpHealthCheckList t) {
return t.getNextPageToken();
}
}
.timeExecute(
{ HttpHealthCheckList list -> list.getItems() },
"compute.httpHealthChecks.list", TAG_SCOPE, SCOPE_GLOBAL
)
httpHealthChecks.each { HttpHealthCheck hc ->
ret << new GoogleHealthCheck(
name: hc.getName(),
selfLink: hc.getSelfLink(),
region: "global",
healthCheckType: GoogleHealthCheck.HealthCheckType.HTTP,
kind: GoogleHealthCheck.HealthCheckKind.httpHealthCheck,
port: hc.getPort(),
requestPath: hc.getRequestPath(),
checkIntervalSec: hc.getCheckIntervalSec(),
timeoutSec: hc.getTimeoutSec(),
healthyThreshold: hc.getHealthyThreshold(),
unhealthyThreshold: hc.getUnhealthyThreshold()
)
}
List httpsHealthChecks = new PaginatedRequest(this) {
@Override
protected ComputeRequest request (String pageToken) {
return compute.httpsHealthChecks().list(project).setPageToken(pageToken)
}
@Override
String getNextPageToken(HttpsHealthCheckList t) {
return t.getNextPageToken();
}
}
.timeExecute(
{ HttpsHealthCheckList list -> list.getItems() },
"compute.httpsHealthChecks.list", TAG_SCOPE, SCOPE_GLOBAL
)
httpsHealthChecks.each { HttpsHealthCheck hc ->
ret << new GoogleHealthCheck(
name: hc.getName(),
selfLink: hc.getSelfLink(),
region: "global",
healthCheckType: GoogleHealthCheck.HealthCheckType.HTTPS,
kind: GoogleHealthCheck.HealthCheckKind.httpsHealthCheck,
port: hc.getPort(),
requestPath: hc.getRequestPath(),
checkIntervalSec: hc.getCheckIntervalSec(),
timeoutSec: hc.getTimeoutSec(),
healthyThreshold: hc.getHealthyThreshold(),
unhealthyThreshold: hc.getUnhealthyThreshold()
)
}
List healthChecks = new PaginatedRequest(this) {
@Override
protected ComputeRequest request (String pageToken) {
return compute.healthChecks().list(project).setPageToken(pageToken)
}
@Override
String getNextPageToken(HealthCheckList t) {
return t.getNextPageToken();
}
}
.timeExecute(
{ HealthCheckList list -> list.getItems() },
"compute.healthChecks.list", TAG_SCOPE, SCOPE_GLOBAL
)
ret.addAll(healthChecks.collect { toGoogleHealthCheck(it, "global") })
def cachingAgent = this
credentials.regions.collect { it.name }.each { String region ->
List regionHealthChecks = new PaginatedRequest(cachingAgent) {
@Override
protected ComputeRequest request (String pageToken) {
return compute.regionHealthChecks().list(project, region).setPageToken(pageToken)
}
@Override
String getNextPageToken(HealthCheckList t) {
return t.getNextPageToken();
}
}
.timeExecute(
{ HealthCheckList list -> list.getItems() },
"compute.regionHealthChecks.list", TAG_SCOPE, SCOPE_REGIONAL, TAG_REGION, region
)
ret.addAll(regionHealthChecks.collect { toGoogleHealthCheck(it, region) })
}
ret
}
private static GoogleHealthCheck toGoogleHealthCheck(HealthCheck hc, String region) {
def newHC = new GoogleHealthCheck(
name: hc.getName(),
selfLink: hc.getSelfLink(),
region: region,
kind: GoogleHealthCheck.HealthCheckKind.healthCheck,
checkIntervalSec: hc.getCheckIntervalSec(),
timeoutSec: hc.getTimeoutSec(),
healthyThreshold: hc.getHealthyThreshold(),
unhealthyThreshold: hc.getUnhealthyThreshold()
)
// Health checks of kind 'healthCheck' are all nested -- the actual health check is contained
// in a field inside a wrapper HealthCheck object. The wrapper object specifies the type of nested
// health check as a string, and the proper field is populated based on the type.
switch(hc.getType()) {
case 'HTTP':
newHC.healthCheckType = GoogleHealthCheck.HealthCheckType.HTTP
newHC.port = hc.getHttpHealthCheck().getPort()
newHC.requestPath = hc.getHttpHealthCheck().getRequestPath()
break
case 'HTTPS':
newHC.healthCheckType = GoogleHealthCheck.HealthCheckType.HTTPS
newHC.port = hc.getHttpsHealthCheck().getPort()
newHC.requestPath = hc.getHttpsHealthCheck().getRequestPath()
break
case 'TCP':
newHC.healthCheckType = GoogleHealthCheck.HealthCheckType.TCP
newHC.port = hc.getTcpHealthCheck().getPort()
break
case 'SSL':
newHC.healthCheckType = GoogleHealthCheck.HealthCheckType.SSL
newHC.port = hc.getSslHealthCheck().getPort()
break
case 'UDP':
newHC.healthCheckType = GoogleHealthCheck.HealthCheckType.UDP
newHC.port = hc.getUdpHealthCheck().getPort()
break
default:
log.warn("Health check ${hc.getName()} has unknown type ${hc.getType()}.")
return
break
}
return newHC
}
private CacheResult buildCacheResult(ProviderCache _, List healthCheckList) {
log.debug("Describing items in ${agentType}")
def cacheResultBuilder = new CacheResultBuilder()
healthCheckList.each { GoogleHealthCheck healthCheck ->
def healthCheckKey = Keys.getHealthCheckKey(accountName, healthCheck.kind as String, healthCheck.getName())
cacheResultBuilder.namespace(HEALTH_CHECKS.ns).keep(healthCheckKey).with {
attributes.healthCheck = healthCheck
}
}
log.debug("Caching ${cacheResultBuilder.namespace(HEALTH_CHECKS.ns).keepSize()} items in ${agentType}")
cacheResultBuilder.build()
}
}
