yakworks.rally.attachment.repo.AttachmentLinkRepo.groovy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rally-domain Show documentation
gorm domain models for core app and testing
There is a newer version: 7.3.76
/*
* Copyright 2021 Yak.Works - Licensed under the Apache License, Version 2.0 (the "License")
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
*/
package yakworks.rally.attachment.repo

import groovy.transform.CompileStatic

import org.springframework.beans.factory.annotation.Autowired

import gorm.tools.model.Persistable
import gorm.tools.problem.ProblemHandler
import gorm.tools.repository.GormRepository
import gorm.tools.repository.model.AbstractLinkedEntityRepo
import yakworks.api.ApiResults
import yakworks.api.Result
import yakworks.rally.attachment.model.Attachable
import yakworks.rally.attachment.model.Attachment
import yakworks.rally.attachment.model.AttachmentLink

@GormRepository
@CompileStatic
class AttachmentLinkRepo extends AbstractLinkedEntityRepo {

    @Autowired(required = false)
    ProblemHandler problemHandler

    AttachmentLinkRepo(){
        super(Attachment, 'attachment')
    }

    @Override
    List addOrRemove(Persistable entity, Object itemParams){
        def list = super.addOrRemove(entity, itemParams)
        updateAttachableHasAttachments(entity, list)
        return list
    }

    /**
     * updates the cached hasAttachments on the attachable entity
     */
    void updateAttachableHasAttachments(Persistable entity, List linkList){
        // update the has attachments
        if(Attachable.isAssignableFrom(entity.class)){
            def attachableEntity = (Attachable)entity
            attachableEntity.setHasAttachments(linkList?.size() as Boolean)
        }
    }

    boolean hasAttachments(Persistable entity) {
        count(entity)
    }

    List listAttachments(Persistable entity) {
        listRelated(entity)
    }

    //
    // @RepoListener
    // void afterPersist(AttachmentLink activity, AfterPersistEvent e) {
    //     //FIXME this is a hack so the events for links get fired after data is inserted
    //     // not very efficient as removes batch inserting for lots of acts so need to rethink this strategy
    //     flush()
    // }

    /**
     * Copies Attachments from the source to target
     *
     * @param fromEntity entity to copy attachments from
     * @param toEntity entity to copy attachments to
     * @return the ApiResults which will be ok or have errors if problem occured with IO
     */
    //FIXME needs a good test
    Result copy(Persistable fromEntity, Persistable toEntity) {
        ApiResults results = ApiResults.OK()
        List attachLinks = queryFor(fromEntity).list()
        for(AttachmentLink attachLink : attachLinks){
            //catch exceptions and move on in case attachment has a bad link we dont want to fail the whole thing
            try {
                Attachment attachmentCopy = Attachment.repo.copy(attachLink.attachment)
                if (attachmentCopy) create(toEntity, attachmentCopy)
            } catch (ex){
                results << problemHandler.handleException(ex)
            }
        }
        return results
    }

}