liuyn
2024-03-11 a87f1c3df03078814ee97ad0c8ac200a232419e9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
global class SelectRepairPDFDeleteBatch implements Database.Batchable<sObject> {
    public String query;
    public List<Repair__c> reList=new List<Repair__c>();
    public List<Id> Ids=new List<String>();
    global SelectRepairPDFDeleteBatch(list<Repair__c> scope) {
        this.reList = scope;
        for(Repair__c re:reList){
            Ids.add(re.Id);
        }
    }
 
    global Database.QueryLocator start(Database.BatchableContext bc) {
        query='SELECT Id,Name from Repair__c where Id in:Ids';
        return Database.getQueryLocator(query);
    }
 
    global void execute(Database.BatchableContext BC, list<Repair__c> repList) {
        List<String> attachmentNameList = new List<String>();
        List<Id> parentIdList = new List<Id>();
        List<ContentDocumentLink> isGeneratedPDFLinkList = new List<ContentDocumentLink>();
        List<ContentDocument> isGeneratedPDFList = new List<ContentDocument>();
        for(Repair__c re : repList){   
            attachmentNameList.add(re.name + '_' + 'MaintenanceCommission.pdf');
            parentIdList.add(re.Id);
        }
        isGeneratedPDFLinkList = [select ContentDocumentId from ContentDocumentLink where LinkedEntityId in: parentIdList];
        List<Id> attachmentIdList = new List<Id>();
        for (ContentDocumentLink PDFLink : isGeneratedPDFLinkList) {
            attachmentIdList.add(PDFLink.ContentDocumentId);
        }
        isGeneratedPDFList = [select Id from ContentDocument where Id in: attachmentIdList and Title in: attachmentNameList];
        if(isGeneratedPDFList.size() > 0){
            delete isGeneratedPDFList;
        }
 
        PageReference pdfPage;
        List<ContentVersion> versions = new List<ContentVersion>();
        for(Repair__c re : repList){       
            pdfPage = new PageReference('/apex/MaintenanceCommissionPDF?id=' + re.Id);
            Blob pdfBody;
            if(System.Test.isRunningTest()) { 
                pdfBody = blob.valueOf('Unit.Test');
            } else {
                pdfBody = pdfPage.getContentAsPDF();
            }
            ContentVersion version = new ContentVersion();
            version.VersionData = pdfBody;
            version.Title = re.name + '_' + 'MaintenanceCommission.pdf';
            version.ContentLocation = 's';
            version.PathOnClient = re.name + '_' + 'MaintenanceCommission.pdf';
            versions.add(version);
        }
 
        if (versions.size() > 0) {
            insert versions;
        }
        List<Id> idList = new List<Id>();
        for (ContentVersion newVersion : versions) {
            idList.add(newVersion.Id);
        }
        versions = [select ContentDocumentId from ContentVersion where Id in: idList];
        List<ContentDocumentLink> links = new List<ContentDocumentLink>();
        Integer i = 0;
        for (ContentVersion newVersion : versions) {
            ContentDocumentLink link = new ContentDocumentLink();
            link.ContentDocumentId = newVersion.ContentDocumentId;
            link.LinkedEntityId = repList[i].Id;
            link.ShareType = 'I';
            link.Visibility = 'AllUsers';
            links.add(link);
            i++;
        }
 
        if (links.size() > 0) {
            insert links;
        }
    }
 
    global void finish(Database.BatchableContext BC) {
 
    }
}