根据输入的参数后缀选择要使用的方法的程序

Question

我们如何实现一个带有参数但基于输入文件后缀（.csv或.ser）后缀的通用程序，它选择用于读取数据的机制。

我在StackOverflow的某处找到了以下代码：但是，我不确定如何安装软件包以及如何设置它。 我读了this，但仍然看不出来。

是否只有Java库才能做到这一点？

import org.apache.commons.io.FilenameUtils;
class FileExtension{
    public static void main(String[] args) {
    String ext1 = FilenameUtils.getExtension("//path//to//your//file");
        if(ext1.equals("csv")){
            //call csv function to read data
        }else if(ext1.equals("ser")){
        }
    }
}

Answer 1

您不需要外部库来获取文件扩展名，像下面这样的简单函数应该可以满足您的需求：

public static String getExtension(String filename) 
{       
    int index = filename.lastIndexOf(".");
    if ( (index == -1) || (index == filename.length() - 1) )
        return "";      
    else            
        return filename.substring(index);   
}