Question

public interface FileReader {

  void readFile();
   void writeFile();
}


 public class XMLReader implements FileReader {

  @Override
   public void readFile() {
    System.out.println("Hellp i am read");

   }

   @Override
   public void writeFile() {
        System.out.println("Hello i am write");
    }


 }

  public class ExcelReader implements FileReader {

  @Override
   public void readFile() {
    System.out.println("Hellp i am read");

   }

   @Override
   public void writeFile() {
        System.out.println("Hello i am write");
    }

}

public class Context {

   FileReader reader ;

   public Context(FileReader reader) {
    super();
    this.reader = reader;
   }

   public void executeRead(){

        reader.readFile();
    }


       }

public class TestStrategy {

public static void main(String[] args) {
    // TODO Auto-generated method stub

    System.out.println(args[0]);
    String s=args[0];
    String[] a=s.split("\\.");
    if(a[1].equals("csv")){
        new Context(new XMLReader()).executeRead();
    }else{
        new Context(new ExcelReader()).executeRead();
    }
}

}

我有一个问题是介绍了更多的文件格式，我们将为它们创建单独的类，但我必须更改TestStrategy类中的if else代码，以便为引入的文件模式创建新对象。

我们可以删除这个if else代码。任何建议。

Answer 1

您可以按名称获得课程。构建Map以配置FileReader以用于每个扩展。

public static void main(String[] args) 
  throws ClassNotFoundException, InstantiationException, IllegalAccessException {
    Map<String, String> extensionToClass = new HashMap<>();
    extensionToClass.put("xml", "de.lhorn.XMLReader");
    extensionToClass.put("xls", "de.lhorn.ExcelReader");

    String s = "foo.xml";
    String[] a = s.split("\\.");
    String extension = a[1];

    // Get the class that is configured for the extension.
    String className = extensionToClass.get(extension);
    Class clazz = Class.forName(className);

    // Create a new instance of this class.
    FileReader reader = (FileReader) clazz.newInstance();

    // Use the FileReader.
    new Context(reader).executeRead();
}

当然，您可以从外部来源阅读extensionToClass。

Answer 2

您可以使用将文件扩展名映射到实现的注册表。

public class Registry {
    static Map<String,FileReader> reg = new HashMap<String,FileReader>();

    public static void register(String ext, FileReader fr ) {
        reg.put( ext, fr );
    }
}

让新添加的实现自己注册，例如

public class XMLReader implements FileReader {
        static {
            Registry.register( "xml", new XMLReader() );
        }
....
public class ExcelReader implements FileReader {
        static {
            Registry.register( "xls", new ExcelReader() );
        }
...

然后你可以简单地在注册表中查找合适的实现，而不需要，如果需要或开关。

用于读取不同文件格式的策略模式

2 个答案: