116 lines
5.5 KiB
Java
116 lines
5.5 KiB
Java
package eu.eudat.model.builder;
|
|
|
|
import eu.eudat.authorization.AuthorizationFlags;
|
|
import eu.eudat.commons.scope.tenant.TenantScope;
|
|
import eu.eudat.convention.ConventionService;
|
|
import eu.eudat.data.StorageFileEntity;
|
|
import eu.eudat.model.*;
|
|
import eu.eudat.query.*;
|
|
import gr.cite.tools.data.builder.BuilderFactory;
|
|
import gr.cite.tools.data.query.QueryFactory;
|
|
import gr.cite.tools.exception.MyApplicationException;
|
|
import gr.cite.tools.fieldset.BaseFieldSet;
|
|
import gr.cite.tools.fieldset.FieldSet;
|
|
import gr.cite.tools.logging.DataLogEntry;
|
|
import gr.cite.tools.logging.LoggerService;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.beans.factory.config.ConfigurableBeanFactory;
|
|
import org.springframework.context.annotation.Scope;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
import java.util.*;
|
|
import java.util.stream.Collectors;
|
|
|
|
@Component
|
|
@Scope(value = ConfigurableBeanFactory.SCOPE_PROTOTYPE)
|
|
public class StorageFileBuilder extends BaseBuilder<StorageFile, StorageFileEntity> {
|
|
|
|
private final QueryFactory queryFactory;
|
|
|
|
private final BuilderFactory builderFactory;
|
|
private final TenantScope tenantScope;
|
|
|
|
private EnumSet<AuthorizationFlags> authorize = EnumSet.of(AuthorizationFlags.None);
|
|
|
|
@Autowired
|
|
public StorageFileBuilder(
|
|
ConventionService conventionService,
|
|
QueryFactory queryFactory,
|
|
BuilderFactory builderFactory, TenantScope tenantScope) {
|
|
super(conventionService, new LoggerService(LoggerFactory.getLogger(StorageFileBuilder.class)));
|
|
this.queryFactory = queryFactory;
|
|
this.builderFactory = builderFactory;
|
|
this.tenantScope = tenantScope;
|
|
}
|
|
|
|
public StorageFileBuilder authorize(EnumSet<AuthorizationFlags> values) {
|
|
this.authorize = values;
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public List<StorageFile> build(FieldSet fields, List<StorageFileEntity> data) throws MyApplicationException {
|
|
this.logger.debug("building for {} items requesting {} fields", Optional.ofNullable(data).map(List::size).orElse(0), Optional.ofNullable(fields).map(FieldSet::getFields).map(Set::size).orElse(0));
|
|
this.logger.trace(new DataLogEntry("requested fields", fields));
|
|
if (fields == null || data == null || fields.isEmpty())
|
|
return new ArrayList<>();
|
|
|
|
FieldSet userFields = fields.extractPrefixed(this.asPrefix(StorageFile._owner));
|
|
Map<UUID, User> userItemsMap = this.collectUsers(userFields, data);
|
|
|
|
List<StorageFile> models = new ArrayList<>();
|
|
for (StorageFileEntity d : data) {
|
|
StorageFile m = new StorageFile();
|
|
if (fields.hasField(this.asIndexer(StorageFile._id))) m.setId(d.getId());
|
|
if (fields.hasField(this.asIndexer(StorageFile._name))) m.setName(d.getName());
|
|
if (fields.hasField(this.asIndexer(StorageFile._fileRef))) m.setFileRef(d.getFileRef());
|
|
if (fields.hasField(this.asIndexer(StorageFile._extension))) m.setExtension(d.getExtension());
|
|
if (fields.hasField(this.asIndexer(StorageFile._createdAt))) m.setCreatedAt(d.getCreatedAt());
|
|
if (fields.hasField(this.asIndexer(StorageFile._mimeType))) m.setMimeType(d.getMimeType());
|
|
if (fields.hasField(this.asIndexer(StorageFile._storageType))) m.setStorageType(d.getStorageType());
|
|
if (fields.hasField(this.asIndexer(StorageFile._purgeAt))) m.setPurgeAt(d.getPurgeAt());
|
|
if (fields.hasField(this.asIndexer(StorageFile._purgedAt))) m.setPurgedAt(d.getPurgedAt());
|
|
if (fields.hasField(this.asIndexer(StorageFile._fullName))) m.setFullName(d.getName() + (d.getExtension().startsWith(".") ? "" : ".") + d.getExtension());
|
|
if (fields.hasField(this.asIndexer(StorageFile._belongsToCurrentTenant))) m.setBelongsToCurrentTenant(this.getBelongsToCurrentTenant(d, this.tenantScope));
|
|
if (!userFields.isEmpty() && userItemsMap != null && userItemsMap.containsKey(d.getOwnerId())) m.setOwner(userItemsMap.get(d.getOwnerId()));
|
|
models.add(m);
|
|
}
|
|
|
|
this.logger.debug("build {} items", Optional.of(models).map(List::size).orElse(0));
|
|
|
|
return models;
|
|
}
|
|
|
|
private Map<UUID, User> collectUsers(FieldSet fields, List<StorageFileEntity> data) throws MyApplicationException {
|
|
if (fields.isEmpty() || data.isEmpty())
|
|
return null;
|
|
this.logger.debug("checking related - {}", User.class.getSimpleName());
|
|
|
|
Map<UUID, User> itemMap;
|
|
if (!fields.hasOtherField(this.asIndexer(User._id))) {
|
|
itemMap = this.asEmpty(
|
|
data.stream().map(StorageFileEntity::getOwnerId).distinct().collect(Collectors.toList()),
|
|
x -> {
|
|
User item = new User();
|
|
item.setId(x);
|
|
return item;
|
|
},
|
|
User::getId);
|
|
} else {
|
|
FieldSet clone = new BaseFieldSet(fields.getFields()).ensure(User._id);
|
|
UserQuery q = this.queryFactory.query(UserQuery.class).authorize(this.authorize).ids(data.stream().map(StorageFileEntity::getOwnerId).distinct().collect(Collectors.toList()));
|
|
itemMap = this.builderFactory.builder(UserBuilder.class).authorize(this.authorize).asForeignKey(q, clone, User::getId);
|
|
}
|
|
if (!fields.hasField(User._id)) {
|
|
itemMap.forEach((id, item) -> {
|
|
if (item != null)
|
|
item.setId(null);
|
|
});
|
|
}
|
|
|
|
return itemMap;
|
|
}
|
|
|
|
}
|