文件仅使用枚举上传指定的图像内容类型

时间:2012-11-14 20:39:26

标签: grails file-upload content-type

我正在使用Grails的HDImageService插件来完成扩展用户上传图像的繁重工作。我创建了ImageService.groovy以保存到我的Amazon S3存储桶。一切正常,用户选择文件,点击发布和wahla图像缩放,存储和显示。我的问题是我不知道如何限制用户上传图像以外的文件。我想只允许上传jpeg,jpg,gif或png类型的文件。我用这些变量创建了一个ENUM类,但我不知道在哪里或如何实现。任何人都可以指出我正确的方向


RegistrationController.groovy:获取文件并保存到存储桶

         if ( params.photo ) {
            MultipartFile file = request.getFile( 'photo' )
            byte[] fileBytes = file.bytes
            ByteArrayInputStream bais = new ByteArrayInputStream( fileBytes )
            BufferedImage image = ImageIO.read( bais )
            def width = image.width
            def height = image.height
            def maxWidth = 500

            // Calculate the ratio that we will need to resize the image
            double ratio = 1.0f
            if ( width > maxWidth ) {
                def r = maxWidth / width
                ratio = r < 1.0f ? r : 1.0f
                bais.reset()
                fileBytes = hdImageService.scale( bais, maxWidth, Math.round( height * ratio ) as int )
            }

            geolink.custPhoto = imageService.storeS3Image(
                imageService.buildPPJPhotoPath( geolink, file.getOriginalFilename() ),
                fileBytes,
                file.contentType
            )
        }

ImageService.groovy:Enum

    String getFormatName( byte[] raw ) {
    try {
        // Create an image input stream on the image
        ImageInputStream iis = ImageIO.createImageInputStream( new ByteArrayInputStream( raw ) )

        // Find all image readers that recognize the image format
        Iterator iter = ImageIO.getImageReaders(iis)
        if (!iter.hasNext()) {
            // No readers found
            log.debug( "Unable to get format" )
            return null;
        }

        // Use the first reader
        ImageReader reader = (ImageReader)iter.next()

        // Close stream
        iis.close()

        // Return the format name
        log.debug( "Format: ${reader.getFormatName() }" )
        return reader.getFormatName()
    }
    catch (IOException e) {
        log.warn( "Unable to determine image format", e )
    }
    // The image could not be read
    return null;
}

ImageFormat getContentType( String filename ) {
    String[] parts = filename.split( '\\.' )
    return ImageFormat.valueOf( parts[parts.length - 1].toUpperCase() )
}}

    public enum ImageFormat {
JPEG( 'image/jpeg', 'jpg' ),
JPG( 'image/jpeg', 'jpg' ),
PNG( 'image/png', 'png' ),
GIF( 'image/gif', 'gif' )

String mimeType
String extension

public ImageFormat( String mime, String ext ) {
    this.mimeType = mime
    this.extension = ext
}

}

1 个答案:

答案 0 :(得分:6)

我是这样做的。将其修改为Enum应该很容易:

def downloadedFile = request.getFile('imageFile')
def okContentTypes = ['image/png', 'image/jpeg', 'image/jpg', 'image/gif']

if (!okContentTypes.contains(downloadedFile.getContentType())) {
    myDomainInstance.errors.rejectValue("image", "Image type must be one of: ${okContentTypes}")
}