diff --git a/src/brainatwork/pattern/creational/factorypattern/ShapeFactory.java b/src/brainatwork/pattern/creational/factorypattern/ShapeFactory.java index 5cbb1e2..87872cb 100644 --- a/src/brainatwork/pattern/creational/factorypattern/ShapeFactory.java +++ b/src/brainatwork/pattern/creational/factorypattern/ShapeFactory.java @@ -4,16 +4,16 @@ public class ShapeFactory { //use getShape method to get object of type shape public Shape getShape(String shapeType){ - if(shapeType == null){ - return null; - } - if(shapeType.equalsIgnoreCase("CIRCLE")){ - return new Circle(); - } - else if(shapeType.equalsIgnoreCase("SQUARE")){ - return new Square(); - } - - return null; + if ("CIRCLE".equalsIgnoreCase(shapeType)) + { + return new Circle(); + } + + if ("SQUARE".equalsIgnoreCase(shapeType)) + { + return new Square(); + } + + return null; } }