Examples with FileMeta org.molgenis.data.file.model.FileMeta used on opensource projects

Example 31 with FileMeta

use of org.molgenis.data.file.model.FileMeta in project molgenis by molgenis.

the class RestServiceTest method toEntityFileValueWithoutFileInRequest.

@Test
public void toEntityFileValueWithoutFileInRequest() throws ParseException {
    int entityId = 12345;
    String fileName = "File name";
    String fileAttrName = "fileAttr";
    String oldEntityTypeId = "oldEntityTypeId";
    EntityType oldEntityType = mock(EntityType.class);
    Entity oldEntity = mock(Entity.class);
    FileMeta storedFileMeta = mock(FileMeta.class);
    Attribute fileAttr = mock(Attribute.class);
    Attribute idAttr = mock(Attribute.class);
    when(fileAttr.getName()).thenReturn(fileAttrName);
    when(fileAttr.getEntity()).thenReturn(oldEntityType);
    when(fileAttr.getDataType()).thenReturn(FILE);
    when(oldEntityType.getId()).thenReturn(oldEntityTypeId);
    when(oldEntityType.getIdAttribute()).thenReturn(idAttr);
    when(idAttr.getDataType()).thenReturn(INT);
    when(oldEntity.getEntity(fileAttrName)).thenReturn(storedFileMeta);
    when(storedFileMeta.get(FileMetaMetaData.FILENAME)).thenReturn(fileName);
    when(dataService.findOneById(fileAttr.getEntity().getId(), entityId)).thenReturn(oldEntity);
    Object result = restService.toEntityValue(fileAttr, fileName, entityId);
    assertEquals(result, storedFileMeta);
}

Example 32 with FileMeta

use of org.molgenis.data.file.model.FileMeta in project molgenis by molgenis.

the class RestService method convertFile.

private FileMeta convertFile(Attribute attr, Object paramValue, Object entityId) {
    FileMeta value;
    if (paramValue != null) {
        /*
			 * If an entity is updated and no new file is passed, use the old file value
			 */
        if (!(paramValue instanceof MultipartFile)) {
            EntityType entityType = attr.getEntity();
            Attribute idAttribute = entityType.getIdAttribute();
            Object idValue = this.toEntityValue(idAttribute, entityId, null);
            Entity oldEntity = dataService.findOneById(entityType.getId(), idValue);
            if (paramValue instanceof String) {
                FileMeta entity = (FileMeta) oldEntity.getEntity(attr.getName());
                if (entity.get(FILENAME).equals(paramValue)) {
                    value = entity;
                } else {
                    throw new MolgenisDataException("Cannot update entity with file attribute without passing file," + " while changing the name of the existing file attribute");
                }
            } else {
                throw new MolgenisDataException(format("Attribute [%s] value is of type [%s] instead of [%s]", attr.getName(), paramValue.getClass().getSimpleName(), MultipartFile.class.getSimpleName()));
            }
        } else {
            MultipartFile multipartFile = (MultipartFile) paramValue;
            String id = idGenerator.generateId();
            try {
                fileStore.store(multipartFile.getInputStream(), id);
            } catch (IOException e) {
                throw new MolgenisDataException(e);
            }
            FileMeta fileEntity = fileMetaFactory.create(id);
            fileEntity.setFilename(multipartFile.getOriginalFilename());
            fileEntity.setContentType(multipartFile.getContentType());
            fileEntity.setSize(multipartFile.getSize());
            ServletUriComponentsBuilder currentRequest = servletUriComponentsBuilderFactory.fromCurrentRequest();
            UriComponents downloadUri = currentRequest.replacePath(FileDownloadController.URI + '/' + id).replaceQuery(null).build();
            fileEntity.setUrl(downloadUri.toUriString());
            dataService.add(FILE_META, fileEntity);
            value = fileEntity;
        }
    } else {
        value = null;
    }
    return value;
}