S3FolderMigrationClientWithTagging.groovy

import com.amazonaws.auth.SystemPropertiesCredentialsProvider
import com.amazonaws.services.s3.model.ObjectMetadata
import com.amazonaws.services.s3.model.ObjectTagging
import com.amazonaws.services.s3.model.Tag
import com.amazonaws.services.s3.transfer.MultipleFileUpload
import com.amazonaws.services.s3.transfer.ObjectMetadataProvider
import com.amazonaws.services.s3.transfer.ObjectTaggingProvider
import com.amazonaws.services.s3.transfer.TransferManagerWithTagging
import com.amazonaws.services.s3.transfer.TransferProgress


class S3FolderMigrationClientWithTagging {
  def loadMultipleFiles = { String directoryName ->
    ObjectTaggingProvider taggingProvider = new ObjectTaggingProvider() {
        ObjectTagging provideObjectTags(File file){
            List<Tag> tagList = new ArrayList<Tag>()
            tagList.add(new Tag(‘costCenter', costCenterName))
            tagList.add(new Tag(‘client', clientName))
            new ObjectTagging(tagList)
        }
    }

    File directory = new File("./${directoryName}" )
    if(!directory.isDirectory()){
        directory.mkdir()
    }

    MultipleFileUpload multipleFileUpload = transferManager.uploadDirectory(bucketName, contextName, directory, false, metadataProvider, taggingProvider)
    TransferProgress transferProgress  = multipleFileUpload.progress
    while (!multipleFileUpload.done){
        Thread.sleep(1000)
    }
    directory.delete()
  }                                
}