Core Java

State Design Pattern in Java – Example Tutorial

State pattern is one of the behavioral design pattern. State design pattern is used when an Object change it’s behavior based on it’s internal state.

If we have to change the behavior of an object based on it’s state, we can have a state variable in the Object and use if-else condition block to perform different actions based on the state. State pattern is used to provide a systematic and lose-coupled way to achieve this through Context and State implementations.

Context is the class that has a State reference to one of the concrete implementations of the State and forwards the request to the state object for processing. Let’s understand this with a simple example.

Suppose we want to implement a TV Remote with a simple button to perform action, if the State is ON, it will turn on the TV and if state is OFF, it will turn off the TV.

We can implement it using if-else condition like below;

package com.journaldev.design.state;

public class TVRemoteBasic {

	private String state="";

	public void setState(String state){
		this.state=state;
	}

	public void doAction(){
		if(state.equalsIgnoreCase("ON")){
			System.out.println("TV is turned ON");
		}else if(state.equalsIgnoreCase("OFF")){
			System.out.println("TV is turned OFF");
		}
	}

	public static void main(String args[]){
		TVRemoteBasic remote = new TVRemoteBasic();

		remote.setState("ON");
		remote.doAction();

		remote.setState("OFF");
		remote.doAction();
	}

}

Notice that client code should know the specific values to use for setting the state of remote, further more if number of states increase then the tight coupling between implementation and the client code will be very hard to maintain and extend.

Now we will use State pattern to implement above TV Remote example.

State Interface

First of all we will create State interface that will define the method that should be implemented by different concrete states and context class.

package com.journaldev.design.state;

public interface State {

	public void doAction();
}

Concrete State Implementations

In our example, we can have two states – one for turning TV on and another to turn it off. So we will create two concrete state implementations for these behaviors.

package com.journaldev.design.state;

public class TVStartState implements State {

	@Override
	public void doAction() {
		System.out.println("TV is turned ON");
	}

}
package com.journaldev.design.state;

public class TVStopState implements State {

	@Override
	public void doAction() {
		System.out.println("TV is turned OFF");
	}

}

Now we are ready to implement our Context object that will change it’s behavior based on it’s internal state.

Context Implementation

package com.journaldev.design.state;

public class TVContext implements State {

	private State tvState;

	public void setState(State state) {
		this.tvState=state;
	}

	public State getState() {
		return this.tvState;
	}

	@Override
	public void doAction() {
		this.tvState.doAction();
	}

}

Notice that Context also implements State and keep a reference of it’s current state and forwards the request to the state implementation.

Test Program

Now let’s write a simple program to test our implementation of TV Remote using State pattern.

package com.journaldev.design.state;

public class TVRemote {

	public static void main(String[] args) {
		TVContext context = new TVContext();
		State tvStartState = new TVStartState();
		State tvStopState = new TVStopState();

		context.setState(tvStartState);
		context.doAction();

		context.setState(tvStopState);
		context.doAction();

	}

}

Output of above program is same as the basic implementation of TV Remote without using any pattern.

The benefits of using State pattern to implement polymorphic behavior is clearly visible, the chances of error are less and it’s very easy to add more states for additional behavior making it more robust, easily maintainable and flexible. Also State pattern helped in avoiding if-else or switch-case conditional logic in this scenario.

State Pattern is very similar to Strategy Pattern, check out Strategy Pattern in Java.

Thats all for State pattern in java, I hope you liked it.
 

Subscribe
Notify of
guest

This site uses Akismet to reduce spam. Learn how your comment data is processed.

7 Comments
Oldest
Newest Most Voted
Inline Feedbacks
View all comments
Alex
Alex
10 years ago

Nice, Thank You!!! :-)

Bhukailas
Bhukailas
10 years ago

Hi Pankaj, I read all you design pattern articles, Based on my understnding, I feel Context and State are supposed to be two separate things but I see TVContext and TvStates are implementing same interface State.

wouldn’t there be any other interface ?

Bhukailas
Bhukailas
10 years ago

one more thing is If we have some 100 states we should write 100 classes, In this case also we need to something, what do you say ?

Jb
Jb
10 years ago

Thanks for the post, but I don’t think this is quite correct. The context can not be a state as it is itself changing. “Object change it’s behavior based on it’s internal state.”

Krishna
9 years ago

HI,

TvContext can not implement state interface. I don’t think that is correct way of state pattern implementation.

Rahil Khan
Rahil Khan
8 years ago

Where should we keep reference of actual class (whose behavior changes as per state), in the states or in the context class so that we can perform state specific tasks???

Cory Huff
Cory Huff
7 years ago

Hi, I never seen anyone do a game manager this way so I went and tried to implement it into a quick program I wrote up. It took some modifications to it and when I got it working the way I wanted it to it didn’t work to bad.

Back to top button