http://git-wip-us.apache.org/repos/asf/hadoop/blob/a23ff8d8/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/definition/YARN-Simplified-V1-API-Layer-For-Services.yaml ---------------------------------------------------------------------- diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/definition/YARN-Simplified-V1-API-Layer-For-Services.yaml b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/definition/YARN-Simplified-V1-API-Layer-For-Services.yaml new file mode 100644 index 0000000..d90ae06 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/definition/YARN-Simplified-V1-API-Layer-For-Services.yaml @@ -0,0 +1,594 @@ +# Hadoop YARN REST APIs for services v1 spec in YAML + +# Licensed to the Apache Software Foundation (ASF) under one or more +# contributor license agreements. See the NOTICE file distributed with +# this work for additional information regarding copyright ownership. +# The ASF licenses this file to You 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. + +swagger: '2.0' +info: + title: "YARN Simplified API layer for services" + description: | + Bringing a new service on YARN today is not a simple experience. The APIs of existing + frameworks are either too low level (native YARN), require writing new code (for frameworks with programmatic APIs) + or writing a complex spec (for declarative frameworks). + + This simplified REST API can be used to create and manage the lifecycle of YARN services. + In most cases, the application owner will not be forced to make any changes to their applications. + This is primarily true if the application is packaged with containerization technologies like Docker. + + This document describes the API specifications (aka. YarnFile) for deploying/managing + containerized services on YARN. The same JSON spec can be used for both REST API + and CLI to manage the services. + + version: "1.0.0" + license: + name: Apache 2.0 + url: http://www.apache.org/licenses/LICENSE-2.0.html +# the domain of the service +host: localhost +port: 8088(default) +# array of all schemes that your API supports +schemes: + - http +consumes: + - application/json +produces: + - application/json +paths: + /app/v1/services/version: + get: + summary: Get current version of the API server. + description: Get current version of the API server. + responses: + 200: + description: Successful request + + /app/v1/services: + get: + summary: (TBD) List of services running in the cluster. + description: Get a list of all currently running services (response includes a minimal projection of the service info). For more details do a GET on a specific service name. + responses: + 200: + description: An array of services + schema: + type: array + items: + $ref: '#/definitions/Service' + default: + description: Unexpected error + schema: + $ref: '#/definitions/ServiceStatus' + post: + summary: Create a service + description: Create a service. The request JSON is a service object with details required for creation. If the request is successful it returns 202 Accepted. A success of this API only confirms success in submission of the service creation request. There is no guarantee that the service will actually reach a RUNNING state. Resource availability and several other factors determines if the service will be deployed in the cluster. It is expected that clients would subsequently call the GET API to get details of the service and determine its state. + parameters: + - name: Service + in: body + description: Service request object + required: true + schema: + $ref: '#/definitions/Service' + responses: + 202: + description: The request to create a service is accepted + 400: + description: Invalid service definition provided in the request body + 500: + description: Failed to create a service + default: + description: Unexpected error + schema: + $ref: '#/definitions/ServiceStatus' + + /app/v1/services/{service_name}: + put: + summary: Update a service or upgrade the binary version of the components of a running service + description: Update the runtime properties of a service. Currently the following operations are supported - update lifetime, stop/start a service. + The PUT operation is also used to orchestrate an upgrade of the service containers to a newer version of their artifacts (TBD). + parameters: + - name: service_name + in: path + description: Service name + required: true + type: string + - name: Service + in: body + description: The updated service definition. It can contain the updated lifetime of a service or the desired state (STOPPED/STARTED) of a service to initiate a start/stop operation against the specified service + required: true + schema: + $ref: '#/definitions/Service' + responses: + 204: + description: Update or upgrade was successful + 404: + description: Service does not exist + default: + description: Unexpected error + schema: + $ref: '#/definitions/ServiceStatus' + delete: + summary: Destroy a service + description: Destroy a service and release all resources. This API might have to return JSON data providing location of logs (TBD), etc. + parameters: + - name: service_name + in: path + description: Service name + required: true + type: string + responses: + 204: + description: Destroy was successful + 404: + description: Service does not exist + default: + description: Unexpected error + schema: + $ref: '#/definitions/ServiceStatus' + get: + summary: Get details of a service. + description: Return the details (including containers) of a running service + parameters: + - name: service_name + in: path + description: Service name + required: true + type: string + responses: + 200: + description: a service object + schema: + type: object + items: + $ref: '#/definitions/Service' + examples: + service_name: logsearch + artifact: + id: logsearch:latest + type: docker + 404: + description: Service does not exist + default: + description: Unexpected error + schema: + $ref: '#/definitions/ServiceStatus' + /app/v1/services/{service_name}/components/{component_name}: + put: + summary: Flex a component's number of instances. + description: Set a component's desired number of instanes + parameters: + - name: service_name + in: path + description: Service name + required: true + type: string + - name: component_name + in: path + description: Component name + required: true + type: string + - name: Component + in: body + description: The definition of a component which contains the updated number of instances. + required: true + schema: + $ref: '#/definitions/Component' + responses: + 200: + description: Flex was successful + 404: + description: Service does not exist + default: + description: Unexpected error + schema: + $ref: '#/definitions/ServiceStatus' +definitions: + Service: + description: a service resource has the following attributes. + required: + - name + - version + properties: + name: + type: string + description: A unique service name. If Registry DNS is enabled, the max length is 63 characters. + version: + type: string + description: Version of the service. + description: + type: string + description: Description of the service. + id: + type: string + description: A unique service id. + artifact: + description: The default artifact for all components of the service except the components which has Artifact type set to SERVICE (optional). + $ref: '#/definitions/Artifact' + resource: + description: The default resource for all components of the service (optional). + $ref: '#/definitions/Resource' + launch_time: + type: string + format: date + description: The time when the service was created, e.g. 2016-03-16T01:01:49.000Z. + number_of_running_containers: + type: integer + format: int64 + description: In get response this provides the total number of running containers for this service (across all components) at the time of request. Note, a subsequent request can return a different number as and when more containers get allocated until it reaches the total number of containers or if a flex request has been made between the two requests. + lifetime: + type: integer + format: int64 + description: Life time (in seconds) of the service from the time it reaches the STARTED state (after which it is automatically destroyed by YARN). For unlimited lifetime do not set a lifetime value. + components: + description: Components of a service. + type: array + items: + $ref: '#/definitions/Component' + configuration: + description: Config properties of a service. Configurations provided at the service/global level are available to all the components. Specific properties can be overridden at the component level. + $ref: '#/definitions/Configuration' + state: + description: State of the service. Specifying a value for this attribute for the PUT payload means update the service to this desired state. + $ref: '#/definitions/ServiceState' + quicklinks: + type: object + description: A blob of key-value pairs of quicklinks to be exported for a service. + additionalProperties: + type: string + queue: + type: string + description: The YARN queue that this service should be submitted to. + kerberos_principal: + description: The principal info of the user who launches the service. + $ref: '#/definitions/KerberosPrincipal' + docker_client_config: + type: string + description: URI of the file containing the docker client configuration (e.g. hdfs:///tmp/config.json). + ResourceInformation: + description: + ResourceInformation determines unit/value of resource types in addition to memory and vcores. It will be part of Resource object. + properties: + value: + type: integer + format: int64 + description: Integer value of the resource. + unit: + type: string + description: Unit of the resource, acceptable values are - p/n/u/m/k/M/G/T/P/Ki/Mi/Gi/Ti/Pi. By default it is empty means no unit. + Resource: + description: + Resource determines the amount of resources (vcores, memory, network, etc.) usable by a container. This field determines the resource to be applied for all the containers of a component or service. The resource specified at the service (or global) level can be overriden at the component level. Only one of profile OR cpu & memory are expected. It raises a validation exception otherwise. + properties: + profile: + type: string + description: Each resource profile has a unique id which is associated with a cluster-level predefined memory, cpus, etc. + cpus: + type: integer + format: int32 + description: Amount of vcores allocated to each container (optional but overrides cpus in profile if specified). + memory: + type: string + description: Amount of memory allocated to each container (optional but overrides memory in profile if specified). Currently accepts only an integer value and default unit is in MB. + additional: + type: object + additionalProperties: + $ref: '#/definitions/ResourceInformation' + description: A map of resource type name to resource type information. Including value (integer), and unit (string). This will be used to specify resource other than cpu and memory. Please refer to example below. + PlacementPolicy: + description: Advanced placement policy of the components of a service. + required: + - constraints + properties: + constraints: + description: Placement constraint details. + type: array + items: + $ref: '#/definitions/PlacementConstraint' + PlacementConstraint: + description: Placement constraint details. + required: + - type + - scope + properties: + name: + description: An optional name associated to this constraint. + type: string + example: C1 + type: + description: The type of placement. + $ref: '#/definitions/PlacementType' + scope: + description: The scope of placement. + $ref: '#/definitions/PlacementScope' + target_tags: + description: The name of the components that this component's placement policy is depending upon are added as target tags. So for affinity say, this component's containers are requesting to be placed on hosts where containers of the target tag component(s) are running on. Target tags can also contain the name of this component, in which case it implies that for anti-affinity say, no more than one container of this component can be placed on a host. Similarly, for cardinality, it would mean that containers of this component is requesting to be placed on hosts where at least minCardinality but no more than maxCardinality containers of the target tag component(s) are running. + type: array + items: + type: string + node_attributes: + description: Node attributes are a set of key:value(s) pairs associated with nodes. + type: object + additionalProperties: + type: array + items: + type: string + node_partitions: + description: Node partitions where the containers of this component can run. + type: array + items: + type: string + min_cardinality: + type: integer + format: int64 + description: When placement type is cardinality, the minimum number of containers of the depending component that a host should have, where containers of this component can be allocated on. + example: 2 + max_cardinality: + type: integer + format: int64 + description: When placement type is cardinality, the maximum number of containers of the depending component that a host should have, where containers of this component can be allocated on. + example: 3 + PlacementType: + description: The type of placement - affinity/anti-affinity/affinity-with-cardinality with containers of another component or containers of the same component (self). + properties: + type: + type: string + enum: + - AFFINITY + - ANTI_AFFINITY + - AFFINITY_WITH_CARDINALITY + PlacementScope: + description: The scope of placement for the containers of a component. + properties: + type: + type: string + enum: + - NODE + - RACK + Artifact: + description: Artifact of a service component. If not specified, component will just run the bare launch command and no artifact will be localized. + required: + - id + properties: + id: + type: string + description: Artifact id. Examples are package location uri for tarball based services, image name for docker, name of service, etc. + type: + type: string + description: Artifact type, like docker, tarball, etc. (optional). For TARBALL type, the specified tarball will be localized to the container local working directory under a folder named lib. For SERVICE type, the service specified will be read and its components will be added into this service. The original component with artifact type SERVICE will be removed (any properties specified in the original component will be ignored). + enum: + - DOCKER + - TARBALL + - SERVICE + default: DOCKER + uri: + type: string + description: Artifact location to support multiple artifact stores (optional). + Component: + description: One or more components of the service. If the service is HBase say, then the component can be a simple role like master or regionserver. If the service is a complex business webapp then a component can be other services say Kafka or Storm. Thereby it opens up the support for complex and nested services. + required: + - name + properties: + name: + type: string + description: Name of the service component (mandatory). If Registry DNS is enabled, the max length is 63 characters. If unique component support is enabled, the max length is lowered to 44 characters. + state: + description: The state of the component + $ref: "#/definitions/ComponentState" + dependencies: + type: array + items: + type: string + description: An array of service components which should be in READY state (as defined by readiness check), before this component can be started. The dependencies across all components of a service should be represented as a DAG. + readiness_check: + description: Readiness check for this component. + $ref: '#/definitions/ReadinessCheck' + artifact: + description: Artifact of the component (optional). If not specified, the service level global artifact takes effect. + $ref: '#/definitions/Artifact' + launch_command: + type: string + description: The custom launch command of this component (optional for DOCKER component, required otherwise). When specified at the component level, it overrides the value specified at the global level (if any). + resource: + description: Resource of this component (optional). If not specified, the service level global resource takes effect. + $ref: '#/definitions/Resource' + number_of_containers: + type: integer + format: int64 + description: Number of containers for this component (optional). If not specified, the service level global number_of_containers takes effect. + containers: + type: array + description: Containers of a started component. Specifying a value for this attribute for the POST payload raises a validation error. This blob is available only in the GET response of a started service. + items: + $ref: '#/definitions/Container' + run_privileged_container: + type: boolean + description: Run all containers of this component in privileged mode (YARN-4262). + placement_policy: + description: Advanced scheduling and placement policies for all containers of this component. + $ref: '#/definitions/PlacementPolicy' + configuration: + description: Config properties for this component. + $ref: '#/definitions/Configuration' + quicklinks: + type: array + items: + type: string + description: A list of quicklink keys defined at the service level, and to be resolved by this component. + restartPolicy: + type: string + description: Policy of restart component. Including ALWAYS (Always restart component even if instance exit code = 0); ON_FAILURE (Only restart component if instance exit code != 0); NEVER (Do not restart in any cases) + enum: + - ALWAYS + - ON_FAILURE + - NEVER + default: ALWAYS + ReadinessCheck: + description: A check to be performed to determine the readiness of a component instance (a container). If no readiness check is specified, the default readiness check will be used unless the yarn.service.default-readiness-check.enabled configuration property is set to false at the component, service, or system level. The artifact field is currently unsupported but may be implemented in the future, enabling a pluggable helper container to support advanced use cases. + required: + - type + properties: + type: + type: string + description: DEFAULT (AM checks whether the container has an IP and optionally performs a DNS lookup for the container hostname), HTTP (AM performs default checks, plus sends a REST call to the container and expects a response code between 200 and 299), or PORT (AM performs default checks, plus attempts to open a socket connection to the container on a specified port). + enum: + - DEFAULT + - HTTP + - PORT + properties: + type: object + description: A blob of key value pairs that will be used to configure the check. + additionalProperties: + type: string + artifact: + description: Artifact of the pluggable readiness check helper container (optional). If specified, this helper container typically hosts the http uri and encapsulates the complex scripts required to perform actual container readiness check. At the end it is expected to respond a 204 No content just like the simplified use case. This pluggable framework benefits service owners who can run services without any packaging modifications. Note, artifacts of type docker only is supported for now. NOT IMPLEMENTED YET + $ref: '#/definitions/Artifact' + Configuration: + description: Set of configuration properties that can be injected into the service components via envs, files and custom pluggable helper docker containers. Files of several standard formats like xml, properties, json, yaml and templates will be supported. + properties: + properties: + type: object + description: A blob of key-value pairs for configuring the YARN service AM. + additionalProperties: + type: string + env: + type: object + description: A blob of key-value pairs which will be appended to the default system properties and handed off to the service at start time. All placeholder references to properties will be substituted before injection. + additionalProperties: + type: string + files: + description: Array of list of files that needs to be created and made available as volumes in the service component containers. + type: array + items: + $ref: '#/definitions/ConfigFile' + ConfigFile: + description: A config file that needs to be created and made available as a volume in a service component container. + properties: + type: + type: string + description: Config file in the standard format like xml, properties, json, yaml, template. + enum: + - XML + - PROPERTIES + - JSON + - YAML + - TEMPLATE + - HADOOP_XML + - STATIC + - ARCHIVE + dest_file: + type: string + description: The path that this configuration file should be created as. If it is an absolute path, it will be mounted into the DOCKER container. Absolute paths are only allowed for DOCKER containers. If it is a relative path, only the file name should be provided, and the file will be created in the container local working directory under a folder named conf. + src_file: + type: string + description: This provides the source location of the configuration file, the content of which is dumped to dest_file post property substitutions, in the format as specified in type. Typically the src_file would point to a source controlled network accessible file maintained by tools like puppet, chef, or hdfs etc. Currently, only hdfs is supported. + properties: + type: object + description: A blob of key value pairs that will be dumped in the dest_file in the format as specified in type. If src_file is specified, src_file content are dumped in the dest_file and these properties will overwrite, if any, existing properties in src_file or be added as new properties in src_file. + additionalProperties: + type: string + Container: + description: An instance of a running service container. + properties: + id: + type: string + description: Unique container id of a running service, e.g. container_e3751_1458061340047_0008_01_000002. + launch_time: + type: string + format: date + description: The time when the container was created, e.g. 2016-03-16T01:01:49.000Z. This will most likely be different from cluster launch time. + ip: + type: string + description: IP address of a running container, e.g. 172.31.42.141. The IP address and hostname attribute values are dependent on the cluster/docker network setup as per YARN-4007. + hostname: + type: string + description: Fully qualified hostname of a running container, e.g. ctr-e3751-1458061340047-0008-01-000002.examplestg.site. The IP address and hostname attribute values are dependent on the cluster/docker network setup as per YARN-4007. + bare_host: + type: string + description: The bare node or host in which the container is running, e.g. cn008.example.com. + state: + description: State of the container of a service. + $ref: '#/definitions/ContainerState' + component_instance_name: + type: string + description: Name of the component instance that this container instance belongs to. Component instance name is named as $COMPONENT_NAME-i, where i is a + monotonically increasing integer. E.g. A componet called nginx can have multiple component instances named as nginx-0, nginx-1 etc. + Each component instance is backed by a container instance. + resource: + description: Resource used for this container. + $ref: '#/definitions/Resource' + artifact: + description: Artifact used for this container. + $ref: '#/definitions/Artifact' + privileged_container: + type: boolean + description: Container running in privileged mode or not. + ServiceState: + description: The current state of a service. + properties: + state: + type: string + description: enum of the state of the service + enum: + - ACCEPTED + - STARTED + - STABLE + - STOPPED + - FAILED + - FLEX + - UPGRADING + ContainerState: + description: The current state of the container of a service. + properties: + state: + type: string + description: enum of the state of the container + enum: + - INIT + - STARTED + - READY + ComponentState: + description: The state of the component + properties: + state: + type: string + description: enum of the state of the component + enum: + - INIT + - FLEXING + - STABLE + - UPGRADING + ServiceStatus: + description: The current status of a submitted service, returned as a response to the GET API. + properties: + diagnostics: + type: string + description: Diagnostic information (if any) for the reason of the current state of the service. It typically has a non-null value, if the service is in a non-running state. + state: + description: Service state. + $ref: '#/definitions/ServiceState' + code: + type: integer + format: int32 + description: An error code specific to a scenario which service owners should be able to use to understand the failure in addition to the diagnostic information. + KerberosPrincipal: + description: The kerberos principal info of the user who launches the service. + properties: + principal_name: + type: string + description: The principal name of the user who launches the service. Note that `_HOST` is required in the `principal_name` field such as `testuser/_h...@example.com` because Hadoop client validates that the server's (in this case, the AM's) principal has hostname present when communicating to the server. + keytab: + type: string + description: The URI of the kerberos keytab. Currently supports only files present on the bare host. URI starts with "file\://" - A path on the local host where the keytab is stored. It is assumed that admin pre-installs the keytabs on the local host before AM launches. + + +
http://git-wip-us.apache.org/repos/asf/hadoop/blob/a23ff8d8/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/log4j-server.properties ---------------------------------------------------------------------- diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/log4j-server.properties b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/log4j-server.properties new file mode 100644 index 0000000..8c679b9 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/log4j-server.properties @@ -0,0 +1,76 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you 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. +# + +# This is the log4j configuration for YARN Services REST API Server + +# Log rotation based on size (100KB) with a max of 10 backup files +log4j.rootLogger=INFO, restservicelog +log4j.threshhold=ALL + +log4j.appender.stdout=org.apache.log4j.ConsoleAppender +log4j.appender.stdout.layout=org.apache.log4j.PatternLayout +log4j.appender.stdout.layout.ConversionPattern=%d{ISO8601} %-5p [%t] %c{2} (%F:%M(%L)) - %m%n + +log4j.appender.restservicelog=org.apache.log4j.RollingFileAppender +log4j.appender.restservicelog.layout=org.apache.log4j.PatternLayout +log4j.appender.restservicelog.File=${REST_SERVICE_LOG_DIR}/restservice.log +log4j.appender.restservicelog.MaxFileSize=1GB +log4j.appender.restservicelog.MaxBackupIndex=10 + +# log layout skips stack-trace creation operations by avoiding line numbers and method +log4j.appender.restservicelog.layout.ConversionPattern=%d{ISO8601} [%t] %-5p %c{2} - %m%n + +# debug edition is much more expensive +#log4j.appender.restservicelog.layout.ConversionPattern=%d{ISO8601} [%t] %-5p %c{2} (%F:%M(%L)) - %m%n + +# configure stderr +# set the conversion pattern of stderr +# Print the date in ISO 8601 format +log4j.appender.stderr=org.apache.log4j.ConsoleAppender +log4j.appender.stderr.Target=System.err +log4j.appender.stderr.layout=org.apache.log4j.PatternLayout +log4j.appender.stderr.layout.ConversionPattern=%d{ISO8601} [%t] %-5p %c{2} - %m%n + +log4j.appender.subprocess=org.apache.log4j.ConsoleAppender +log4j.appender.subprocess.layout=org.apache.log4j.PatternLayout +log4j.appender.subprocess.layout.ConversionPattern=[%c{1}]: %m%n + +# for debugging REST API Service +#log4j.logger.org.apache.hadoop.yarn.services=DEBUG + +# uncomment to debug service lifecycle issues +#log4j.logger.org.apache.hadoop.yarn.service.launcher=DEBUG +#log4j.logger.org.apache.hadoop.yarn.service=DEBUG + +# uncomment for YARN operations +#log4j.logger.org.apache.hadoop.yarn.client=DEBUG + +# uncomment this to debug security problems +#log4j.logger.org.apache.hadoop.security=DEBUG + +#crank back on some noise +log4j.logger.org.apache.hadoop.util.NativeCodeLoader=ERROR +log4j.logger.org.apache.hadoop.hdfs=WARN +log4j.logger.org.apache.hadoop.hdfs.shortcircuit=ERROR + +log4j.logger.org.apache.hadoop.yarn.server.nodemanager.containermanager.monitor=WARN +log4j.logger.org.apache.hadoop.yarn.server.nodemanager.NodeStatusUpdaterImpl=WARN +log4j.logger.org.apache.zookeeper=WARN +log4j.logger.org.apache.curator.framework.state=ERROR +log4j.logger.org.apache.curator.framework.imps=WARN + +log4j.logger.org.mortbay.log=DEBUG http://git-wip-us.apache.org/repos/asf/hadoop/blob/a23ff8d8/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/webapps/api-server/app ---------------------------------------------------------------------- diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/webapps/api-server/app b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/webapps/api-server/app new file mode 100644 index 0000000..6a077b1 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/resources/webapps/api-server/app @@ -0,0 +1,16 @@ +# Licensed to the Apache Software Foundation (ASF) under one or more +# contributor license agreements. See the NOTICE file distributed with +# this work for additional information regarding copyright ownership. +# The ASF licenses this file to You 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. + +DON'T DELETE. REST WEBAPP RUN SCRIPT WILL STOP WORKING. http://git-wip-us.apache.org/repos/asf/hadoop/blob/a23ff8d8/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/webapp/WEB-INF/web.xml ---------------------------------------------------------------------- diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/webapp/WEB-INF/web.xml b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/webapp/WEB-INF/web.xml new file mode 100644 index 0000000..1282c9f --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/main/webapp/WEB-INF/web.xml @@ -0,0 +1,36 @@ +<?xml version="1.0" encoding="UTF-8"?> +<!-- + 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. +--> +<web-app xmlns="http://java.sun.com/xml/ns/javaee"; xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"; + xsi:schemaLocation="http://java.sun.com/xml/ns/javaee http://java.sun.com/xml/ns/javaee/web-app_3_0.xsd"; + version="3.0"> + + <servlet> + <servlet-name>Jersey REST API</servlet-name> + <servlet-class>com.sun.jersey.spi.container.servlet.ServletContainer</servlet-class> + <init-param> + <param-name>com.sun.jersey.config.property.packages</param-name> + <param-value>org.apache.hadoop.yarn.service.webapp,org.apache.hadoop.yarn.service.api,org.apache.hadoop.yarn.service.api.records</param-value> + </init-param> + <init-param> + <param-name>com.sun.jersey.api.json.POJOMappingFeature</param-name> + <param-value>true</param-value> + </init-param> + <load-on-startup>1</load-on-startup> + </servlet> + <servlet-mapping> + <servlet-name>Jersey REST API</servlet-name> + <url-pattern>/*</url-pattern> + </servlet-mapping> +</web-app> http://git-wip-us.apache.org/repos/asf/hadoop/blob/a23ff8d8/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/ServiceClientTest.java ---------------------------------------------------------------------- diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/ServiceClientTest.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/ServiceClientTest.java new file mode 100644 index 0000000..75b9486 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/ServiceClientTest.java @@ -0,0 +1,210 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.hadoop.yarn.service; + +import org.apache.hadoop.conf.Configuration; +import org.apache.hadoop.yarn.api.records.ApplicationId; +import org.apache.hadoop.yarn.exceptions.ApplicationNotFoundException; +import org.apache.hadoop.yarn.exceptions.YarnException; +import org.apache.hadoop.yarn.service.api.records.Artifact; +import org.apache.hadoop.yarn.service.api.records.Component; +import org.apache.hadoop.yarn.service.api.records.Container; +import org.apache.hadoop.yarn.service.api.records.ContainerState; +import org.apache.hadoop.yarn.service.api.records.Resource; +import org.apache.hadoop.yarn.service.api.records.Service; +import org.apache.hadoop.yarn.service.client.ServiceClient; +import org.apache.hadoop.yarn.service.utils.ServiceApiUtil; +import org.apache.hadoop.yarn.service.utils.SliderFileSystem; + +import java.io.FileNotFoundException; +import java.io.IOException; +import java.util.ArrayList; +import java.util.HashSet; +import java.util.List; +import java.util.Set; +import java.util.stream.Collectors; + +/** + * A mock version of ServiceClient - This class is design + * to simulate various error conditions that will happen + * when a consumer class calls ServiceClient. + */ +public class ServiceClientTest extends ServiceClient { + + private Configuration conf = new Configuration(); + private Service goodServiceStatus = buildLiveGoodService(); + private boolean initialized; + private Set<String> expectedInstances = new HashSet<>(); + + public ServiceClientTest() { + super(); + } + + @Override + public void init(Configuration conf) { + if (!initialized) { + super.init(conf); + initialized = true; + } + } + + @Override + public void stop() { + // This is needed for testing API Server which uses client to get status + // and then perform an action. + } + + public void forceStop() { + expectedInstances.clear(); + super.stop(); + } + + @Override + public Configuration getConfig() { + return conf; + } + + @Override + public ApplicationId actionCreate(Service service) throws IOException { + ServiceApiUtil.validateAndResolveService(service, + new SliderFileSystem(conf), getConfig()); + return ApplicationId.newInstance(System.currentTimeMillis(), 1); + } + + @Override + public Service getStatus(String appName) throws FileNotFoundException { + if ("jenkins".equals(appName)) { + return goodServiceStatus; + } else { + throw new FileNotFoundException("Service " + appName + " not found"); + } + } + + @Override + public int actionStart(String serviceName) + throws YarnException, IOException { + if (serviceName != null && serviceName.equals("jenkins")) { + return EXIT_SUCCESS; + } else { + throw new ApplicationNotFoundException(""); + } + } + + @Override + public int actionStop(String serviceName, boolean waitForAppStopped) + throws YarnException, IOException { + if (serviceName == null) { + throw new NullPointerException(); + } + if (serviceName.equals("jenkins")) { + return EXIT_SUCCESS; + } else if (serviceName.equals("jenkins-second-stop")) { + return EXIT_COMMAND_ARGUMENT_ERROR; + } else { + throw new ApplicationNotFoundException(""); + } + } + + @Override + public int actionDestroy(String serviceName) { + if (serviceName != null) { + if (serviceName.equals("jenkins")) { + return EXIT_SUCCESS; + } else if (serviceName.equals("jenkins-already-stopped")) { + return EXIT_SUCCESS; + } else if (serviceName.equals("jenkins-doesn't-exist")) { + return EXIT_NOT_FOUND; + } else if (serviceName.equals("jenkins-error-cleaning-registry")) { + return EXIT_OTHER_FAILURE; + } + } + throw new IllegalArgumentException(); + } + + @Override + public int initiateUpgrade(Service service) throws YarnException, + IOException { + if (service.getName() != null && service.getName().equals("jenkins")) { + return EXIT_SUCCESS; + } else { + throw new IllegalArgumentException(); + } + } + + @Override + public int actionUpgrade(Service service, List<Container> compInstances) + throws IOException, YarnException { + if (service.getName() != null && service.getName().equals("jenkins") + && compInstances != null) { + Set<String> actualInstances = compInstances.stream().map( + Container::getComponentInstanceName).collect(Collectors.toSet()); + if (actualInstances.equals(expectedInstances)) { + return EXIT_SUCCESS; + } + } + throw new IllegalArgumentException(); + } + + Service getGoodServiceStatus() { + return goodServiceStatus; + } + + void setExpectedInstances(Set<String> instances) { + if (instances != null) { + expectedInstances.addAll(instances); + } + } + + static Service buildGoodService() { + Service service = new Service(); + service.setName("jenkins"); + service.setVersion("v1"); + Artifact artifact = new Artifact(); + artifact.setType(Artifact.TypeEnum.DOCKER); + artifact.setId("jenkins:latest"); + Resource resource = new Resource(); + resource.setCpus(1); + resource.setMemory("2048"); + List<Component> components = new ArrayList<>(); + for (int i = 0; i < 2; i++) { + Component c = new Component(); + c.setName("jenkins" + i); + c.setNumberOfContainers(2L); + c.setArtifact(artifact); + c.setLaunchCommand(""); + c.setResource(resource); + components.add(c); + } + service.setComponents(components); + return service; + } + + static Service buildLiveGoodService() { + Service service = buildGoodService(); + Component comp = service.getComponents().iterator().next(); + List<Container> containers = new ArrayList<>(); + for (int i = 0; i < comp.getNumberOfContainers(); i++) { + Container container = new Container(); + container.setComponentInstanceName(comp.getName() + "-" + (i + 1)); + container.setState(ContainerState.READY); + containers.add(container); + } + comp.setContainers(containers); + return service; + } +} http://git-wip-us.apache.org/repos/asf/hadoop/blob/a23ff8d8/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/TestApiServer.java ---------------------------------------------------------------------- diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/TestApiServer.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/TestApiServer.java new file mode 100644 index 0000000..733b9bc --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/TestApiServer.java @@ -0,0 +1,623 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.hadoop.yarn.service; + +import static org.junit.Assert.*; + +import java.io.BufferedWriter; +import java.io.File; +import java.io.FileWriter; +import java.util.ArrayList; +import java.util.HashSet; +import java.util.List; +import java.util.Set; + +import javax.servlet.http.HttpServletRequest; +import javax.ws.rs.Path; +import javax.ws.rs.core.Response; +import javax.ws.rs.core.Response.Status; + +import com.google.common.collect.Sets; +import org.apache.commons.io.FileUtils; +import org.apache.hadoop.conf.Configuration; +import org.apache.hadoop.yarn.service.api.records.Artifact; +import org.apache.hadoop.yarn.service.api.records.Artifact.TypeEnum; +import org.apache.hadoop.yarn.service.api.records.Component; +import org.apache.hadoop.yarn.service.api.records.ComponentState; +import org.apache.hadoop.yarn.service.api.records.Container; +import org.apache.hadoop.yarn.service.api.records.ContainerState; +import org.apache.hadoop.yarn.service.api.records.Resource; +import org.apache.hadoop.yarn.service.api.records.Service; +import org.apache.hadoop.yarn.service.api.records.ServiceState; +import org.apache.hadoop.yarn.service.api.records.ServiceStatus; +import org.apache.hadoop.yarn.service.conf.RestApiConstants; +import org.apache.hadoop.yarn.service.webapp.ApiServer; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; +import org.mockito.Mockito; + +/** + * Test case for ApiServer REST API. + * + */ +public class TestApiServer { + private ApiServer apiServer; + private HttpServletRequest request; + private ServiceClientTest mockServerClient; + + @Before + public void setup() throws Exception { + request = Mockito.mock(HttpServletRequest.class); + Mockito.when(request.getRemoteUser()) + .thenReturn(System.getProperty("user.name")); + mockServerClient = new ServiceClientTest(); + Configuration conf = new Configuration(); + conf.set("yarn.api-service.service.client.class", + ServiceClientTest.class.getName()); + apiServer = new ApiServer(conf); + apiServer.setServiceClient(mockServerClient); + } + + @After + public void teardown() { + mockServerClient.forceStop(); + } + + @Test + public void testPathAnnotation() { + assertNotNull(this.apiServer.getClass().getAnnotation(Path.class)); + assertTrue("The controller has the annotation Path", + this.apiServer.getClass().isAnnotationPresent(Path.class)); + final Path path = this.apiServer.getClass() + .getAnnotation(Path.class); + assertEquals("The path has /v1 annotation", "/v1", path.value()); + } + + @Test + public void testGetVersion() { + final Response actual = apiServer.getVersion(); + assertEquals("Version number is", Response.ok().build().getStatus(), + actual.getStatus()); + } + + @Test + public void testBadCreateService() { + Service service = new Service(); + // Test for invalid argument + final Response actual = apiServer.createService(request, service); + assertEquals("Create service is ", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testGoodCreateService() throws Exception { + String json = "{\"auths\": " + + "{\"https://index.docker.io/v1/\": " + + "{\"auth\": \"foobarbaz\"}," + + "\"registry.example.com\": " + + "{\"auth\": \"bazbarfoo\"}}}"; + File dockerTmpDir = new File("target", "docker-tmp"); + FileUtils.deleteQuietly(dockerTmpDir); + dockerTmpDir.mkdirs(); + String dockerConfig = dockerTmpDir + "/config.json"; + BufferedWriter bw = new BufferedWriter(new FileWriter(dockerConfig)); + bw.write(json); + bw.close(); + Service service = ServiceClientTest.buildGoodService(); + final Response actual = apiServer.createService(request, service); + assertEquals("Create service is ", + Response.status(Status.ACCEPTED).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testInternalServerErrorDockerClientConfigMissingCreateService() { + Service service = new Service(); + service.setName("jenkins"); + service.setVersion("v1"); + service.setDockerClientConfig("/does/not/exist/config.json"); + Artifact artifact = new Artifact(); + artifact.setType(TypeEnum.DOCKER); + artifact.setId("jenkins:latest"); + Resource resource = new Resource(); + resource.setCpus(1); + resource.setMemory("2048"); + List<Component> components = new ArrayList<>(); + Component c = new Component(); + c.setName("jenkins"); + c.setNumberOfContainers(1L); + c.setArtifact(artifact); + c.setLaunchCommand(""); + c.setResource(resource); + components.add(c); + service.setComponents(components); + final Response actual = apiServer.createService(request, service); + assertEquals("Create service is ", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testBadGetService() { + final String serviceName = "nonexistent-jenkins"; + final Response actual = apiServer.getService(request, serviceName); + assertEquals("Get service is ", + Response.status(Status.NOT_FOUND).build().getStatus(), + actual.getStatus()); + ServiceStatus serviceStatus = (ServiceStatus) actual.getEntity(); + assertEquals("Response code don't match", + RestApiConstants.ERROR_CODE_APP_NAME_INVALID, serviceStatus.getCode()); + assertEquals("Response diagnostics don't match", + "Service " + serviceName + " not found", + serviceStatus.getDiagnostics()); + } + + @Test + public void testBadGetService2() { + final Response actual = apiServer.getService(request, null); + assertEquals("Get service is ", + Response.status(Status.NOT_FOUND).build().getStatus(), + actual.getStatus()); + ServiceStatus serviceStatus = (ServiceStatus) actual.getEntity(); + assertEquals("Response code don't match", + RestApiConstants.ERROR_CODE_APP_NAME_INVALID, serviceStatus.getCode()); + assertEquals("Response diagnostics don't match", + "Service name cannot be null.", serviceStatus.getDiagnostics()); + } + + @Test + public void testGoodGetService() { + final Response actual = apiServer.getService(request, "jenkins"); + assertEquals("Get service is ", + Response.status(Status.OK).build().getStatus(), actual.getStatus()); + } + + @Test + public void testBadDeleteService() { + final Response actual = apiServer.deleteService(request, "no-jenkins"); + assertEquals("Delete service is ", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testBadDeleteService2() { + final Response actual = apiServer.deleteService(request, null); + assertEquals("Delete service is ", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testBadDeleteService3() { + final Response actual = apiServer.deleteService(request, + "jenkins-doesn't-exist"); + assertEquals("Delete service is ", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testBadDeleteService4() { + final Response actual = apiServer.deleteService(request, + "jenkins-error-cleaning-registry"); + assertEquals("Delete service is ", + Response.status(Status.INTERNAL_SERVER_ERROR).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testGoodDeleteService() { + final Response actual = apiServer.deleteService(request, "jenkins"); + assertEquals("Delete service is ", + Response.status(Status.OK).build().getStatus(), actual.getStatus()); + } + + @Test + public void testDeleteStoppedService() { + final Response actual = apiServer.deleteService(request, + "jenkins-already-stopped"); + assertEquals("Delete service is ", + Response.status(Status.OK).build().getStatus(), actual.getStatus()); + } + + @Test + public void testDecreaseContainerAndStop() { + Service service = new Service(); + service.setState(ServiceState.STOPPED); + service.setName("jenkins"); + Artifact artifact = new Artifact(); + artifact.setType(TypeEnum.DOCKER); + artifact.setId("jenkins:latest"); + Resource resource = new Resource(); + resource.setCpus(1); + resource.setMemory("2048"); + List<Component> components = new ArrayList<Component>(); + Component c = new Component(); + c.setName("jenkins"); + c.setNumberOfContainers(0L); + c.setArtifact(artifact); + c.setLaunchCommand(""); + c.setResource(resource); + components.add(c); + service.setComponents(components); + final Response actual = apiServer.updateService(request, "jenkins", + service); + assertEquals("update service is ", + Response.status(Status.OK).build().getStatus(), actual.getStatus()); + } + + @Test + public void testBadDecreaseContainerAndStop() { + Service service = new Service(); + service.setState(ServiceState.STOPPED); + service.setName("no-jenkins"); + Artifact artifact = new Artifact(); + artifact.setType(TypeEnum.DOCKER); + artifact.setId("jenkins:latest"); + Resource resource = new Resource(); + resource.setCpus(1); + resource.setMemory("2048"); + List<Component> components = new ArrayList<Component>(); + Component c = new Component(); + c.setName("no-jenkins"); + c.setNumberOfContainers(-1L); + c.setArtifact(artifact); + c.setLaunchCommand(""); + c.setResource(resource); + components.add(c); + service.setComponents(components); + System.out.println("before stop"); + final Response actual = apiServer.updateService(request, "no-jenkins", + service); + assertEquals("flex service is ", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testIncreaseContainersAndStart() { + Service service = new Service(); + service.setState(ServiceState.STARTED); + service.setName("jenkins"); + Artifact artifact = new Artifact(); + artifact.setType(TypeEnum.DOCKER); + artifact.setId("jenkins:latest"); + Resource resource = new Resource(); + resource.setCpus(1); + resource.setMemory("2048"); + List<Component> components = new ArrayList<Component>(); + Component c = new Component(); + c.setName("jenkins"); + c.setNumberOfContainers(2L); + c.setArtifact(artifact); + c.setLaunchCommand(""); + c.setResource(resource); + components.add(c); + service.setComponents(components); + final Response actual = apiServer.updateService(request, "jenkins", + service); + assertEquals("flex service is ", + Response.status(Status.OK).build().getStatus(), actual.getStatus()); + } + + @Test + public void testBadStartServices() { + Service service = new Service(); + service.setState(ServiceState.STARTED); + service.setName("no-jenkins"); + Artifact artifact = new Artifact(); + artifact.setType(TypeEnum.DOCKER); + artifact.setId("jenkins:latest"); + Resource resource = new Resource(); + resource.setCpus(1); + resource.setMemory("2048"); + List<Component> components = new ArrayList<Component>(); + Component c = new Component(); + c.setName("jenkins"); + c.setNumberOfContainers(2L); + c.setArtifact(artifact); + c.setLaunchCommand(""); + c.setResource(resource); + components.add(c); + service.setComponents(components); + final Response actual = apiServer.updateService(request, "no-jenkins", + service); + assertEquals("start service is ", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testGoodStartServices() { + Service service = new Service(); + service.setState(ServiceState.STARTED); + service.setName("jenkins"); + Artifact artifact = new Artifact(); + artifact.setType(TypeEnum.DOCKER); + artifact.setId("jenkins:latest"); + Resource resource = new Resource(); + resource.setCpus(1); + resource.setMemory("2048"); + List<Component> components = new ArrayList<Component>(); + Component c = new Component(); + c.setName("jenkins"); + c.setNumberOfContainers(2L); + c.setArtifact(artifact); + c.setLaunchCommand(""); + c.setResource(resource); + components.add(c); + service.setComponents(components); + final Response actual = apiServer.updateService(request, "jenkins", + service); + assertEquals("start service is ", + Response.status(Status.OK).build().getStatus(), actual.getStatus()); + } + + @Test + public void testBadStopServices() { + Service service = new Service(); + service.setState(ServiceState.STOPPED); + service.setName("no-jenkins"); + Artifact artifact = new Artifact(); + artifact.setType(TypeEnum.DOCKER); + artifact.setId("jenkins:latest"); + Resource resource = new Resource(); + resource.setCpus(1); + resource.setMemory("2048"); + List<Component> components = new ArrayList<Component>(); + Component c = new Component(); + c.setName("no-jenkins"); + c.setNumberOfContainers(-1L); + c.setArtifact(artifact); + c.setLaunchCommand(""); + c.setResource(resource); + components.add(c); + service.setComponents(components); + System.out.println("before stop"); + final Response actual = apiServer.updateService(request, "no-jenkins", + service); + assertEquals("stop service is ", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testGoodStopServices() { + Service service = new Service(); + service.setState(ServiceState.STOPPED); + service.setName("jenkins"); + System.out.println("before stop"); + final Response actual = apiServer.updateService(request, "jenkins", + service); + assertEquals("stop service is ", + Response.status(Status.OK).build().getStatus(), actual.getStatus()); + } + + @Test + public void testBadSecondStopServices() throws Exception { + Service service = new Service(); + service.setState(ServiceState.STOPPED); + service.setName("jenkins-second-stop"); + // simulates stop on an already stopped service + System.out.println("before second stop"); + final Response actual = apiServer.updateService(request, + "jenkins-second-stop", service); + assertEquals("stop service should have thrown 400 Bad Request: ", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + ServiceStatus serviceStatus = (ServiceStatus) actual.getEntity(); + assertEquals("Stop service should have failed with service already stopped", + "Service jenkins-second-stop is already stopped", + serviceStatus.getDiagnostics()); + } + + @Test + public void testUpdateService() { + Service service = new Service(); + service.setState(ServiceState.STARTED); + service.setName("no-jenkins"); + Artifact artifact = new Artifact(); + artifact.setType(TypeEnum.DOCKER); + artifact.setId("jenkins:latest"); + Resource resource = new Resource(); + resource.setCpus(1); + resource.setMemory("2048"); + List<Component> components = new ArrayList<Component>(); + Component c = new Component(); + c.setName("no-jenkins"); + c.setNumberOfContainers(-1L); + c.setArtifact(artifact); + c.setLaunchCommand(""); + c.setResource(resource); + components.add(c); + service.setComponents(components); + System.out.println("before stop"); + final Response actual = apiServer.updateService(request, "no-jenkins", + service); + assertEquals("update service is ", + Response.status(Status.BAD_REQUEST) + .build().getStatus(), actual.getStatus()); + } + + @Test + public void testUpdateComponent() { + Response actual = apiServer.updateComponent(request, "jenkins", + "jenkins-master", null); + ServiceStatus serviceStatus = (ServiceStatus) actual.getEntity(); + assertEquals("Update component should have failed with 400 bad request", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + assertEquals("Update component should have failed with no data error", + "No component data provided", serviceStatus.getDiagnostics()); + + Component comp = new Component(); + actual = apiServer.updateComponent(request, "jenkins", "jenkins-master", + comp); + serviceStatus = (ServiceStatus) actual.getEntity(); + assertEquals("Update component should have failed with 400 bad request", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + assertEquals("Update component should have failed with no count error", + "No container count provided", serviceStatus.getDiagnostics()); + + comp.setNumberOfContainers(-1L); + actual = apiServer.updateComponent(request, "jenkins", "jenkins-master", + comp); + serviceStatus = (ServiceStatus) actual.getEntity(); + assertEquals("Update component should have failed with 400 bad request", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + assertEquals("Update component should have failed with no count error", + "Invalid number of containers specified -1", serviceStatus.getDiagnostics()); + + comp.setName("jenkins-slave"); + comp.setNumberOfContainers(1L); + actual = apiServer.updateComponent(request, "jenkins", "jenkins-master", + comp); + serviceStatus = (ServiceStatus) actual.getEntity(); + assertEquals("Update component should have failed with 400 bad request", + Response.status(Status.BAD_REQUEST).build().getStatus(), + actual.getStatus()); + assertEquals( + "Update component should have failed with component name mismatch " + + "error", + "Component name in the request object (jenkins-slave) does not match " + + "that in the URI path (jenkins-master)", + serviceStatus.getDiagnostics()); + } + + @Test + public void testInitiateUpgrade() { + Service goodService = ServiceClientTest.buildLiveGoodService(); + goodService.setVersion("v2"); + goodService.setState(ServiceState.UPGRADING); + final Response actual = apiServer.updateService(request, + goodService.getName(), goodService); + assertEquals("Initiate upgrade is ", + Response.status(Status.ACCEPTED).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testUpgradeSingleInstance() { + Service goodService = ServiceClientTest.buildLiveGoodService(); + Component comp = goodService.getComponents().iterator().next(); + Container container = comp.getContainers().iterator().next(); + container.setState(ContainerState.UPGRADING); + + // To be able to upgrade, the service needs to be in UPGRADING + // and container state needs to be in NEEDS_UPGRADE. + Service serviceStatus = mockServerClient.getGoodServiceStatus(); + serviceStatus.setState(ServiceState.UPGRADING); + Container liveContainer = serviceStatus.getComponents().iterator().next() + .getContainers().iterator().next(); + liveContainer.setState(ContainerState.NEEDS_UPGRADE); + mockServerClient.setExpectedInstances(Sets.newHashSet( + liveContainer.getComponentInstanceName())); + + final Response actual = apiServer.updateComponentInstance(request, + goodService.getName(), comp.getName(), + container.getComponentInstanceName(), container); + assertEquals("Instance upgrade is ", + Response.status(Status.ACCEPTED).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testUpgradeMultipleInstances() { + Service goodService = ServiceClientTest.buildLiveGoodService(); + Component comp = goodService.getComponents().iterator().next(); + comp.getContainers().forEach(container -> + container.setState(ContainerState.UPGRADING)); + + // To be able to upgrade, the service needs to be in UPGRADING + // and container state needs to be in NEEDS_UPGRADE. + Service serviceStatus = mockServerClient.getGoodServiceStatus(); + serviceStatus.setState(ServiceState.UPGRADING); + Set<String> expectedInstances = new HashSet<>(); + serviceStatus.getComponents().iterator().next().getContainers().forEach( + container -> { + container.setState(ContainerState.NEEDS_UPGRADE); + expectedInstances.add(container.getComponentInstanceName()); + } + ); + mockServerClient.setExpectedInstances(expectedInstances); + + final Response actual = apiServer.updateComponentInstances(request, + goodService.getName(), comp.getContainers()); + assertEquals("Instance upgrade is ", + Response.status(Status.ACCEPTED).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testUpgradeComponent() { + Service goodService = ServiceClientTest.buildLiveGoodService(); + Component comp = goodService.getComponents().iterator().next(); + comp.setState(ComponentState.UPGRADING); + + // To be able to upgrade, the service needs to be in UPGRADING + // and component state needs to be in NEEDS_UPGRADE. + Service serviceStatus = mockServerClient.getGoodServiceStatus(); + serviceStatus.setState(ServiceState.UPGRADING); + Component liveComp = serviceStatus.getComponent(comp.getName()); + liveComp.setState(ComponentState.NEEDS_UPGRADE); + Set<String> expectedInstances = new HashSet<>(); + liveComp.getContainers().forEach(container -> { + expectedInstances.add(container.getComponentInstanceName()); + container.setState(ContainerState.NEEDS_UPGRADE); + }); + mockServerClient.setExpectedInstances(expectedInstances); + + final Response actual = apiServer.updateComponent(request, + goodService.getName(), comp.getName(), comp); + assertEquals("Component upgrade is ", + Response.status(Status.ACCEPTED).build().getStatus(), + actual.getStatus()); + } + + @Test + public void testUpgradeMultipleComps() { + Service goodService = ServiceClientTest.buildLiveGoodService(); + goodService.getComponents().forEach(comp -> + comp.setState(ComponentState.UPGRADING)); + + // To be able to upgrade, the live service needs to be in UPGRADING + // and component states needs to be in NEEDS_UPGRADE. + Service serviceStatus = mockServerClient.getGoodServiceStatus(); + serviceStatus.setState(ServiceState.UPGRADING); + Set<String> expectedInstances = new HashSet<>(); + serviceStatus.getComponents().forEach(liveComp -> { + liveComp.setState(ComponentState.NEEDS_UPGRADE); + liveComp.getContainers().forEach(liveContainer -> { + expectedInstances.add(liveContainer.getComponentInstanceName()); + liveContainer.setState(ContainerState.NEEDS_UPGRADE); + }); + }); + mockServerClient.setExpectedInstances(expectedInstances); + + final Response actual = apiServer.updateComponents(request, + goodService.getName(), goodService.getComponents()); + assertEquals("Component upgrade is ", + Response.status(Status.ACCEPTED).build().getStatus(), + actual.getStatus()); + } +} http://git-wip-us.apache.org/repos/asf/hadoop/blob/a23ff8d8/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/client/TestApiServiceClient.java ---------------------------------------------------------------------- diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/client/TestApiServiceClient.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/client/TestApiServiceClient.java new file mode 100644 index 0000000..6cf0880 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/client/TestApiServiceClient.java @@ -0,0 +1,314 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.hadoop.yarn.service.client; + +import static org.junit.Assert.*; + +import java.io.IOException; +import java.util.HashMap; + +import javax.servlet.ServletException; +import javax.servlet.http.HttpServlet; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; + +import com.google.common.collect.Lists; +import org.apache.hadoop.conf.Configuration; +import org.apache.hadoop.yarn.exceptions.YarnException; +import org.eclipse.jetty.server.Server; +import org.eclipse.jetty.server.ServerConnector; +import org.eclipse.jetty.servlet.ServletContextHandler; +import org.eclipse.jetty.servlet.ServletHolder; +import org.eclipse.jetty.util.thread.QueuedThreadPool; +import org.junit.AfterClass; +import org.junit.BeforeClass; +import org.junit.Test; +import static org.apache.hadoop.yarn.service.exceptions.LauncherExitCodes.*; + +/** + * Test case for CLI to API Service. + * + */ +public class TestApiServiceClient { + private static ApiServiceClient asc; + private static ApiServiceClient badAsc; + private static Server server; + + /** + * A mocked version of API Service for testing purpose. + * + */ + @SuppressWarnings("serial") + public static class TestServlet extends HttpServlet { + + @Override + protected void doGet(HttpServletRequest req, HttpServletResponse resp) + throws ServletException, IOException { + System.out.println("Get was called"); + if (req.getPathInfo() != null + && req.getPathInfo().contains("nonexistent-app")) { + resp.setStatus(HttpServletResponse.SC_NOT_FOUND); + } else { + resp.setStatus(HttpServletResponse.SC_OK); + } + } + + @Override + protected void doPost(HttpServletRequest req, HttpServletResponse resp) + throws ServletException, IOException { + resp.setStatus(HttpServletResponse.SC_OK); + } + + @Override + protected void doPut(HttpServletRequest req, HttpServletResponse resp) + throws ServletException, IOException { + resp.setStatus(HttpServletResponse.SC_OK); + } + + @Override + protected void doDelete(HttpServletRequest req, HttpServletResponse resp) + throws ServletException, IOException { + resp.setStatus(HttpServletResponse.SC_OK); + } + + } + + @BeforeClass + public static void setup() throws Exception { + server = new Server(8088); + ((QueuedThreadPool)server.getThreadPool()).setMaxThreads(10); + ServletContextHandler context = new ServletContextHandler(); + context.setContextPath("/app"); + server.setHandler(context); + context.addServlet(new ServletHolder(TestServlet.class), "/*"); + ((ServerConnector)server.getConnectors()[0]).setHost("localhost"); + server.start(); + + Configuration conf = new Configuration(); + conf.set("yarn.resourcemanager.webapp.address", + "localhost:8088"); + asc = new ApiServiceClient(); + asc.serviceInit(conf); + + Configuration conf2 = new Configuration(); + conf2.set("yarn.resourcemanager.webapp.address", + "localhost:8089"); + badAsc = new ApiServiceClient(); + badAsc.serviceInit(conf2); + } + + @AfterClass + public static void tearDown() throws Exception { + server.stop(); + } + + @Test + public void testLaunch() { + String fileName = "target/test-classes/example-app.json"; + String appName = "example-app"; + long lifetime = 3600L; + String queue = "default"; + try { + int result = asc.actionLaunch(fileName, appName, lifetime, queue); + assertEquals(EXIT_SUCCESS, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testBadLaunch() { + String fileName = "unknown_file"; + String appName = "unknown_app"; + long lifetime = 3600L; + String queue = "default"; + try { + int result = badAsc.actionLaunch(fileName, appName, lifetime, queue); + assertEquals(EXIT_EXCEPTION_THROWN, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testStatus() { + String appName = "nonexistent-app"; + try { + String result = asc.getStatusString(appName); + assertEquals("Status reponse don't match", + " Service " + appName + " not found", result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testStop() { + String appName = "example-app"; + try { + int result = asc.actionStop(appName); + assertEquals(EXIT_SUCCESS, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testBadStop() { + String appName = "unknown_app"; + try { + int result = badAsc.actionStop(appName); + assertEquals(EXIT_EXCEPTION_THROWN, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testStart() { + String appName = "example-app"; + try { + int result = asc.actionStart(appName); + assertEquals(EXIT_SUCCESS, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testBadStart() { + String appName = "unknown_app"; + try { + int result = badAsc.actionStart(appName); + assertEquals(EXIT_EXCEPTION_THROWN, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testSave() { + String fileName = "target/test-classes/example-app.json"; + String appName = "example-app"; + long lifetime = 3600L; + String queue = "default"; + try { + int result = asc.actionSave(fileName, appName, lifetime, queue); + assertEquals(EXIT_SUCCESS, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testBadSave() { + String fileName = "unknown_file"; + String appName = "unknown_app"; + long lifetime = 3600L; + String queue = "default"; + try { + int result = badAsc.actionSave(fileName, appName, lifetime, queue); + assertEquals(EXIT_EXCEPTION_THROWN, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testFlex() { + String appName = "example-app"; + HashMap<String, String> componentCounts = new HashMap<String, String>(); + try { + int result = asc.actionFlex(appName, componentCounts); + assertEquals(EXIT_SUCCESS, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testBadFlex() { + String appName = "unknown_app"; + HashMap<String, String> componentCounts = new HashMap<String, String>(); + try { + int result = badAsc.actionFlex(appName, componentCounts); + assertEquals(EXIT_EXCEPTION_THROWN, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testDestroy() { + String appName = "example-app"; + try { + int result = asc.actionDestroy(appName); + assertEquals(EXIT_SUCCESS, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testBadDestroy() { + String appName = "unknown_app"; + try { + int result = badAsc.actionDestroy(appName); + assertEquals(EXIT_EXCEPTION_THROWN, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testInitiateServiceUpgrade() { + String appName = "example-app"; + String upgradeFileName = "target/test-classes/example-app.json"; + try { + int result = asc.initiateUpgrade(appName, upgradeFileName, false); + assertEquals(EXIT_SUCCESS, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testInstancesUpgrade() { + String appName = "example-app"; + try { + int result = asc.actionUpgradeInstances(appName, Lists.newArrayList( + "comp-1", "comp-2")); + assertEquals(EXIT_SUCCESS, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + @Test + public void testComponentsUpgrade() { + String appName = "example-app"; + try { + int result = asc.actionUpgradeComponents(appName, Lists.newArrayList( + "comp")); + assertEquals(EXIT_SUCCESS, result); + } catch (IOException | YarnException e) { + fail(); + } + } + + +} http://git-wip-us.apache.org/repos/asf/hadoop/blob/a23ff8d8/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/client/TestSystemServiceManagerImpl.java ---------------------------------------------------------------------- diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/client/TestSystemServiceManagerImpl.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/client/TestSystemServiceManagerImpl.java new file mode 100644 index 0000000..d39083d --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-services/hadoop-yarn-services-api/src/test/java/org/apache/hadoop/yarn/service/client/TestSystemServiceManagerImpl.java @@ -0,0 +1,182 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.hadoop.yarn.service.client; + +import org.apache.hadoop.conf.Configuration; +import org.apache.hadoop.security.UserGroupInformation; +import org.apache.hadoop.yarn.api.records.ApplicationId; +import org.apache.hadoop.yarn.exceptions.YarnException; +import org.apache.hadoop.yarn.service.api.records.Service; +import org.apache.hadoop.yarn.service.conf.YarnServiceConf; +import org.junit.After; +import org.junit.Assert; +import org.junit.Before; +import org.junit.Test; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import java.io.File; +import java.io.IOException; +import java.util.HashMap; +import java.util.HashSet; +import java.util.Iterator; +import java.util.Map; +import java.util.Set; + +/** + * Test class for system service manager. + */ +public class TestSystemServiceManagerImpl { + + private static final Logger LOG = + LoggerFactory.getLogger(TestSystemServiceManagerImpl.class); + private SystemServiceManagerImpl systemService; + private Configuration conf; + private String resourcePath = "system-services"; + + private String[] users = new String[] {"user1", "user2"}; + private static Map<String, Set<String>> loadedServices = new HashMap<>(); + private static Map<String, Set<String>> submittedServices = new HashMap<>(); + + @Before + public void setup() { + File file = new File( + getClass().getClassLoader().getResource(resourcePath).getFile()); + conf = new Configuration(); + conf.set(YarnServiceConf.YARN_SERVICES_SYSTEM_SERVICE_DIRECTORY, + file.getAbsolutePath()); + systemService = new SystemServiceManagerImpl() { + @Override ServiceClient getServiceClient() { + return new TestServiceClient(); + } + }; + systemService.init(conf); // do not call explicit start + + constructUserService(users[0], "example-app1"); + constructUserService(users[1], "example-app1", "example-app2"); + } + + @After + public void teadDown() { + systemService.stop(); + } + + @Test + public void testSystemServiceSubmission() throws Exception { + systemService.start(); + + /* verify for ignored sevices count */ + Map<String, Integer> ignoredUserServices = + systemService.getIgnoredUserServices(); + Assert.assertEquals(1, ignoredUserServices.size()); + Assert.assertTrue("User user1 doesn't exist.", + ignoredUserServices.containsKey(users[0])); + int count = ignoredUserServices.get(users[0]); + Assert.assertEquals(1, count); + Assert.assertEquals(1, + systemService.getBadFileNameExtensionSkipCounter()); + Assert.assertEquals(1, systemService.getBadDirSkipCounter()); + + Map<String, Set<Service>> userServices = + systemService.getSyncUserServices(); + Assert.assertEquals(loadedServices.size(), userServices.size()); + verifyForScannedUserServices(userServices); + + verifyForLaunchedUserServices(); + + // 2nd time launch service to handle if service exist scenario + systemService.launchUserService(userServices); + verifyForLaunchedUserServices(); + } + + private void verifyForScannedUserServices( + Map<String, Set<Service>> userServices) { + for (String user : users) { + Set<Service> services = userServices.get(user); + Set<String> serviceNames = loadedServices.get(user); + Assert.assertEquals(serviceNames.size(), services.size()); + Iterator<Service> iterator = services.iterator(); + while (iterator.hasNext()) { + Service next = iterator.next(); + Assert.assertTrue( + "Service name doesn't exist in expected userService " + + serviceNames, serviceNames.contains(next.getName())); + } + } + } + + public void constructUserService(String user, String... serviceNames) { + Set<String> service = loadedServices.get(user); + if (service == null) { + service = new HashSet<>(); + for (String serviceName : serviceNames) { + service.add(serviceName); + } + loadedServices.put(user, service); + } + } + + class TestServiceClient extends ServiceClient { + @Override + protected void serviceStart() throws Exception { + // do nothing + } + + @Override + protected void serviceStop() throws Exception { + // do nothing + } + + @Override + protected void serviceInit(Configuration configuration) + throws Exception { + // do nothing + } + + @Override + public ApplicationId actionCreate(Service service) + throws YarnException, IOException { + String userName = + UserGroupInformation.getCurrentUser().getShortUserName(); + Set<String> services = submittedServices.get(userName); + if (services == null) { + services = new HashSet<>(); + submittedServices.put(userName, services); + } + if (services.contains(service.getName())) { + String message = "Failed to create service " + service.getName() + + ", because it already exists."; + throw new YarnException(message); + } + services.add(service.getName()); + return ApplicationId.newInstance(System.currentTimeMillis(), 1); + } + } + + private void verifyForLaunchedUserServices() { + Assert.assertEquals(loadedServices.size(), submittedServices.size()); + for (Map.Entry<String, Set<String>> entry : submittedServices.entrySet()) { + String user = entry.getKey(); + Set<String> serviceSet = entry.getValue(); + Assert.assertTrue(loadedServices.containsKey(user)); + Set<String> services = loadedServices.get(user); + Assert.assertEquals(services.size(), serviceSet.size()); + Assert.assertTrue(services.containsAll(serviceSet)); + } + } +} --------------------------------------------------------------------- To unsubscribe, e-mail: common-commits-unsubscr...@hadoop.apache.org For additional commands, e-mail: common-commits-h...@hadoop.apache.org
