From d8dc84a3d56df839895f1c417a4d9cbee763d262 Mon Sep 17 00:00:00 2001
From: 高章伟 <gaozhangwei@prec-tech.com>
Date: 星期五, 03 三月 2023 14:50:59 +0800
Subject: [PATCH] gzw 测试环境代码更新

---
 force-app/main/default/classes/SyncAccountContactToAWS.cls |  135 +++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 125 insertions(+), 10 deletions(-)

diff --git a/force-app/main/default/classes/SyncAccountContactToAWS.cls b/force-app/main/default/classes/SyncAccountContactToAWS.cls
index df049b2..58a43c2 100644
--- a/force-app/main/default/classes/SyncAccountContactToAWS.cls
+++ b/force-app/main/default/classes/SyncAccountContactToAWS.cls
@@ -10,47 +10,162 @@
 global class  SyncAccountContactToAWS implements Schedulable,Database.Batchable<SObject>,Database.AllowsCallouts{
 
     global String query;
-    
+    global String scheduleId{set;get;}
+    global Map<Id,Contact> ContactMap{set;get;}
     global SyncAccountContactToAWS(String query) {
         this.query = query;
+    }
+    global SyncAccountContactToAWS(String query,Map<Id,Contact> ContactMap) {
+        this.query = query;
+        this.ContactMap = ContactMap;
     }
     global SyncAccountContactToAWS() {
         this.query = 'SELECT id,Doctor_Division1__c,Doctor_Division1_Encrypted__c,Name,Name_Encrypted__c,Type__c,Type_Encrypted__c,AWS_Data_Id__c,Contact__c, Contact__r.Doctor_Division1_Encrypted__c,Contact__r.LastName_Encrypted__c, Contact__r.Type_Encrypted__c FROM Agency_Contact__c WHERE AWS_Data_Id__c =\'\' And Contact__c != null order by lastmodifieddate desc';
     }
+    global SyncAccountContactToAWS(String query,String scId) {
+        if(String.isBlank(query)||String.isEmpty(query)){
+            this.query = 'SELECT id,Doctor_Division1__c,Doctor_Division1_Encrypted__c,Name,Name_Encrypted__c,Type__c,Type_Encrypted__c,AWS_Data_Id__c,Contact__c, Contact__r.Doctor_Division1_Encrypted__c,Contact__r.LastName_Encrypted__c, Contact__r.Type_Encrypted__c FROM Agency_Contact__c WHERE AWS_Data_Id__c =\'\' And Contact__c != null order by lastmodifieddate desc';
+        }else{
+            this.query = query;
+        }
+        system.debug('This query from Batch:'+this.query);
+        this.scheduleId = scId;
+    }
+    //zhj 2022-01-03 start
+    global SyncAccountContactToAWS(String query,String scId,Map<Id,Contact> ContactMap) {
+        if(String.isBlank(query)||String.isEmpty(query)){
+            this.query = 'SELECT id,Doctor_Division1__c,Doctor_Division1_Encrypted__c,Name,Name_Encrypted__c,Type__c,Type_Encrypted__c,AWS_Data_Id__c,Contact__c, Contact__r.Doctor_Division1_Encrypted__c,Contact__r.LastName_Encrypted__c, Contact__r.Type_Encrypted__c FROM Agency_Contact__c WHERE AWS_Data_Id__c =\'\' And Contact__c != null order by lastmodifieddate desc';
+        }else{
+            this.query = query;
+        }
+        system.debug('This query from Batch:'+this.query);
+        this.scheduleId = scId;
+        this.ContactMap = ContactMap;
+    }
+    //zhj 2022-01-03 end
     global Database.QueryLocator start(Database.BatchableContext bc) {
-        system.debug('Query by custom soql:'+this.query);
+        system.debug('Query by custom soql:'+this.query);  
         return Database.getQueryLocator(this.query);
     }
     global void execute(Database.BatchableContext BC, list<Agency_Contact__c> scope) {
+        System.debug('ContactMap = ' + ContactMap);
         if(scope!=null && scope.size()>0){
+            //zhj MEBG鏂版柟妗堟敼閫� 2022-11-28 start
             for(Agency_Contact__c  aContact:scope){
-                aContact.Doctor_Division1_Encrypted__c = aContact.Contact__c!=null&&aContact.Contact__r.Doctor_Division1_Encrypted__c!=''?aContact.Contact__r.Doctor_Division1_Encrypted__c:'';
-                aContact.Name_Encrypted__c =  aContact.Contact__c!=null&&aContact.Contact__r.LastName_Encrypted__c!=''?aContact.Contact__r.LastName_Encrypted__c:'';
-                aContact.Type_Encrypted__c =  aContact.Contact__c!=null&&aContact.Contact__r.Type_Encrypted__c!=''?aContact.Contact__r.Type_Encrypted__c:'';
+                //aContact.Doctor_Division1_Encrypted__c = aContact.Contact__c!=null&&aContact.Contact__r.Doctor_Division1_Encrypted__c!=''?aContact.Contact__r.Doctor_Division1_Encrypted__c:'';
+                //aContact.Name_Encrypted__c =  aContact.Contact__c!=null&&aContact.Contact__r.LastName_Encrypted__c!=''?aContact.Contact__r.LastName_Encrypted__c:'';
+                //aContact.Type_Encrypted__c =  aContact.Contact__c!=null&&aContact.Contact__r.Type_Encrypted__c!=''?aContact.Contact__r.Type_Encrypted__c:'';
                 aContact.AWS_Data_Id__c = '';
             }
             system.debug('Agency Contact Info to AWS:'+JSON.serialize(scope));
-            AWSServiceTool2.EncryptPushCore(JSON.serialize(scope),'Agency_Contact__c');
+
+            Map<String,PIHelper.PIIntegration> staticResource = new Map<String,PIHelper.PIIntegration>();
+            staticResource.put('Contact',PIHelper.getPIIntegrationInfo('Contact'));
+            staticResource.put('Agency_Contact__c',PIHelper.getPIIntegrationInfo('Agency_Contact__c'));
+            Map<String, Map<String, PI_Field_Policy_Detail__c>> mmsp = new Map<String, Map<String,PI_Field_Policy_Detail__c>>();
+            for (String key : staticResource.keySet()) {
+                mmsp.put(key, new Map<String,PI_Field_Policy_Detail__c>());
+                for (PI_Field_Policy_Detail__c detail : staticResource.get(key).PIDetails) {
+                    mmsp.get(key).put(detail.SF_Field_API_Name__c, detail);
+                }
+            }
+
+            System.debug('mmsp = ' + mmsp);
+            List<AWSServiceTool2V2.EncryptPushRequestBody> EncryptPushList = new List<AWSServiceTool2V2.EncryptPushRequestBody>();
+            for(Agency_Contact__c ac : scope){
+                AWSServiceTool2V2.EncryptPushRequestBody EncryptPush = new AWSServiceTool2V2.EncryptPushRequestBody();
+                EncryptPush.dataId = ac.AWS_Data_Id__c != null ?ac.AWS_Data_Id__c:'';
+                EncryptPush.sfRecordId = ac.Id;
+                EncryptPush.fieldsMapping = new Map<String, List<AWSServiceTool2V2.EncryptPushRes>>();
+                List<AWSServiceTool2V2.EncryptPushRes> resList = new List<AWSServiceTool2V2.EncryptPushRes>();
+                AWSServiceTool2V2.EncryptPushRes res= new AWSServiceTool2V2.EncryptPushRes();
+                res.isQueryDb = true;
+                res.value = '';
+                res.table = staticResource.get('Contact').awsTableName;
+                //res.dataId = ContactMap.get(ac.Contact__c).AWS_Data_Id__c;
+                res.dataId = (ContactMap == null || ContactMap.size() == 0) ?'':ContactMap.get(ac.Contact__c).AWS_Data_Id__c;
+                res.field = mmsp.get('Contact').get('LastName').AWS_Field_API__c;
+                resList.add(res);
+
+                List<AWSServiceTool2V2.EncryptPushRes> resList2 = new List<AWSServiceTool2V2.EncryptPushRes>();
+                AWSServiceTool2V2.EncryptPushRes res2= new AWSServiceTool2V2.EncryptPushRes();
+                res2.isQueryDb = true;
+                res2.value = '';
+                res2.table = staticResource.get('Contact').awsTableName;
+                //res2.dataId = ContactMap.get(ac.Contact__c).AWS_Data_Id__c;
+                res2.dataId = (ContactMap == null || ContactMap.size() == 0) ?'':ContactMap.get(ac.Contact__c).AWS_Data_Id__c;
+                res2.field = mmsp.get('Contact').get('Type__c').AWS_Field_API__c;
+                resList2.add(res2);
+
+                List<AWSServiceTool2V2.EncryptPushRes> resList3 = new List<AWSServiceTool2V2.EncryptPushRes>();
+                AWSServiceTool2V2.EncryptPushRes res3= new AWSServiceTool2V2.EncryptPushRes();
+                res3.isQueryDb = true;
+                res3.value = '';
+                res3.table = staticResource.get('Contact').awsTableName;
+                //res3.dataId = ContactMap.get(ac.Contact__c).AWS_Data_Id__c;
+                res3.dataId = (ContactMap == null || ContactMap.size() == 0) ?'':ContactMap.get(ac.Contact__c).AWS_Data_Id__c;
+                res3.field = mmsp.get('Contact').get('Doctor_Division1__c').AWS_Field_API__c;
+                resList3.add(res3);
+
+                System.debug('Agency_Contact__c Name = ' +mmsp.get('Agency_Contact__c').get('Name').AWS_Field_API__c);
+                System.debug('resList = ' + resList);
+                EncryptPush.fieldsMapping.put(mmsp.get('Agency_Contact__c').get('Name').AWS_Field_API__c, resList);
+                EncryptPush.fieldsMapping.put(mmsp.get('Agency_Contact__c').get('Type__c').AWS_Field_API__c, resList2);
+                EncryptPush.fieldsMapping.put(mmsp.get('Agency_Contact__c').get('Doctor_Division1__c').AWS_Field_API__c, resList3);
+                EncryptPushList.add(EncryptPush);
+            }
+            System.debug('EncryptPushListdataId = ' + JSON.serialize(EncryptPushList[0].dataId));
+            System.debug('EncryptPushListsfRecordId = ' + JSON.serialize(EncryptPushList[0].sfRecordId));
+            System.debug('EncryptPushListfieldsMapping = ' + JSON.serialize(EncryptPushList[0].fieldsMapping));
+            //AWSServiceTool2.EncryptPushCore(JSON.serialize(scope),'Agency_Contact__c');
+            if(!Test.isRunningTest()){
+                AwsServiceTool2V2.EncryptPushCoreV2(Json.serialize(EncryptPushList),JSON.serialize(scope), 'Agency_Contact__c');
+            }
+            //zhj MEBG鏂版柟妗堟敼閫� 2022-11-28 end
         }      
     }
 
     global void execute(SchedulableContext SC) {
-        Id execBTId = Database.executeBatch(new SyncAccountContactToAWS(), 1);
+        system.debug('this query from schedule execute method:'+this.query);
+        Id execBTId = Database.executeBatch(new SyncAccountContactToAWS(this.query,SC.getTriggerId(),this.ContactMap), 1);       
+        system.debug('Batch Job Id:'+execBTId);
     }
 
-    public static void assignOnceOneMinuteLater() {
+    public static void assignOnceOneMinuteLater(List<Agency_Contact__c> acList,Map<Id,Contact> cm) {
+        system.debug('ACList:'+JSON.serialize(acList));
         String hour = String.valueOf(Datetime.now().hour());
         String min = String.valueOf(Datetime.now().minute() + 1); 
         String ss = String.valueOf(Datetime.now().second());
         //parse to cron expression
         String nextFireTime = ss + ' ' + min + ' ' + hour + ' * * ?';
-        SyncAccountContactToAWS s = new SyncAccountContactToAWS(); 
-        System.schedule('Job Started At ' + String.valueOf(Datetime.now()), nextFireTime, s);
+        //Prepare the soql to agency account
+        String soqlForAgencyAccount = '';
+        if(acList!=null && acList.size()>0){
+            soqlForAgencyAccount = 'SELECT id,Doctor_Division1__c,Doctor_Division1_Encrypted__c,Name,Name_Encrypted__c,Type__c,Type_Encrypted__c,AWS_Data_Id__c,Contact__c, Contact__r.Doctor_Division1_Encrypted__c,Contact__r.LastName_Encrypted__c, Contact__r.Type_Encrypted__c FROM Agency_Contact__c WHERE id in ';
+            List<String> acIds = new List<String>();
+            for(Agency_Contact__c ac:acList ){
+                acIds.add(ac.Id);
+            }
+            soqlForAgencyAccount = soqlForAgencyAccount +  '(\'' + String.join(acIds, '\', \'') + '\')';
+        }
+        system.debug('SOQL for agency:'+soqlForAgencyAccount);
+        system.debug('cm  = '+cm);
+        SyncAccountContactToAWS s = new SyncAccountContactToAWS(soqlForAgencyAccount,cm);
+        try{
+            System.schedule('SyncAccountContactToAWS:' + String.valueOf(Datetime.now()), nextFireTime, s);
+        }catch(Exception e){
+            system.debug('Exception from SyncAccountContactToAWS:'+e.getMessage());
+        }      
     }
 
     global void finish(Database.BatchableContext BC) {
+        system.debug('Finish batch job Id:'+BC.getJobId()); 
         AsyncApexJob a = [SELECT Id, Status, NumberOfErrors, JobItemsProcessed, TotalJobItems, CreatedBy.Email FROM AsyncApexJob WHERE Id = :BC.getJobId()];
         //then use the active job id and abort it
         system.abortJob(a.id);
+        system.debug('Finish schedule job Id:'+scheduleId); 
+        if(scheduleId!=null&&scheduleId!=''){
+            system.abortJob(scheduleId);
+        }
     }
 }
\ No newline at end of file

--
Gitblit v1.9.1