最近在写一个数据生成工具，代码中需要判断各种数据类型，有一个枚举中定义了很多类型，根据这些不同类型有不同的操作，于是我写了一大排的if，把自己都给搞晕了，代码大致如下：
数据类型枚举定义：
```java
public enum GenerateDataType {

/**
     * 自增，对应java的Long
     */
    AutoIncre(),
    /**
     * 字段串类型
     */
    String(),
    /**
     * 数字
     */
    Number(),
    /**
     * 生成uuid，带横线-
     */
    UUID(),
    /**
     * 生成uuid，并去除-
     */
    uuid(),
    /**
     * 日期，如果没有配置format属性，默认生成yyyy-MM-dd HH:mm:ss格式的日期
     */
    Date(),
    /**
     * json对象
     */
    Object();

private GenerateDataType(){
    }
}
```
每个数据类型对应的操作判断的代码：
```java
        if(dataType.equals(GenerateDataType.AutoIncre.name())){
            //操作
        }
        if(dataType.equals(GenerateDataType.String.name())){
            //操作
        }
        if(dataType.equals(GenerateDataType.Number.name())){
            //操作
        }
        if(dataType.equals(GenerateDataType.Date.name())){
            //操作
        }
        ...
```
上面这样各种判断，可苦了我，于是想整改重构一下这些代码，将所有类型的操作都实现一个接口，采用工厂模式根据不同类型调用不同接口实现，这样就去掉了if判断，让代码更优雅，具体代码如下：
定义各种数据类型公用的操作接口：
```java
public interface ValueHandle {

public String handleValue(String column,GenerateDataContext context);

}

```
定义各个数据类型的操作实现类
自增长类型
```java
public class AutoIncreValueHandle implements ValueHandle {

@Override
    public String handleValue(String column, GenerateDataContext context) {
        return null;
    }
}
```
数字类型
```java
public class NumberValueHandle implements ValueHandle {

@Override
    public String handleValue(String column, GenerateDataContext context) {
		return null
	｝
}

```
。。。
再定义工厂
```java
public class DataTypeValueFactory {

static Map<String, ValueHandle> operationMap = new HashMap<>();

static {
        operationMap.put(GenerateDataType.AutoIncre.name(), new AutoIncreValueHandle());
        operationMap.put(GenerateDataType.String.name(), new StringValueHandle());
        operationMap.put(GenerateDataType.Number.name(), new NumberValueHandle());
        operationMap.put(GenerateDataType.UUID.name(), new UUIDsValueHandle());
        operationMap.put(GenerateDataType.uuid.name(), new UuidValueHandle());
        operationMap.put(GenerateDataType.Date.name(), new DateValueHandle());
        operationMap.put(GenerateDataType.Object.name(), new DateValueHandle());
        // more operators
    }

public static ValueHandle getDataTypeValueHandle(String dataType) throws Exception {
        ValueHandle dataTypeValueHandle = operationMap.get(dataType);
        if (dataTypeValueHandle == null) {
            throw new Exception("dataType:"+dataType +" is unsupported data type!");
        }
        return dataTypeValueHandle;
    }
}
```
从工厂中取的对应类型的实现来调用不同的类型操作
```java
ValueHandle valueHandle = DataTypeValueFactory.getDataTypeValueHandle("Number");
Object columnData = valueHandle.handleValue("age",context);
```
这样Map替代了if判断，工厂模式与接口的设计拆开解藕各种数据类型的操作，就不需要写if判断了，整个代码优雅不少，修改一个类型的操作不会影响其他类型，减少出错

ClassInstance

java如何减少if判断让代码优雅一点：采用Map+工厂模式实现

相关文章

推荐文章

热门文章

标签列表