├── .gitignore ├── .travis.yml ├── CHANGELOG.md ├── LICENSE ├── README.md ├── docker ├── Dockerfile └── docker-entrypoint.sh ├── docs └── examples │ ├── hello-world-docker.json │ ├── hello-world.json │ ├── hello-world.properties │ └── kibana-minimesos.properties ├── mvnw ├── mvnw.cmd ├── pom.xml └── src └── main └── java └── com └── containersolutions └── mesos ├── HealthCheckFactory.java ├── MesosFramework.java ├── config └── autoconfigure │ ├── FrameworkSchedulerConfiguration.java │ └── TaskActuatorConfiguration.java ├── controllers └── CountController.java └── scheduler └── config └── FrameworkMesosConfigProperties.java /.gitignore: -------------------------------------------------------------------------------- 1 | *.class 2 | 3 | # Mobile Tools for Java (J2ME) 4 | .mtj.tmp/ 5 | 6 | # Package Files # 7 | *.jar 8 | *.war 9 | *.ear 10 | 11 | # virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml 12 | hs_err_pid* 13 | 14 | .idea 15 | *.iml 16 | target 17 | -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- 1 | language: java 2 | sudo: required 3 | services: 4 | - docker 5 | jdk: 6 | - oraclejdk8 7 | 8 | after_success: 9 | - "mvn -DskipTests package" 10 | - export JAR_FILE=$(ls target/*.jar) 11 | - cp $JAR_FILE $TRAVIS_BUILD_DIR/docker 12 | - docker login -e $DOCKER_EMAIL -u $DOCKER_USER -p $DOCKER_PASS 13 | - export REPO=containersol/mesosframework 14 | - export TAG=`if [ "$TRAVIS_BRANCH" == "master" ]; then echo "snapshot"; else echo $TRAVIS_BRANCH ; fi` 15 | - docker build -t $REPO:$COMMIT $TRAVIS_BUILD_DIR/docker 16 | - docker tag $REPO:$COMMIT $REPO:$TAG 17 | - docker tag $REPO:$COMMIT $REPO:travis-$TRAVIS_BUILD_NUMBER 18 | - docker push $REPO 19 | 20 | before_deploy: 21 | - echo $(ls $TRAVIS_BUILD_DIR) 22 | deploy: 23 | provider: releases 24 | api_key: "$GITHUB_OAUTH_TOKEN" 25 | file: $JAR_FILE 26 | skip_cleanup: true 27 | on: 28 | tags: true 29 | -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- 1 | # [0.1.0](https://github.com/ContainerSolutions/mesosframework/releases/tag/0.1.0) 2 | 3 | Initial release. -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | Apache License 2 | Version 2.0, January 2004 3 | http://www.apache.org/licenses/ 4 | 5 | TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 6 | 7 | 1. Definitions. 8 | 9 | "License" shall mean the terms and conditions for use, reproduction, 10 | and distribution as defined by Sections 1 through 9 of this document. 11 | 12 | "Licensor" shall mean the copyright owner or entity authorized by 13 | the copyright owner that is granting the License. 14 | 15 | "Legal Entity" shall mean the union of the acting entity and all 16 | other entities that control, are controlled by, or are under common 17 | control with that entity. For the purposes of this definition, 18 | "control" means (i) the power, direct or indirect, to cause the 19 | direction or management of such entity, whether by contract or 20 | otherwise, or (ii) ownership of fifty percent (50%) or more of the 21 | outstanding shares, or (iii) beneficial ownership of such entity. 22 | 23 | "You" (or "Your") shall mean an individual or Legal Entity 24 | exercising permissions granted by this License. 25 | 26 | "Source" form shall mean the preferred form for making modifications, 27 | including but not limited to software source code, documentation 28 | source, and configuration files. 29 | 30 | "Object" form shall mean any form resulting from mechanical 31 | transformation or translation of a Source form, including but 32 | not limited to compiled object code, generated documentation, 33 | and conversions to other media types. 34 | 35 | "Work" shall mean the work of authorship, whether in Source or 36 | Object form, made available under the License, as indicated by a 37 | copyright notice that is included in or attached to the work 38 | (an example is provided in the Appendix below). 39 | 40 | "Derivative Works" shall mean any work, whether in Source or Object 41 | form, that is based on (or derived from) the Work and for which the 42 | editorial revisions, annotations, elaborations, or other modifications 43 | represent, as a whole, an original work of authorship. For the purposes 44 | of this License, Derivative Works shall not include works that remain 45 | separable from, or merely link (or bind by name) to the interfaces of, 46 | the Work and Derivative Works thereof. 47 | 48 | "Contribution" shall mean any work of authorship, including 49 | the original version of the Work and any modifications or additions 50 | to that Work or Derivative Works thereof, that is intentionally 51 | submitted to Licensor for inclusion in the Work by the copyright owner 52 | or by an individual or Legal Entity authorized to submit on behalf of 53 | the copyright owner. For the purposes of this definition, "submitted" 54 | means any form of electronic, verbal, or written communication sent 55 | to the Licensor or its representatives, including but not limited to 56 | communication on electronic mailing lists, source code control systems, 57 | and issue tracking systems that are managed by, or on behalf of, the 58 | Licensor for the purpose of discussing and improving the Work, but 59 | excluding communication that is conspicuously marked or otherwise 60 | designated in writing by the copyright owner as "Not a Contribution." 61 | 62 | "Contributor" shall mean Licensor and any individual or Legal Entity 63 | on behalf of whom a Contribution has been received by Licensor and 64 | subsequently incorporated within the Work. 65 | 66 | 2. Grant of Copyright License. Subject to the terms and conditions of 67 | this License, each Contributor hereby grants to You a perpetual, 68 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 69 | copyright license to reproduce, prepare Derivative Works of, 70 | publicly display, publicly perform, sublicense, and distribute the 71 | Work and such Derivative Works in Source or Object form. 72 | 73 | 3. Grant of Patent License. Subject to the terms and conditions of 74 | this License, each Contributor hereby grants to You a perpetual, 75 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 76 | (except as stated in this section) patent license to make, have made, 77 | use, offer to sell, sell, import, and otherwise transfer the Work, 78 | where such license applies only to those patent claims licensable 79 | by such Contributor that are necessarily infringed by their 80 | Contribution(s) alone or by combination of their Contribution(s) 81 | with the Work to which such Contribution(s) was submitted. If You 82 | institute patent litigation against any entity (including a 83 | cross-claim or counterclaim in a lawsuit) alleging that the Work 84 | or a Contribution incorporated within the Work constitutes direct 85 | or contributory patent infringement, then any patent licenses 86 | granted to You under this License for that Work shall terminate 87 | as of the date such litigation is filed. 88 | 89 | 4. Redistribution. You may reproduce and distribute copies of the 90 | Work or Derivative Works thereof in any medium, with or without 91 | modifications, and in Source or Object form, provided that You 92 | meet the following conditions: 93 | 94 | (a) You must give any other recipients of the Work or 95 | Derivative Works a copy of this License; and 96 | 97 | (b) You must cause any modified files to carry prominent notices 98 | stating that You changed the files; and 99 | 100 | (c) You must retain, in the Source form of any Derivative Works 101 | that You distribute, all copyright, patent, trademark, and 102 | attribution notices from the Source form of the Work, 103 | excluding those notices that do not pertain to any part of 104 | the Derivative Works; and 105 | 106 | (d) If the Work includes a "NOTICE" text file as part of its 107 | distribution, then any Derivative Works that You distribute must 108 | include a readable copy of the attribution notices contained 109 | within such NOTICE file, excluding those notices that do not 110 | pertain to any part of the Derivative Works, in at least one 111 | of the following places: within a NOTICE text file distributed 112 | as part of the Derivative Works; within the Source form or 113 | documentation, if provided along with the Derivative Works; or, 114 | within a display generated by the Derivative Works, if and 115 | wherever such third-party notices normally appear. The contents 116 | of the NOTICE file are for informational purposes only and 117 | do not modify the License. You may add Your own attribution 118 | notices within Derivative Works that You distribute, alongside 119 | or as an addendum to the NOTICE text from the Work, provided 120 | that such additional attribution notices cannot be construed 121 | as modifying the License. 122 | 123 | You may add Your own copyright statement to Your modifications and 124 | may provide additional or different license terms and conditions 125 | for use, reproduction, or distribution of Your modifications, or 126 | for any such Derivative Works as a whole, provided Your use, 127 | reproduction, and distribution of the Work otherwise complies with 128 | the conditions stated in this License. 129 | 130 | 5. Submission of Contributions. Unless You explicitly state otherwise, 131 | any Contribution intentionally submitted for inclusion in the Work 132 | by You to the Licensor shall be under the terms and conditions of 133 | this License, without any additional terms or conditions. 134 | Notwithstanding the above, nothing herein shall supersede or modify 135 | the terms of any separate license agreement you may have executed 136 | with Licensor regarding such Contributions. 137 | 138 | 6. Trademarks. This License does not grant permission to use the trade 139 | names, trademarks, service marks, or product names of the Licensor, 140 | except as required for reasonable and customary use in describing the 141 | origin of the Work and reproducing the content of the NOTICE file. 142 | 143 | 7. Disclaimer of Warranty. Unless required by applicable law or 144 | agreed to in writing, Licensor provides the Work (and each 145 | Contributor provides its Contributions) on an "AS IS" BASIS, 146 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or 147 | implied, including, without limitation, any warranties or conditions 148 | of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A 149 | PARTICULAR PURPOSE. You are solely responsible for determining the 150 | appropriateness of using or redistributing the Work and assume any 151 | risks associated with Your exercise of permissions under this License. 152 | 153 | 8. Limitation of Liability. In no event and under no legal theory, 154 | whether in tort (including negligence), contract, or otherwise, 155 | unless required by applicable law (such as deliberate and grossly 156 | negligent acts) or agreed to in writing, shall any Contributor be 157 | liable to You for damages, including any direct, indirect, special, 158 | incidental, or consequential damages of any character arising as a 159 | result of this License or out of the use or inability to use the 160 | Work (including but not limited to damages for loss of goodwill, 161 | work stoppage, computer failure or malfunction, or any and all 162 | other commercial damages or losses), even if such Contributor 163 | has been advised of the possibility of such damages. 164 | 165 | 9. Accepting Warranty or Additional Liability. While redistributing 166 | the Work or Derivative Works thereof, You may choose to offer, 167 | and charge a fee for, acceptance of support, warranty, indemnity, 168 | or other liability obligations and/or rights consistent with this 169 | License. However, in accepting such obligations, You may act only 170 | on Your own behalf and on Your sole responsibility, not on behalf 171 | of any other Contributor, and only if You agree to indemnify, 172 | defend, and hold each Contributor harmless for any liability 173 | incurred by, or claims asserted against, such Contributor by reason 174 | of your accepting any such warranty or additional liability. 175 | 176 | END OF TERMS AND CONDITIONS 177 | 178 | APPENDIX: How to apply the Apache License to your work. 179 | 180 | To apply the Apache License to your work, attach the following 181 | boilerplate notice, with the fields enclosed by brackets "{}" 182 | replaced with your own identifying information. (Don't include 183 | the brackets!) The text should be enclosed in the appropriate 184 | comment syntax for the file format. We also recommend that a 185 | file or class name and description of purpose be included on the 186 | same "printed page" as the copyright notice for easier 187 | identification within third-party archives. 188 | 189 | Copyright {yyyy} {name of copyright owner} 190 | 191 | Licensed under the Apache License, Version 2.0 (the "License"); 192 | you may not use this file except in compliance with the License. 193 | You may obtain a copy of the License at 194 | 195 | http://www.apache.org/licenses/LICENSE-2.0 196 | 197 | Unless required by applicable law or agreed to in writing, software 198 | distributed under the License is distributed on an "AS IS" BASIS, 199 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 200 | See the License for the specific language governing permissions and 201 | limitations under the License. 202 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # mesos-framework 2 | Create a mesos framework with just an application.properties file! 3 | 4 | # Introduction 5 | Writing Frameworks for Apache Mesos is hard. We created the [Mesos Starter](www.github.com/containersolutions/mesos-starter) project to reduce the amount of boilerplate. 6 | 7 | MesosFramework is a pre-built version of Mesos-Starter, that allows users to run a binary or Docker container as a Framework via a single configuration file. 8 | 9 | # Features 10 | (Most features come from the [Mesos-Starter](https://github.com/ContainerSolutions/mesos-starter) project) 11 | 12 | - [x] State stored in ZooKeeper 13 | - [x] Mesos Authorisation 14 | - [ ] ZooKeeper Authorisation (requires testing) 15 | - [x] Jar mode (no docker) 16 | - [x] Resource specification (including port) 17 | - [x] Import Kibana.yml settings file 18 | - [x] "Spread" orchestration strategy (Spreads instances across distinct hosts) 19 | - [x] Decoupled from Kibana. Use any version. 20 | - [x] Decoupled from Mesos. Use any version 0.25+ 21 | - [x] Horizontal scaling via JMX 22 | 23 | Added by MesosFramework: 24 | - [x] Live horizontal scaling via a REST endpoint 25 | - [x] Single endpoint to check health of all instances 26 | - [x] Customizable mesos healthcheck command 27 | 28 | # Quick start 29 | Once you have decided upon a binary or Docker container to run, you simply need to call: 30 | 31 | `docker run -d containersol/mesosframework:latest --spring.config.location=your.application.properties` 32 | 33 | or if you don't want to use Docker: 34 | 35 | `java -jar mesosframework-0.0.1.jar --spring.config.location=your.application.properties` 36 | 37 | # Application properties 38 | All options can be specified as either: 39 | - A cli parameter: `--mesos.command=pwd` 40 | - A properties file: `mesos.command=pwd` 41 | - Java options: `-Dmesos.command=pwd` 42 | - Environmental variables: `MESOS_COMMAND=pwd` 43 | In that order of preference. 44 | 45 | To pass a configuration file, the following property must be set: 46 | - `--spring.config.location=my.properties` (Or the env var `SPRING_CONFIG_LOCATION`, etc.) 47 | 48 | Valid configration options are specified by the [Mesos Starter](www.github.com/containersolutions/mesos-starter) project. 49 | 50 | Please see the [examples](./docs/examples) folder for examples. 51 | 52 | You can also provide new options with `${my.option}` inside the properties file. Now you can provide that option with `--my.option=Hi!` 53 | 54 | # Extensions to the Mesos-starter project 55 | This project adds to the Mesos-starter project on the following points: 56 | 57 | ## Horizontal scaling 58 | It adds an endpoint at the following location to control the number of instances in the cluster. The endpoint matches the properties file definition of the same name: 59 | 60 | `GET /mesos/resources/count` Returns the current number of requested instances. For example to get the current number of instances: 61 | 62 | ``` 63 | $ curl -s http://localhost:8080/mesos/resources/count 64 | 3 65 | ``` 66 | 67 | `POST /mesos/resources/count` with a body of type `Integer` will set the number of requested instances. For example, to set the number of instances to 1: 68 | 69 | ``` 70 | $ curl -XPOST -H 'Content-Type: text/plain' http://localhost:8080/mesos/resources/count -d 1 71 | ``` 72 | 73 | ## Healthcheck 74 | An extra healthchecks to monitor the state of the running tasks is exposed. 75 | 76 | `GET /health` returns the health of the running tasks. If `count == running instances` the health will be `UP`. For example: 77 | 78 | ``` 79 | $ curl http://localhost:8080/health 80 | {"status":"UP","task":{"status":"UP","mesos.resources.count":1,"instances":1,"TASK_STAGING":0,"TASK_STARTING":0,"TASK_RUNNING":3,"TASK_KILLING":0,"TASK_FINISHED":0,"TASK_FAILED":0,"TASK_KILLED":0,"TASK_LOST":0,"TASK_ERROR":0},"diskSpace":{"status":"UP","total":9896046592,"free":2796249088,"threshold":10485760}}% 81 | ``` 82 | 83 | # Sponsors 84 | This project is sponsored by Cisco Cloud Services. 85 | 86 | # License 87 | Apache License 2.0 -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- 1 | FROM ubuntu:14.04 2 | 3 | RUN echo "deb http://ppa.launchpad.net/openjdk-r/ppa/ubuntu trusty main" > /etc/apt/sources.list.d/openjdk.list && \ 4 | apt-key adv --keyserver keyserver.ubuntu.com --recv 86F44E2A && \ 5 | apt-get update && \ 6 | apt-get -y install openjdk-8-jre-headless && \ 7 | rm -rf /var/lib/apt/lists/* 8 | 9 | RUN echo "deb http://repos.mesosphere.io/ubuntu/ trusty main" > /etc/apt/sources.list.d/mesosphere.list && \ 10 | apt-key adv --keyserver keyserver.ubuntu.com --recv E56151BF && \ 11 | apt-get -y update && \ 12 | apt-get -y install mesos=0.25.0-0.2.70.ubuntu1404 && \ 13 | rm -rf /var/lib/apt/lists/* 14 | 15 | # Clean up APT when done. 16 | RUN apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* 17 | 18 | ADD mesosframework-*.jar /tmp/mesosframework.jar 19 | ADD docker-entrypoint.sh /tmp/docker-entrypoint.sh 20 | 21 | # Use baseimage-docker's init system. 22 | ENTRYPOINT ["/tmp/docker-entrypoint.sh"] -------------------------------------------------------------------------------- /docker/docker-entrypoint.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | set -x 3 | exec java $JAVA_OPTS -jar /tmp/mesosframework.jar "$@" 4 | -------------------------------------------------------------------------------- /docs/examples/hello-world-docker.json: -------------------------------------------------------------------------------- 1 | { 2 | "id": "hello-world-docker", 3 | "cpus": 0.1, 4 | "mem": 128, 5 | "instances": 1, 6 | "args": [ 7 | "--spring.application.name=hello-world-docker", 8 | "--mesos.framework.name=hello-world-docker", 9 | "--mesos.master=zk://$ZOOKEEPER:2181/mesos", 10 | "--mesos.docker.image=training/webapp:latest", 11 | "--mesos.docker.network=BRIDGE", 12 | "--mesos.resources.mem=64", 13 | "--mesos.resources.cpu=0.1", 14 | "--mesos.resources.count=3", 15 | "--mesos.command=python app.py", 16 | "--logging.level.com.containersolutions.mesos=DEBUG", 17 | "--mesos.zookeeper.server=$ZOOKEEPER:2181", 18 | "--server.port=31100" 19 | ], 20 | "env": { 21 | "JAVA_OPTS": "-Xms32m -Xmx100m" 22 | }, 23 | "container": { 24 | "type": "DOCKER", 25 | "docker": { 26 | "image": "containersol/mesosframework:0.1.0", 27 | "network": "BRIDGE" 28 | } 29 | }, 30 | "ports": [ 31 | 31100 32 | ], 33 | "requirePorts": true 34 | } -------------------------------------------------------------------------------- /docs/examples/hello-world.json: -------------------------------------------------------------------------------- 1 | { 2 | "id": "hello-world", 3 | "cpus": 0.1, 4 | "mem": 64, 5 | "instances": 1, 6 | "cmd": "unset MESOS_RESOURCES ; mv *.jar scheduler.jar ; java -jar scheduler.jar --spring.config.location=./hello-world.properties --mesos.zookeeper.server=$ZOOKEEPER:2181", 7 | "uris": [ 8 | "https://github.com/ContainerSolutions/mesosframework/releases/download/0.1.0/mesosframework-0.1.0.jar", 9 | "https://raw.githubusercontent.com/ContainerSolutions/mesosframework/master/docs/examples/hello-world.properties" 10 | ], 11 | "env": { 12 | "JAVA_OPTS": "-Xms32m -Xmx64m" 13 | } 14 | } 15 | -------------------------------------------------------------------------------- /docs/examples/hello-world.properties: -------------------------------------------------------------------------------- 1 | # Introduction 2 | # This example uses minimesos to run a simple hello world webapp. 3 | # First, start minimesos with three agents. 4 | 5 | # Required application name for Spring 6 | spring.application.name=hello-world 7 | 8 | # Mesos framework name 9 | mesos.framework.name=hello-world 10 | 11 | # Mesos settings 12 | mesos.master=zk://${mesos.zookeeper.server}/mesos 13 | # mesos.zookeeper.server=$ZOOKEEPER:2181 14 | 15 | # Task resources 16 | mesos.resources.cpus=0.1 17 | mesos.resources.mem=64 18 | mesos.resources.count=3 19 | 20 | # Docker image 21 | mesos.docker.image=training/webapp:latest 22 | mesos.docker.network=BRIDGE 23 | 24 | # Command to run 25 | mesos.command=python app.py 26 | 27 | # Debugging 28 | logging.level.com.containersolutions.mesos=DEBUG -------------------------------------------------------------------------------- /docs/examples/kibana-minimesos.properties: -------------------------------------------------------------------------------- 1 | # Introduction 2 | # This example uses minimesos to run kibana. 3 | # First, start minimesos with three agents, and then start an elasticsearch instance with `docker run -d elasticsearch:latest` 4 | 5 | # Required application name for Spring 6 | spring.application.name=kibana-minimesos 7 | 8 | # Mesos framework name 9 | mesos.framework.name=kibana-minimesos 10 | 11 | # Mesos settings 12 | mesos.master=zk://${mesos.zookeeper.server}/mesos 13 | #mesos.zookeeper.server=172.17.0.3:2181 # Usually injected. 14 | 15 | # ES location 16 | # elasticsearch.http=http://172.17.0.2:9200 # Usually injected. 17 | 18 | # Task resources 19 | mesos.resources.cpus=0.5 20 | mesos.resources.mem=256 21 | mesos.resources.count=3 22 | 23 | # Docker mode 24 | #mesos.docker.image=kibana:latest 25 | #mesos.docker.network=BRIDGE 26 | #mesos.command=kibana --elasticsearch ${elasticsearch.http} 27 | #mesos.healthCheck.command=wget -q localhost:5061 28 | 29 | # Command mode 30 | mesos.resources.ports.UI_5061.host=ANY 31 | mesos.uri[0]=https://gist.githubusercontent.com/philwinder/592a1ab2db40431c1b08/raw/kibana.yml 32 | mesos.uri[1]=https://download.elastic.co/kibana/kibana/kibana-4.4.1-linux-x64.tar.gz 33 | mesos.command= cp $MESOS_SANDBOX/kibana.yml $MESOS_SANDBOX/kibana-*/config/kibana.yml ; cd kibana-* ; bin/kibana --port=$UI_5061 --elasticsearch ${elasticsearch.http} 34 | mesos.healthCheck.command=wget -q localhost:$UI_5061 35 | 36 | # Log level 37 | logging.level.com.containersolutions.mesos=DEBUG 38 | -------------------------------------------------------------------------------- /mvnw: -------------------------------------------------------------------------------- 1 | #!/bin/sh 2 | # ---------------------------------------------------------------------------- 3 | # Licensed to the Apache Software Foundation (ASF) under one 4 | # or more contributor license agreements. See the NOTICE file 5 | # distributed with this work for additional information 6 | # regarding copyright ownership. The ASF licenses this file 7 | # to you under the Apache License, Version 2.0 (the 8 | # "License"); you may not use this file except in compliance 9 | # with the License. You may obtain a copy of the License at 10 | # 11 | # http://www.apache.org/licenses/LICENSE-2.0 12 | # 13 | # Unless required by applicable law or agreed to in writing, 14 | # software distributed under the License is distributed on an 15 | # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 16 | # KIND, either express or implied. See the License for the 17 | # specific language governing permissions and limitations 18 | # under the License. 19 | # ---------------------------------------------------------------------------- 20 | 21 | # ---------------------------------------------------------------------------- 22 | # Maven2 Start Up Batch script 23 | # 24 | # Required ENV vars: 25 | # ------------------ 26 | # JAVA_HOME - location of a JDK home dir 27 | # 28 | # Optional ENV vars 29 | # ----------------- 30 | # M2_HOME - location of maven2's installed home dir 31 | # MAVEN_OPTS - parameters passed to the Java VM when running Maven 32 | # e.g. to debug Maven itself, use 33 | # set MAVEN_OPTS=-Xdebug -Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=8000 34 | # MAVEN_SKIP_RC - flag to disable loading of mavenrc files 35 | # ---------------------------------------------------------------------------- 36 | 37 | if [ -z "$MAVEN_SKIP_RC" ] ; then 38 | 39 | if [ -f /etc/mavenrc ] ; then 40 | . /etc/mavenrc 41 | fi 42 | 43 | if [ -f "$HOME/.mavenrc" ] ; then 44 | . "$HOME/.mavenrc" 45 | fi 46 | 47 | fi 48 | 49 | # OS specific support. $var _must_ be set to either true or false. 50 | cygwin=false; 51 | darwin=false; 52 | mingw=false 53 | case "`uname`" in 54 | CYGWIN*) cygwin=true ;; 55 | MINGW*) mingw=true;; 56 | Darwin*) darwin=true 57 | # 58 | # Look for the Apple JDKs first to preserve the existing behaviour, and then look 59 | # for the new JDKs provided by Oracle. 60 | # 61 | if [ -z "$JAVA_HOME" ] && [ -L /System/Library/Frameworks/JavaVM.framework/Versions/CurrentJDK ] ; then 62 | # 63 | # Apple JDKs 64 | # 65 | export JAVA_HOME=/System/Library/Frameworks/JavaVM.framework/Versions/CurrentJDK/Home 66 | fi 67 | 68 | if [ -z "$JAVA_HOME" ] && [ -L /System/Library/Java/JavaVirtualMachines/CurrentJDK ] ; then 69 | # 70 | # Apple JDKs 71 | # 72 | export JAVA_HOME=/System/Library/Java/JavaVirtualMachines/CurrentJDK/Contents/Home 73 | fi 74 | 75 | if [ -z "$JAVA_HOME" ] && [ -L "/Library/Java/JavaVirtualMachines/CurrentJDK" ] ; then 76 | # 77 | # Oracle JDKs 78 | # 79 | export JAVA_HOME=/Library/Java/JavaVirtualMachines/CurrentJDK/Contents/Home 80 | fi 81 | 82 | if [ -z "$JAVA_HOME" ] && [ -x "/usr/libexec/java_home" ]; then 83 | # 84 | # Apple JDKs 85 | # 86 | export JAVA_HOME=`/usr/libexec/java_home` 87 | fi 88 | ;; 89 | esac 90 | 91 | if [ -z "$JAVA_HOME" ] ; then 92 | if [ -r /etc/gentoo-release ] ; then 93 | JAVA_HOME=`java-config --jre-home` 94 | fi 95 | fi 96 | 97 | if [ -z "$M2_HOME" ] ; then 98 | ## resolve links - $0 may be a link to maven's home 99 | PRG="$0" 100 | 101 | # need this for relative symlinks 102 | while [ -h "$PRG" ] ; do 103 | ls=`ls -ld "$PRG"` 104 | link=`expr "$ls" : '.*-> \(.*\)$'` 105 | if expr "$link" : '/.*' > /dev/null; then 106 | PRG="$link" 107 | else 108 | PRG="`dirname "$PRG"`/$link" 109 | fi 110 | done 111 | 112 | saveddir=`pwd` 113 | 114 | M2_HOME=`dirname "$PRG"`/.. 115 | 116 | # make it fully qualified 117 | M2_HOME=`cd "$M2_HOME" && pwd` 118 | 119 | cd "$saveddir" 120 | # echo Using m2 at $M2_HOME 121 | fi 122 | 123 | # For Cygwin, ensure paths are in UNIX format before anything is touched 124 | if $cygwin ; then 125 | [ -n "$M2_HOME" ] && 126 | M2_HOME=`cygpath --unix "$M2_HOME"` 127 | [ -n "$JAVA_HOME" ] && 128 | JAVA_HOME=`cygpath --unix "$JAVA_HOME"` 129 | [ -n "$CLASSPATH" ] && 130 | CLASSPATH=`cygpath --path --unix "$CLASSPATH"` 131 | fi 132 | 133 | # For Migwn, ensure paths are in UNIX format before anything is touched 134 | if $mingw ; then 135 | [ -n "$M2_HOME" ] && 136 | M2_HOME="`(cd "$M2_HOME"; pwd)`" 137 | [ -n "$JAVA_HOME" ] && 138 | JAVA_HOME="`(cd "$JAVA_HOME"; pwd)`" 139 | # TODO classpath? 140 | fi 141 | 142 | if [ -z "$JAVA_HOME" ]; then 143 | javaExecutable="`which javac`" 144 | if [ -n "$javaExecutable" ] && ! [ "`expr \"$javaExecutable\" : '\([^ ]*\)'`" = "no" ]; then 145 | # readlink(1) is not available as standard on Solaris 10. 146 | readLink=`which readlink` 147 | if [ ! `expr "$readLink" : '\([^ ]*\)'` = "no" ]; then 148 | if $darwin ; then 149 | javaHome="`dirname \"$javaExecutable\"`" 150 | javaExecutable="`cd \"$javaHome\" && pwd -P`/javac" 151 | else 152 | javaExecutable="`readlink -f \"$javaExecutable\"`" 153 | fi 154 | javaHome="`dirname \"$javaExecutable\"`" 155 | javaHome=`expr "$javaHome" : '\(.*\)/bin'` 156 | JAVA_HOME="$javaHome" 157 | export JAVA_HOME 158 | fi 159 | fi 160 | fi 161 | 162 | if [ -z "$JAVACMD" ] ; then 163 | if [ -n "$JAVA_HOME" ] ; then 164 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then 165 | # IBM's JDK on AIX uses strange locations for the executables 166 | JAVACMD="$JAVA_HOME/jre/sh/java" 167 | else 168 | JAVACMD="$JAVA_HOME/bin/java" 169 | fi 170 | else 171 | JAVACMD="`which java`" 172 | fi 173 | fi 174 | 175 | if [ ! -x "$JAVACMD" ] ; then 176 | echo "Error: JAVA_HOME is not defined correctly." >&2 177 | echo " We cannot execute $JAVACMD" >&2 178 | exit 1 179 | fi 180 | 181 | if [ -z "$JAVA_HOME" ] ; then 182 | echo "Warning: JAVA_HOME environment variable is not set." 183 | fi 184 | 185 | CLASSWORLDS_LAUNCHER=org.codehaus.plexus.classworlds.launcher.Launcher 186 | 187 | # For Cygwin, switch paths to Windows format before running java 188 | if $cygwin; then 189 | [ -n "$M2_HOME" ] && 190 | M2_HOME=`cygpath --path --windows "$M2_HOME"` 191 | [ -n "$JAVA_HOME" ] && 192 | JAVA_HOME=`cygpath --path --windows "$JAVA_HOME"` 193 | [ -n "$CLASSPATH" ] && 194 | CLASSPATH=`cygpath --path --windows "$CLASSPATH"` 195 | fi 196 | 197 | # traverses directory structure from process work directory to filesystem root 198 | # first directory with .mvn subdirectory is considered project base directory 199 | find_maven_basedir() { 200 | local basedir=$(pwd) 201 | local wdir=$(pwd) 202 | while [ "$wdir" != '/' ] ; do 203 | if [ -d "$wdir"/.mvn ] ; then 204 | basedir=$wdir 205 | break 206 | fi 207 | wdir=$(cd "$wdir/.."; pwd) 208 | done 209 | echo "${basedir}" 210 | } 211 | 212 | # concatenates all lines of a file 213 | concat_lines() { 214 | if [ -f "$1" ]; then 215 | echo "$(tr -s '\n' ' ' < "$1")" 216 | fi 217 | } 218 | 219 | export MAVEN_PROJECTBASEDIR=${MAVEN_BASEDIR:-$(find_maven_basedir)} 220 | MAVEN_OPTS="$(concat_lines "$MAVEN_PROJECTBASEDIR/.mvn/jvm.config") $MAVEN_OPTS" 221 | 222 | # Provide a "standardized" way to retrieve the CLI args that will 223 | # work with both Windows and non-Windows executions. 224 | MAVEN_CMD_LINE_ARGS="$MAVEN_CONFIG $@" 225 | export MAVEN_CMD_LINE_ARGS 226 | 227 | WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain 228 | 229 | exec "$JAVACMD" \ 230 | $MAVEN_OPTS \ 231 | -classpath "$MAVEN_PROJECTBASEDIR/.mvn/wrapper/maven-wrapper.jar" \ 232 | "-Dmaven.home=${M2_HOME}" "-Dmaven.multiModuleProjectDirectory=${MAVEN_PROJECTBASEDIR}" \ 233 | ${WRAPPER_LAUNCHER} "$@" 234 | -------------------------------------------------------------------------------- /mvnw.cmd: -------------------------------------------------------------------------------- 1 | @REM ---------------------------------------------------------------------------- 2 | @REM Licensed to the Apache Software Foundation (ASF) under one 3 | @REM or more contributor license agreements. See the NOTICE file 4 | @REM distributed with this work for additional information 5 | @REM regarding copyright ownership. The ASF licenses this file 6 | @REM to you under the Apache License, Version 2.0 (the 7 | @REM "License"); you may not use this file except in compliance 8 | @REM with the License. You may obtain a copy of the License at 9 | @REM 10 | @REM http://www.apache.org/licenses/LICENSE-2.0 11 | @REM 12 | @REM Unless required by applicable law or agreed to in writing, 13 | @REM software distributed under the License is distributed on an 14 | @REM "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 15 | @REM KIND, either express or implied. See the License for the 16 | @REM specific language governing permissions and limitations 17 | @REM under the License. 18 | @REM ---------------------------------------------------------------------------- 19 | 20 | @REM ---------------------------------------------------------------------------- 21 | @REM Maven2 Start Up Batch script 22 | @REM 23 | @REM Required ENV vars: 24 | @REM JAVA_HOME - location of a JDK home dir 25 | @REM 26 | @REM Optional ENV vars 27 | @REM M2_HOME - location of maven2's installed home dir 28 | @REM MAVEN_BATCH_ECHO - set to 'on' to enable the echoing of the batch commands 29 | @REM MAVEN_BATCH_PAUSE - set to 'on' to wait for a key stroke before ending 30 | @REM MAVEN_OPTS - parameters passed to the Java VM when running Maven 31 | @REM e.g. to debug Maven itself, use 32 | @REM set MAVEN_OPTS=-Xdebug -Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=8000 33 | @REM MAVEN_SKIP_RC - flag to disable loading of mavenrc files 34 | @REM ---------------------------------------------------------------------------- 35 | 36 | @REM Begin all REM lines with '@' in case MAVEN_BATCH_ECHO is 'on' 37 | @echo off 38 | @REM enable echoing my setting MAVEN_BATCH_ECHO to 'on' 39 | @if "%MAVEN_BATCH_ECHO%" == "on" echo %MAVEN_BATCH_ECHO% 40 | 41 | @REM set %HOME% to equivalent of $HOME 42 | if "%HOME%" == "" (set "HOME=%HOMEDRIVE%%HOMEPATH%") 43 | 44 | @REM Execute a user defined script before this one 45 | if not "%MAVEN_SKIP_RC%" == "" goto skipRcPre 46 | @REM check for pre script, once with legacy .bat ending and once with .cmd ending 47 | if exist "%HOME%\mavenrc_pre.bat" call "%HOME%\mavenrc_pre.bat" 48 | if exist "%HOME%\mavenrc_pre.cmd" call "%HOME%\mavenrc_pre.cmd" 49 | :skipRcPre 50 | 51 | @setlocal 52 | 53 | set ERROR_CODE=0 54 | 55 | @REM To isolate internal variables from possible post scripts, we use another setlocal 56 | @setlocal 57 | 58 | @REM ==== START VALIDATION ==== 59 | if not "%JAVA_HOME%" == "" goto OkJHome 60 | 61 | echo. 62 | echo Error: JAVA_HOME not found in your environment. >&2 63 | echo Please set the JAVA_HOME variable in your environment to match the >&2 64 | echo location of your Java installation. >&2 65 | echo. 66 | goto error 67 | 68 | :OkJHome 69 | if exist "%JAVA_HOME%\bin\java.exe" goto init 70 | 71 | echo. 72 | echo Error: JAVA_HOME is set to an invalid directory. >&2 73 | echo JAVA_HOME = "%JAVA_HOME%" >&2 74 | echo Please set the JAVA_HOME variable in your environment to match the >&2 75 | echo location of your Java installation. >&2 76 | echo. 77 | goto error 78 | 79 | @REM ==== END VALIDATION ==== 80 | 81 | :init 82 | 83 | set MAVEN_CMD_LINE_ARGS=%* 84 | 85 | @REM Find the project base dir, i.e. the directory that contains the folder ".mvn". 86 | @REM Fallback to current working directory if not found. 87 | 88 | set MAVEN_PROJECTBASEDIR=%MAVEN_BASEDIR% 89 | IF NOT "%MAVEN_PROJECTBASEDIR%"=="" goto endDetectBaseDir 90 | 91 | set EXEC_DIR=%CD% 92 | set WDIR=%EXEC_DIR% 93 | :findBaseDir 94 | IF EXIST "%WDIR%"\.mvn goto baseDirFound 95 | cd .. 96 | IF "%WDIR%"=="%CD%" goto baseDirNotFound 97 | set WDIR=%CD% 98 | goto findBaseDir 99 | 100 | :baseDirFound 101 | set MAVEN_PROJECTBASEDIR=%WDIR% 102 | cd "%EXEC_DIR%" 103 | goto endDetectBaseDir 104 | 105 | :baseDirNotFound 106 | set MAVEN_PROJECTBASEDIR=%EXEC_DIR% 107 | cd "%EXEC_DIR%" 108 | 109 | :endDetectBaseDir 110 | 111 | IF NOT EXIST "%MAVEN_PROJECTBASEDIR%\.mvn\jvm.config" goto endReadAdditionalConfig 112 | 113 | @setlocal EnableExtensions EnableDelayedExpansion 114 | for /F "usebackq delims=" %%a in ("%MAVEN_PROJECTBASEDIR%\.mvn\jvm.config") do set JVM_CONFIG_MAVEN_PROPS=!JVM_CONFIG_MAVEN_PROPS! %%a 115 | @endlocal & set JVM_CONFIG_MAVEN_PROPS=%JVM_CONFIG_MAVEN_PROPS% 116 | 117 | :endReadAdditionalConfig 118 | 119 | SET MAVEN_JAVA_EXE="%JAVA_HOME%\bin\java.exe" 120 | 121 | set WRAPPER_JAR="".\.mvn\wrapper\maven-wrapper.jar"" 122 | set WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain 123 | 124 | %MAVEN_JAVA_EXE% %JVM_CONFIG_MAVEN_PROPS% %MAVEN_OPTS% %MAVEN_DEBUG_OPTS% -classpath %WRAPPER_JAR% "-Dmaven.multiModuleProjectDirectory=%MAVEN_PROJECTBASEDIR%" %WRAPPER_LAUNCHER% %MAVEN_CMD_LINE_ARGS% 125 | if ERRORLEVEL 1 goto error 126 | goto end 127 | 128 | :error 129 | set ERROR_CODE=1 130 | 131 | :end 132 | @endlocal & set ERROR_CODE=%ERROR_CODE% 133 | 134 | if not "%MAVEN_SKIP_RC%" == "" goto skipRcPost 135 | @REM check for post script, once with legacy .bat ending and once with .cmd ending 136 | if exist "%HOME%\mavenrc_post.bat" call "%HOME%\mavenrc_post.bat" 137 | if exist "%HOME%\mavenrc_post.cmd" call "%HOME%\mavenrc_post.cmd" 138 | :skipRcPost 139 | 140 | @REM pause the script if MAVEN_BATCH_PAUSE is set to 'on' 141 | if "%MAVEN_BATCH_PAUSE%" == "on" pause 142 | 143 | if "%MAVEN_TERMINATE_CMD%" == "on" exit %ERROR_CODE% 144 | 145 | exit /B %ERROR_CODE% -------------------------------------------------------------------------------- /pom.xml: -------------------------------------------------------------------------------- 1 | 2 | 4 | 4.0.0 5 | 6 | com.github.containersolutions 7 | mesosframework 8 | 0.1.0 9 | jar 10 | 11 | mesosframework 12 | Apache Mesos Framework for any application 13 | 14 | 15 | org.springframework.boot 16 | spring-boot-starter-parent 17 | 1.3.3.RELEASE 18 | 19 | 20 | 21 | 22 | 23 | jitpack.io 24 | https://jitpack.io 25 | 26 | 27 | 28 | 29 | UTF-8 30 | 1.8 31 | 32 | 33 | 34 | 35 | org.springframework.boot 36 | spring-boot-starter 37 | 38 | 39 | 40 | org.springframework.boot 41 | spring-boot-starter-test 42 | test 43 | 44 | 45 | 46 | org.springframework.boot 47 | spring-boot-actuator 48 | 49 | 50 | 51 | org.springframework.boot 52 | spring-boot-starter-web 53 | 54 | 55 | 56 | com.github.containersolutions.mesos-starter 57 | spring-boot-starter-mesos 58 | 0.2.1 59 | 60 | 61 | 62 | 63 | 64 | 65 | 66 | 67 | 68 | 69 | 70 | 71 | org.springframework.boot 72 | spring-boot-maven-plugin 73 | 74 | 75 | org.springframework.boot 76 | spring-boot-maven-plugin 77 | 78 | 79 | 80 | repackage 81 | 82 | 83 | 84 | 85 | 86 | 87 | 88 | 89 | 90 | -------------------------------------------------------------------------------- /src/main/java/com/containersolutions/mesos/HealthCheckFactory.java: -------------------------------------------------------------------------------- 1 | package com.containersolutions.mesos; 2 | 3 | import com.containersolutions.mesos.scheduler.MesosProtoFactory; 4 | import com.containersolutions.mesos.scheduler.config.FrameworkMesosConfigProperties; 5 | import org.apache.mesos.Protos; 6 | import org.springframework.beans.factory.annotation.Autowired; 7 | 8 | import java.util.List; 9 | 10 | public class HealthCheckFactory implements MesosProtoFactory> { 11 | @Autowired 12 | FrameworkMesosConfigProperties mesosConfig; 13 | 14 | @Override 15 | public Protos.HealthCheck create(List resources) { 16 | // Dev note: HealthCheck does not work in 0.25: https://issues.apache.org/jira/browse/MESOS-3738 17 | Protos.HealthCheck healthDefault = Protos.HealthCheck.getDefaultInstance(); 18 | if (mesosConfig.getHealthCheck() == null) { 19 | return healthDefault; 20 | } else { 21 | return Protos.HealthCheck.newBuilder() 22 | .setCommand( 23 | Protos.CommandInfo.newBuilder() 24 | .setValue(mesosConfig.getHealthCheck().getOrDefault("command", "echo Success")) 25 | ) 26 | .setTimeoutSeconds(Double.valueOf(mesosConfig.getHealthCheck().getOrDefault("timeout", Double.toString(healthDefault.getTimeoutSeconds())))) 27 | .setDelaySeconds(Double.valueOf(mesosConfig.getHealthCheck().getOrDefault("delay", Double.toString(healthDefault.getDelaySeconds())))) 28 | .setConsecutiveFailures(Integer.valueOf(mesosConfig.getHealthCheck().getOrDefault("consecutiveFailures", Integer.toString(healthDefault.getConsecutiveFailures())))) 29 | .setGracePeriodSeconds(Double.valueOf(mesosConfig.getHealthCheck().getOrDefault("gracePeriod", Double.toString(healthDefault.getGracePeriodSeconds())))) 30 | .setIntervalSeconds(Double.valueOf(mesosConfig.getHealthCheck().getOrDefault("interval", Double.toString(healthDefault.getIntervalSeconds())))) 31 | .build(); 32 | } 33 | } 34 | } 35 | -------------------------------------------------------------------------------- /src/main/java/com/containersolutions/mesos/MesosFramework.java: -------------------------------------------------------------------------------- 1 | package com.containersolutions.mesos; 2 | 3 | import org.springframework.boot.SpringApplication; 4 | import org.springframework.boot.autoconfigure.SpringBootApplication; 5 | 6 | @SpringBootApplication 7 | public class MesosFramework { 8 | public static void main(String[] args) { 9 | SpringApplication.run(MesosFramework.class, args); 10 | } 11 | } 12 | -------------------------------------------------------------------------------- /src/main/java/com/containersolutions/mesos/config/autoconfigure/FrameworkSchedulerConfiguration.java: -------------------------------------------------------------------------------- 1 | package com.containersolutions.mesos.config.autoconfigure; 2 | 3 | import com.containersolutions.mesos.HealthCheckFactory; 4 | import com.containersolutions.mesos.controllers.CountController; 5 | import com.containersolutions.mesos.scheduler.InstanceCount; 6 | import com.containersolutions.mesos.scheduler.TaskInfoFactory; 7 | import com.containersolutions.mesos.scheduler.TaskInfoFactoryCommand; 8 | import com.containersolutions.mesos.scheduler.TaskInfoFactoryDocker; 9 | import com.containersolutions.mesos.scheduler.config.FrameworkMesosConfigProperties; 10 | import org.apache.commons.logging.Log; 11 | import org.apache.commons.logging.LogFactory; 12 | import org.apache.mesos.Protos; 13 | import org.springframework.beans.factory.annotation.Qualifier; 14 | import org.springframework.boot.autoconfigure.condition.ConditionalOnBean; 15 | import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; 16 | import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; 17 | import org.springframework.context.annotation.Bean; 18 | import org.springframework.context.annotation.Configuration; 19 | import org.springframework.context.annotation.Import; 20 | import org.springframework.context.annotation.Primary; 21 | 22 | @Configuration 23 | @Import({MesosSchedulerConfiguration.class}) 24 | public class FrameworkSchedulerConfiguration { 25 | protected final Log logger = LogFactory.getLog(getClass()); 26 | 27 | @Bean 28 | @ConditionalOnMissingBean(HealthCheckFactory.class) 29 | public HealthCheckFactory healthCheckFactory() { 30 | return new HealthCheckFactory(); 31 | } 32 | 33 | @Bean 34 | @ConditionalOnMissingBean(FrameworkMesosConfigProperties.class) 35 | public FrameworkMesosConfigProperties frameworkMesosConfigProperties() { 36 | return new FrameworkMesosConfigProperties(); 37 | } 38 | 39 | @Bean(name = "defaultTaskInfoFactoryDocker") 40 | @ConditionalOnProperty(prefix = "mesos.docker", name = {"image"}) 41 | public TaskInfoFactoryDocker defaulTaskInfoFactoryDocker() { 42 | return new TaskInfoFactoryDocker(); 43 | } 44 | 45 | @Bean(name = "defaultTaskInfoFactoryCommand") 46 | public TaskInfoFactoryCommand defaultTaskInfoFactoryCommand() { 47 | return new TaskInfoFactoryCommand(); 48 | } 49 | 50 | // Overridden TaskInfoFactoryBean 51 | @Bean(name = "frameworkTaskInfoFactoryDocker") 52 | @ConditionalOnProperty(prefix = "mesos.docker", name = {"image"}) 53 | @Primary 54 | public TaskInfoFactory taskInfoFactoryDocker(@Qualifier("defaultTaskInfoFactoryDocker") TaskInfoFactoryDocker taskInfoFactoryDocker, HealthCheckFactory healthCheckFactory) { 55 | return (taskId, offer, resources, executionParameters) -> { 56 | Protos.TaskInfo taskInfo = taskInfoFactoryDocker.create(taskId, offer, resources, executionParameters); 57 | Protos.TaskInfo built = Protos.TaskInfo.newBuilder() 58 | .mergeFrom(taskInfo) 59 | .setHealthCheck(healthCheckFactory.create(resources)) 60 | .build(); 61 | logger.debug(built); 62 | return built; 63 | }; 64 | } 65 | 66 | @Bean 67 | @ConditionalOnMissingBean(name = "frameworkTaskInfoFactoryDocker") 68 | @Primary 69 | public TaskInfoFactory taskInfoFactoryCommand(@Qualifier("defaultTaskInfoFactoryCommand") TaskInfoFactoryCommand taskInfoFactoryCommand, HealthCheckFactory healthCheckFactory) { 70 | return (taskId, offer, resources, executionParameters) -> { 71 | Protos.TaskInfo taskInfo = taskInfoFactoryCommand.create(taskId, offer, resources, executionParameters); 72 | Protos.TaskInfo built = Protos.TaskInfo.newBuilder() 73 | .mergeFrom(taskInfo) 74 | .setHealthCheck(healthCheckFactory.create(resources)) 75 | .build(); 76 | logger.debug(built); 77 | return built; 78 | }; 79 | } 80 | 81 | @Bean 82 | @ConditionalOnBean(value = InstanceCount.class) 83 | public CountController countController(InstanceCount instanceCount) { 84 | return new CountController(instanceCount); 85 | } 86 | } 87 | -------------------------------------------------------------------------------- /src/main/java/com/containersolutions/mesos/config/autoconfigure/TaskActuatorConfiguration.java: -------------------------------------------------------------------------------- 1 | package com.containersolutions.mesos.config.autoconfigure; 2 | 3 | import com.containersolutions.mesos.scheduler.InstanceCount; 4 | import com.containersolutions.mesos.scheduler.events.StatusUpdateEvent; 5 | import com.containersolutions.mesos.scheduler.state.StateRepository; 6 | import org.apache.mesos.Protos; 7 | import org.springframework.beans.factory.annotation.Autowired; 8 | import org.springframework.boot.actuate.health.AbstractHealthIndicator; 9 | import org.springframework.boot.actuate.health.Health; 10 | import org.springframework.boot.actuate.health.HealthIndicator; 11 | import org.springframework.context.ApplicationListener; 12 | import org.springframework.context.annotation.Bean; 13 | import org.springframework.context.annotation.Configuration; 14 | 15 | import java.util.List; 16 | import java.util.Map; 17 | import java.util.concurrent.ConcurrentHashMap; 18 | import java.util.stream.Collectors; 19 | 20 | @Configuration 21 | public class TaskActuatorConfiguration implements ApplicationListener { 22 | @Autowired 23 | StateRepository stateRepository; 24 | 25 | @Autowired 26 | InstanceCount instanceCount; 27 | 28 | protected Map taskStatusMap = new ConcurrentHashMap<>(); 29 | 30 | @Bean 31 | public HealthIndicator taskHealthIndicator() { 32 | return new AbstractHealthIndicator() { 33 | @Override 34 | protected void doHealthCheck(Health.Builder builder) throws Exception { 35 | if (correctNumberOfInstances()) { 36 | builder.up(); 37 | } else { 38 | builder.down(); 39 | } 40 | builder.withDetail("mesos.resources.count", instanceCount.getCount()); 41 | builder.withDetail("instances", stateRepository.allTaskInfos().size()); 42 | 43 | for (Protos.TaskState taskState : Protos.TaskState.values()) { 44 | Map state = getTasksForState(taskState); 45 | builder.withDetail(taskState.name(), state.size()); 46 | } 47 | } 48 | }; 49 | } 50 | 51 | @Override 52 | public void onApplicationEvent(StatusUpdateEvent event) { 53 | updateTaskStatusList(event); 54 | List taskIdStrings = getStoredTaskList(); 55 | if (correctNumberOfInstances()) { // If cluster is valid, remove old tasks. 56 | taskStatusMap = taskStatusMap.entrySet().stream() 57 | .filter(entry -> taskIdStrings.contains(entry.getKey())) 58 | .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)); 59 | } 60 | } 61 | 62 | private boolean correctNumberOfInstances() { 63 | return stateRepository.allTaskInfos().size() == instanceCount.getCount(); 64 | } 65 | 66 | private void updateTaskStatusList(StatusUpdateEvent event) { 67 | taskStatusMap.put(getKey(event.getTaskStatus().getTaskId()), event.getTaskStatus()); 68 | } 69 | 70 | private List getStoredTaskList() { 71 | return stateRepository.allTaskInfos().stream().map(taskInfo -> getKey(taskInfo.getTaskId())).collect(Collectors.toList()); 72 | } 73 | 74 | private String getKey(Protos.TaskID taskId) { 75 | return taskId.getValue(); 76 | } 77 | 78 | private Map getTasksForState(Protos.TaskState state) { 79 | return taskStatusMap.entrySet().stream() 80 | .filter(entry -> entry.getValue().getState().equals(state)) 81 | .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)); 82 | } 83 | } 84 | 85 | -------------------------------------------------------------------------------- /src/main/java/com/containersolutions/mesos/controllers/CountController.java: -------------------------------------------------------------------------------- 1 | package com.containersolutions.mesos.controllers; 2 | 3 | import com.containersolutions.mesos.scheduler.InstanceCount; 4 | import org.springframework.beans.factory.annotation.Autowired; 5 | import org.springframework.http.HttpStatus; 6 | import org.springframework.http.MediaType; 7 | import org.springframework.http.ResponseEntity; 8 | import org.springframework.stereotype.Controller; 9 | import org.springframework.web.bind.annotation.RequestBody; 10 | import org.springframework.web.bind.annotation.RequestMapping; 11 | import org.springframework.web.bind.annotation.RequestMethod; 12 | 13 | @Controller 14 | @RequestMapping("/mesos/resources") 15 | public class CountController { 16 | private InstanceCount instanceCount; 17 | 18 | @Autowired 19 | public CountController(InstanceCount instanceCount) { 20 | this.instanceCount = instanceCount; 21 | } 22 | 23 | @RequestMapping(value = "/count", method = RequestMethod.POST, consumes = MediaType.TEXT_PLAIN_VALUE) 24 | public ResponseEntity countEndpointPost(@RequestBody String count) { 25 | instanceCount.setCount(Integer.parseInt(count)); 26 | return ResponseEntity.status(HttpStatus.ACCEPTED).build(); 27 | } 28 | 29 | @RequestMapping(value = "/count", method = RequestMethod.GET, produces = MediaType.TEXT_PLAIN_VALUE) 30 | public ResponseEntity countEndpointGet() { 31 | return new ResponseEntity<>(Integer.toString(instanceCount.getCount()), HttpStatus.OK); 32 | } 33 | } 34 | -------------------------------------------------------------------------------- /src/main/java/com/containersolutions/mesos/scheduler/config/FrameworkMesosConfigProperties.java: -------------------------------------------------------------------------------- 1 | package com.containersolutions.mesos.scheduler.config; 2 | 3 | import org.springframework.boot.context.properties.ConfigurationProperties; 4 | 5 | import java.util.Map; 6 | 7 | @ConfigurationProperties(prefix = "mesos") 8 | public class FrameworkMesosConfigProperties extends MesosConfigProperties { 9 | private Map healthCheck; 10 | 11 | public Map getHealthCheck() { 12 | return healthCheck; 13 | } 14 | 15 | public void setHealthCheck(Map healthCheck) { 16 | this.healthCheck = healthCheck; 17 | } 18 | } 19 | --------------------------------------------------------------------------------