Skip to content

Commit

Permalink
Merge pull request #2355 from PaladinCloud/feat/next-446/remove-legac…
Browse files Browse the repository at this point in the history
…y-event

feat:[NEXT-446] Remove eventbridge event sent to legacy collector
  • Loading branch information
ershad-paladin authored Jan 23, 2025
2 parents 5d94579 + c71e0b1 commit 6c1b624
Showing 1 changed file with 1 addition and 60 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -55,24 +55,14 @@ public class JobScheduler {
@Autowired
CredentialProvider credentialProvider;

@Value("${aws.eventbridge.bus.details}")
private String awsBusDetails;

@Value("${gcp.eventbridge.bus.details}")
private String gcpBusDetails;

@Value("${azure.eventbridge.bus.details}")
private String azureBusDetails;

@Value("${vulnerability.eventbridge.bus.details}")
private String vulnerabilityBusDetails;


private boolean azureEnabled;
private boolean gcpEnabled;

private boolean awsEnabled;

private boolean qualysEnabled;
private boolean aquaEnabled;

Expand Down Expand Up @@ -106,20 +96,10 @@ public void scheduleCollectorJobs() {
logger.info("Current milliseconds: {} ", System.currentTimeMillis());
logger.info("Job Scheduler for collector is running...");

EventBridgeClient eventBrClient = getEventBridgeClient();
List<PutEventsRequestEntry> putEventsRequestEntries = new ArrayList<>();

try {
ConfigUtil.setConfigProperties();
azureEnabled = Boolean.parseBoolean(env.getProperty(AZURE_ENABLED));
awsEnabled = Boolean.parseBoolean(env.getProperty(AWS_ENABLED));
boolean compositePluginEnabled = isCompositeEnabled();
if (!compositePluginEnabled && awsEnabled) {
addCollectorEvent(putEventsRequestEntries, awsBusDetails);
}
if (!compositePluginEnabled && azureEnabled) {
addCollectorEvent(putEventsRequestEntries, azureBusDetails);
}

// Sending SQS message to trigger Data-Collector
String[] plugins = pluginUsingV1.split(",");
List<String> configuredPlugins = dataCollectorSQSServic.pluginsUsingVersion1AndConfigured(plugins);
Expand All @@ -130,29 +110,10 @@ public void scheduleCollectorJobs() {
dataCollectorSQSServic.sendSQSMessage(plugin);
}

if (!putEventsRequestEntries.isEmpty()) {
PutEventsRequest eventsRequest = PutEventsRequest.builder().entries(putEventsRequestEntries)
.build();

PutEventsResponse result = eventBrClient.putEvents(eventsRequest);

for (PutEventsResultEntry resultEntry : result.entries()) {
if (resultEntry.eventId() != null) {
logger.info(EVENT_ID, resultEntry.eventId());
} else {
logger.info(FAILED_WITH_ERROR_CODE, resultEntry.errorCode());
}
}
}

} catch (EventBridgeException e) {
logger.error(e.awsErrorDetails().errorMessage());
System.exit(1);
} catch (Exception e) {
logger.error(e.getMessage());
System.exit(1);
}
eventBrClient.close();
}

private void putPluginRuleRequestEntries(int batchNo, String busDetails, List<PutEventsRequestEntry> reqEntryList, String pluginType) {
Expand Down Expand Up @@ -220,26 +181,6 @@ public void schedulePluginCollectorJobs() {
eventBrClient.close();
}


private void addCollectorEvent(List<PutEventsRequestEntry> putEventsRequestEntries, String busDetails) {
String detailString = null;

// populate events for each event bus
String[] busDetailsArray = busDetails.split(",");
for (String busDetail : busDetailsArray) {
String cloudName = busDetail.split(":")[0].split("-")[1];
Event event = populateEventForCollector(cloudName);
detailString = getMarshalledEvent(detailString, event);
PutEventsRequestEntry reqEntry = PutEventsRequestEntry.builder().source(EVENT_SOURCE).detailType(EVENT_DETAIL_TYPE).detail(detailString).eventBusName(busDetail.split(":")[0]).build();

// Add the PutEventsRequestEntry to a putEventsRequestEntries
putEventsRequestEntries.add(reqEntry);

// print the request entry
logger.info(REQUEST_ENTRY, reqEntry);
}
}

private void addPluginCollectorEvent(List<PutEventsRequestEntry> putEventsRequestEntries, String busDetails, String pluginType) {
String detailString = null;
//For custom plugins like qualys, aqua, tenable- "aqua-saasdev-aws_gcp_azure:145"
Expand Down

0 comments on commit 6c1b624

Please sign in to comment.