2015-11-01 16:35:01 +01:00
package provider
2015-09-12 15:10:03 +02:00
2015-09-09 22:39:08 +02:00
import (
2015-11-01 16:35:01 +01:00
"errors"
2015-11-13 11:50:32 +01:00
"net/url"
2015-09-24 17:16:13 +02:00
"strconv"
"text/template"
2015-09-10 22:54:37 +02:00
"github.com/BurntSushi/ty/fun"
2015-09-24 14:32:37 +02:00
log "github.com/Sirupsen/logrus"
2015-11-01 16:35:01 +01:00
"github.com/emilevauge/traefik/types"
2015-09-12 15:10:03 +02:00
"github.com/gambol99/go-marathon"
2015-09-09 22:39:08 +02:00
)
2015-11-01 19:29:47 +01:00
// Marathon holds configuration of the Marathon provider.
2015-11-02 19:48:34 +01:00
type Marathon struct {
2015-11-13 11:50:32 +01:00
baseProvider
2015-09-10 15:13:35 +02:00
Endpoint string
Domain string
NetworkInterface string
2016-01-18 11:52:18 +01:00
Basic * MarathonBasic
2015-11-13 11:50:32 +01:00
marathonClient lightMarathonClient
}
2016-01-18 11:52:18 +01:00
// MarathonBasic holds basic authentication specific configurations
type MarathonBasic struct {
HTTPBasicAuthUser string
HTTPBasicPassword string
}
2015-11-13 11:50:32 +01:00
type lightMarathonClient interface {
Applications ( url . Values ) ( * marathon . Applications , error )
2015-11-22 23:32:31 +01:00
AllTasks ( v url . Values ) ( * marathon . Tasks , error )
2015-09-09 22:39:08 +02:00
}
2015-11-01 19:29:47 +01:00
// Provide allows the provider to provide configurations to traefik
// using the given configuration channel.
2015-11-02 19:48:34 +01:00
func ( provider * Marathon ) Provide ( configurationChan chan <- types . ConfigMessage ) error {
2015-09-09 22:39:08 +02:00
config := marathon . NewDefaultConfig ( )
config . URL = provider . Endpoint
2015-09-10 15:13:35 +02:00
config . EventsInterface = provider . NetworkInterface
2016-01-18 11:52:18 +01:00
if provider . Basic != nil {
config . HTTPBasicAuthUser = provider . Basic . HTTPBasicAuthUser
config . HTTPBasicPassword = provider . Basic . HTTPBasicPassword
}
2015-09-24 17:16:13 +02:00
client , err := marathon . NewClient ( config )
if err != nil {
2015-09-24 14:32:37 +02:00
log . Errorf ( "Failed to create a client for marathon, error: %s" , err )
2015-10-01 12:04:25 +02:00
return err
2015-09-24 17:16:13 +02:00
}
provider . marathonClient = client
update := make ( marathon . EventsChannel , 5 )
if provider . Watch {
if err := client . AddEventsListener ( update , marathon . EVENTS_APPLICATIONS ) ; err != nil {
log . Errorf ( "Failed to register for subscriptions, %s" , err )
} else {
go func ( ) {
for {
event := <- update
log . Debug ( "Marathon event receveived" , event )
configuration := provider . loadMarathonConfig ( )
if configuration != nil {
2015-11-13 11:50:32 +01:00
configurationChan <- types . ConfigMessage {
ProviderName : "marathon" ,
Configuration : configuration ,
}
2015-09-09 23:09:16 +02:00
}
2015-09-24 17:16:13 +02:00
}
} ( )
2015-09-09 23:09:16 +02:00
}
2015-09-09 22:39:08 +02:00
}
2015-09-24 17:16:13 +02:00
configuration := provider . loadMarathonConfig ( )
2015-11-13 11:50:32 +01:00
configurationChan <- types . ConfigMessage {
ProviderName : "marathon" ,
Configuration : configuration ,
}
2015-10-01 12:04:25 +02:00
return nil
2015-09-09 22:39:08 +02:00
}
2015-11-02 19:48:34 +01:00
func ( provider * Marathon ) loadMarathonConfig ( ) * types . Configuration {
2015-10-08 21:21:51 +02:00
var MarathonFuncMap = template . FuncMap {
2015-11-13 11:50:32 +01:00
"getPort" : provider . getPort ,
"getWeight" : provider . getWeight ,
"getDomain" : provider . getDomain ,
"getProtocol" : provider . getProtocol ,
"getPassHostHeader" : provider . getPassHostHeader ,
"getFrontendValue" : provider . getFrontendValue ,
"getFrontendRule" : provider . getFrontendRule ,
"replace" : replace ,
2015-10-08 21:21:51 +02:00
}
2015-09-09 22:39:08 +02:00
applications , err := provider . marathonClient . Applications ( nil )
2015-09-12 15:10:03 +02:00
if err != nil {
2015-09-24 14:32:37 +02:00
log . Errorf ( "Failed to create a client for marathon, error: %s" , err )
2015-09-09 22:39:08 +02:00
return nil
}
2015-09-10 22:54:37 +02:00
2015-11-22 23:32:31 +01:00
tasks , err := provider . marathonClient . AllTasks ( ( url . Values { "status" : [ ] string { "running" } } ) )
2015-09-12 15:10:03 +02:00
if err != nil {
2015-09-24 14:32:37 +02:00
log . Errorf ( "Failed to create a client for marathon, error: %s" , err )
2015-09-09 22:39:08 +02:00
return nil
}
2015-09-10 22:54:37 +02:00
//filter tasks
filteredTasks := fun . Filter ( func ( task marathon . Task ) bool {
2015-11-13 11:50:32 +01:00
return taskFilter ( task , applications )
2015-09-10 22:54:37 +02:00
} , tasks . Tasks ) . ( [ ] marathon . Task )
//filter apps
filteredApps := fun . Filter ( func ( app marathon . Application ) bool {
2015-11-13 11:50:32 +01:00
return applicationFilter ( app , filteredTasks )
2015-09-10 22:54:37 +02:00
} , applications . Apps ) . ( [ ] marathon . Application )
2015-09-09 22:39:08 +02:00
templateObjects := struct {
Applications [ ] marathon . Application
Tasks [ ] marathon . Task
Domain string
} {
2015-09-10 22:54:37 +02:00
filteredApps ,
filteredTasks ,
2015-09-09 22:39:08 +02:00
provider . Domain ,
}
2015-11-13 11:50:32 +01:00
configuration , err := provider . getConfiguration ( "templates/marathon.tmpl" , MarathonFuncMap , templateObjects )
2015-09-09 22:39:08 +02:00
if err != nil {
2015-11-13 11:50:32 +01:00
log . Error ( err )
2015-09-09 22:39:08 +02:00
}
2015-11-13 11:50:32 +01:00
return configuration
}
2015-09-09 22:39:08 +02:00
2015-11-13 11:50:32 +01:00
func taskFilter ( task marathon . Task , applications * marathon . Applications ) bool {
if len ( task . Ports ) == 0 {
log . Debug ( "Filtering marathon task without port %s" , task . AppID )
return false
}
2015-12-05 10:59:01 -08:00
application , err := getApplication ( task , applications . Apps )
if err != nil {
2015-11-13 11:50:32 +01:00
log . Errorf ( "Unable to get marathon application from task %s" , task . AppID )
return false
}
if application . Labels [ "traefik.enable" ] == "false" {
log . Debugf ( "Filtering disabled marathon task %s" , task . AppID )
return false
}
2015-12-05 10:59:01 -08:00
//filter indeterminable task port
portIndexLabel := application . Labels [ "traefik.portIndex" ]
portValueLabel := application . Labels [ "traefik.port" ]
if portIndexLabel != "" && portValueLabel != "" {
log . Debugf ( "Filtering marathon task %s specifying both traefik.portIndex and traefik.port labels" , task . AppID )
return false
}
if portIndexLabel == "" && portValueLabel == "" && len ( application . Ports ) > 1 {
log . Debugf ( "Filtering marathon task %s with more than 1 port and no traefik.portIndex or traefik.port label" , task . AppID )
return false
}
if portIndexLabel != "" {
index , err := strconv . Atoi ( application . Labels [ "traefik.portIndex" ] )
if err != nil || index < 0 || index > len ( application . Ports ) - 1 {
log . Debugf ( "Filtering marathon task %s with unexpected value for traefik.portIndex label" , task . AppID )
return false
}
}
if portValueLabel != "" {
port , err := strconv . Atoi ( application . Labels [ "traefik.port" ] )
if err != nil {
log . Debugf ( "Filtering marathon task %s with unexpected value for traefik.port label" , task . AppID )
return false
}
var foundPort bool
for _ , exposedPort := range task . Ports {
if port == exposedPort {
foundPort = true
break
}
}
if ! foundPort {
log . Debugf ( "Filtering marathon task %s without a matching port for traefik.port label" , task . AppID )
return false
}
}
2015-11-13 11:50:32 +01:00
//filter healthchecks
if application . HasHealthChecks ( ) {
if task . HasHealthCheckResults ( ) {
for _ , healthcheck := range task . HealthCheckResult {
// found one bad healthcheck, return false
if ! healthcheck . Alive {
log . Debugf ( "Filtering marathon task %s with bad healthcheck" , task . AppID )
return false
}
}
} else {
log . Debugf ( "Filtering marathon task %s with bad healthcheck" , task . AppID )
return false
}
2015-09-09 22:39:08 +02:00
}
2015-11-13 11:50:32 +01:00
return true
}
2015-09-09 22:39:08 +02:00
2015-11-13 11:50:32 +01:00
func applicationFilter ( app marathon . Application , filteredTasks [ ] marathon . Task ) bool {
return fun . Exists ( func ( task marathon . Task ) bool {
return task . AppID == app . ID
} , filteredTasks )
2015-09-10 22:54:37 +02:00
}
2015-10-27 00:26:35 +01:00
func getApplication ( task marathon . Task , apps [ ] marathon . Application ) ( marathon . Application , error ) {
2015-09-10 22:54:37 +02:00
for _ , application := range apps {
2015-09-12 15:10:03 +02:00
if application . ID == task . AppID {
2015-10-27 00:26:35 +01:00
return application , nil
2015-09-10 22:54:37 +02:00
}
}
2015-10-27 00:26:35 +01:00
return marathon . Application { } , errors . New ( "Application not found: " + task . AppID )
2015-09-12 15:10:03 +02:00
}
2015-10-23 09:49:19 +02:00
2015-11-02 19:48:34 +01:00
func ( provider * Marathon ) getLabel ( application marathon . Application , label string ) ( string , error ) {
2015-10-23 09:49:19 +02:00
for key , value := range application . Labels {
if key == label {
return value , nil
}
}
return "" , errors . New ( "Label not found:" + label )
}
2015-12-05 10:59:01 -08:00
func ( provider * Marathon ) getPort ( task marathon . Task , applications [ ] marathon . Application ) string {
application , err := getApplication ( task , applications )
if err != nil {
log . Errorf ( "Unable to get marathon application from task %s" , task . AppID )
return ""
}
if portIndexLabel , err := provider . getLabel ( application , "traefik.portIndex" ) ; err == nil {
if index , err := strconv . Atoi ( portIndexLabel ) ; err == nil {
return strconv . Itoa ( task . Ports [ index ] )
}
}
if portValueLabel , err := provider . getLabel ( application , "traefik.port" ) ; err == nil {
return portValueLabel
}
2015-11-13 11:50:32 +01:00
for _ , port := range task . Ports {
return strconv . Itoa ( port )
}
return ""
}
func ( provider * Marathon ) getWeight ( task marathon . Task , applications [ ] marathon . Application ) string {
application , errApp := getApplication ( task , applications )
if errApp != nil {
log . Errorf ( "Unable to get marathon application from task %s" , task . AppID )
return "0"
}
if label , err := provider . getLabel ( application , "traefik.weight" ) ; err == nil {
return label
}
return "0"
}
func ( provider * Marathon ) getDomain ( application marathon . Application ) string {
if label , err := provider . getLabel ( application , "traefik.domain" ) ; err == nil {
return label
}
return provider . Domain
}
func ( provider * Marathon ) getProtocol ( task marathon . Task , applications [ ] marathon . Application ) string {
application , errApp := getApplication ( task , applications )
if errApp != nil {
log . Errorf ( "Unable to get marathon application from task %s" , task . AppID )
return "http"
}
if label , err := provider . getLabel ( application , "traefik.protocol" ) ; err == nil {
return label
}
return "http"
}
func ( provider * Marathon ) getPassHostHeader ( application marathon . Application ) string {
if passHostHeader , err := provider . getLabel ( application , "traefik.frontend.passHostHeader" ) ; err == nil {
return passHostHeader
}
return "false"
2015-10-23 09:49:19 +02:00
}
2015-11-13 11:50:32 +01:00
// getFrontendValue returns the frontend value for the specified application, using
2015-11-01 19:29:47 +01:00
// it's label. It returns a default one if the label is not present.
2015-11-13 11:50:32 +01:00
func ( provider * Marathon ) getFrontendValue ( application marathon . Application ) string {
2015-10-23 09:49:19 +02:00
if label , err := provider . getLabel ( application , "traefik.frontend.value" ) ; err == nil {
return label
}
2015-11-13 11:50:32 +01:00
return getEscapedName ( application . ID ) + "." + provider . Domain
2015-10-23 09:49:19 +02:00
}
2015-11-13 11:50:32 +01:00
// getFrontendRule returns the frontend rule for the specified application, using
2015-11-01 19:29:47 +01:00
// it's label. It returns a default one (Host) if the label is not present.
2015-11-13 11:50:32 +01:00
func ( provider * Marathon ) getFrontendRule ( application marathon . Application ) string {
2015-10-23 09:49:19 +02:00
if label , err := provider . getLabel ( application , "traefik.frontend.rule" ) ; err == nil {
return label
}
return "Host"
}