파일 확장자, 파일 분류 가져오기 소스

파일 확장자 (예: image.png 의 .png를) 등을 가져오거나 파일의 타입이 이미지인지, 일반 파일인지 가져오고 싶을때 위 소스를 사용.

public class Extensions {
    
    private static final String[] IMG_TYPES = {"jpg","png","gif"};
    private static final String[] FILE_TYPES = {"xlsx","xls","pptx","ppt","pdf"};
    
    /**
     * 확장자 가져오기
     * @param String
     * @throws Exception
     */
    public static String getExtType(String extension) throws Exception {
        
        String ext = extension;
        
        int pos = ext.lastIndexOf( "." );
        ext = ext.substring( pos + 1 );
        
        return ext;
    }
    
    /**
     * 확장자가 이미지 타입인지 확인
     * @param String
     * @throws Exception
     */
    public static boolean isImage(String extension) throws Exception {
        
        String ext = getExtType(extension);
        
        int extensionLength = IMG_TYPES.length;
        
        for(int i = 0; i < extensionLength; i++) {
            if(ext.equalsIgnoreCase(IMG_TYPES[i])) {
                return true;
            }
        }
        
        return false;
    }
    
    /**
     * 확장자가 파일 타입인지 확인
     * @param String
     * @throws Exception
     */
    public static boolean isFile(String extension) throws Exception {
        
        String ext = getExtType(extension);
        
        int extensionLength = FILE_TYPES.length;
        
        for(int i = 0; i < extensionLength; i++) {
            if(ext.equalsIgnoreCase(FILE_TYPES[i])) {
                return true;
            }
        }
        
        return false;
    }
    
}

3.번 , 4.번 줄에 필요한 확장자를 넣어 사용한다.