/**
 * 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.camel.scala.dsl

import org.apache.camel.model.LoadBalanceDefinition
import org.apache.camel.scala.dsl.builder.RouteBuilder
import org.apache.camel.Exchange

/**
 * Scala enrichment for Camel's LoadBalanceDefinition
 */
case class SLoadBalanceDefinition(override val target: LoadBalanceDefinition)(implicit val builder: RouteBuilder) extends SAbstractDefinition[LoadBalanceDefinition] {

  def circuitbreaker(threshold: Int, halfOpenAfter: Long, classes: Class[_]*)
  = wrap(target.circuitBreaker(threshold, halfOpenAfter, classes: _*))

  def failover(classes: Class[_]*) = wrap(target.failover(classes: _*))

  def failover = wrap(target.failover)

  def roundrobin = wrap(target.roundRobin)

  def random = wrap(target.random)

  def sticky(expression: Exchange => Any) = wrap(target.sticky(expression))

  def topic = wrap(target.topic)
  
  def weighted(roundRobin: Boolean, distributionRatio: String, distributionRatioDelimiter :String = ",") 
      = wrap(target.weighted(roundRobin, distributionRatio, distributionRatioDelimiter))
  def custom(ref: String) = wrap(target.custom(ref))

  override def wrap(block: => Unit) : SLoadBalanceDefinition = super.wrap(block).asInstanceOf[SLoadBalanceDefinition] 
}
