Scala Tutorial - Learn Function Composition Using AndThen

By Nadim Bahadoor | Last updated: October 24, 2019 at 17:25 pm

Overview

In this tutorial, we will learn how to create value functions which are defined using the val keyword as opposed to using the def keyword.

 

Val functions inherit the andThen function and we will show how to use the andThen function to compose two functions together.

 

Mathematically speaking, (f andThen g)(x) = g(f(x)). The results of the first function f(x) is ran first and will be passed as input to the second function g(x).

Steps

1. Assume a pre-calculated total cost amount

Let's start with a simple totalCost value which represents the total cost in dollar figure for a particular customer buying donuts from your store.


println("Step 1: Assume a pre-calculated total cost amount")
val totalCost: Double = 10

 

2. How to define a val function to apply discount to total cost

Similar to the example from the previous tutorial on defining function with val keyword, let's define a val function which will apply some discount dollar value from the total cost figure.


println("\nStep 2: How to define a val function to apply discount to total cost")
val applyDiscountValFunction = (amount: Double) => {
  println("Apply discount function")
  val discount = 2 // fetch discount from database
  amount - discount
}

 

3. How to call a val function
Calling the val function applyDiscountValFunction from Step 2 is very straight-forward. You simply need to pass it the totalCost value from Step 1.


println("\nStep 3: How to call a val function")
println(s"Total cost of 5 donuts with discount = ${applyDiscountValFunction(totalCost)}")

You should see the following output when you run your Scala application in IntelliJ:


Step 3: How to call a val function
Apply discount function
Total cost of 5 donuts with discount = 8.0

 

4. How to define a val function to apply tax to total cost

Let's go ahead and define another val function which should apply some tax amount to the totalCost value.


println("\nStep 4: How to define a val function to apply tax to total cost")
val applyTaxValFunction = (amount: Double) => {
  println("Apply tax function")
  val tax = 1 // fetch tax from database
  amount + tax
}

 

5. How to call andThen on a val function

As we've seen from the previous tutorial on defining function with val keyword, val function inherits an andThen function.

 

Calling andThen will take the result from the first function and pass it as input parameter to the second function. Let's use the andThen semantics to apply discount andThen apply tax to the totalCost figure as shown below.


println("\nStep 5: How to call andThen on a val function")
println(s"Total cost of 5 donuts = ${ (applyDiscountValFunction andThen applyTaxValFunction)(totalCost) }")

You should see the following output when you run your Scala application in IntelliJ:


Step 5: How to call andThen on a val function
Apply discount function
Apply tax function
Total cost of 5 donuts = 9.0

NOTE:

  • The apply discount function was called first andThen the apply tax function was called.
  • The output from the first apply discount function was also passed through as input parameter to the second apply tax function

This concludes our tutorial on Scala Tutorial - Learn Function Composition Using AndThen and I hope you've found it useful!

 

Stay in touch via Facebook and Twitter for upcoming tutorials!

 

Don't forget to like and share this page :)

Summary

In this tutorial, we went over the following:

  • How to define a value of type Double to represent total cost
  • How to define a val function to apply discount to total cost
  • How to call a val function
  • How to define a value function to apply tax to total cost
  • How to call andThen on a val function

Tip

  • val functions are instances of FunctionN classes and if you look at the Scala documentation for say Function1 class, you will notice that val function will inherit other methods such as andThen or compose which allow for function composition.

Source Code

The source code is available on the allaboutscala GitHub repository.

 

What's Next

In the next tutorial, I will show you how to use the compose function inherited when you define val function.

Nadim Bahadoor on FacebookNadim Bahadoor on GithubNadim Bahadoor on LinkedinNadim Bahadoor on Twitter
Nadim Bahadoor
Technology and Finance Consultant with over 14 years of hands-on experience building large scale systems in the Financial (Electronic Trading Platforms), Risk, Insurance and Life Science sectors. I am self-driven and passionate about Finance, Distributed Systems, Functional Programming, Big Data, Semantic Data (Graph) and Machine Learning.
Other allaboutscala.com tutorials you may like: