-
Notifications
You must be signed in to change notification settings - Fork 277
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #208 from tmobile/release/1.3
Release/1.3
- Loading branch information
Showing
204 changed files
with
22,023 additions
and
12,465 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
205 changes: 205 additions & 0 deletions
205
...api-admin/src/main/java/com/tmobile/pacman/api/admin/repository/service/AdminService.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,205 @@ | ||
| package com.tmobile.pacman.api.admin.repository.service; | ||
|
|
||
| import java.util.ArrayList; | ||
| import java.util.HashMap; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import java.util.stream.Collectors; | ||
|
|
||
| import org.slf4j.Logger; | ||
| import org.slf4j.LoggerFactory; | ||
| import org.springframework.beans.factory.annotation.Autowired; | ||
| import org.springframework.stereotype.Service; | ||
|
|
||
| import com.amazonaws.services.cloudwatchevents.model.DisableRuleRequest; | ||
| import com.amazonaws.services.cloudwatchevents.model.EnableRuleRequest; | ||
| import com.amazonaws.services.cloudwatchevents.model.ListRulesRequest; | ||
| import com.amazonaws.services.cloudwatchevents.model.ListRulesResult; | ||
| import com.amazonaws.services.cloudwatchevents.model.RuleState; | ||
| import com.tmobile.pacman.api.admin.common.AdminConstants; | ||
| import com.tmobile.pacman.api.admin.config.PacmanConfiguration; | ||
| import com.tmobile.pacman.api.admin.exceptions.PacManException; | ||
| import com.tmobile.pacman.api.admin.repository.JobExecutionManagerRepository; | ||
| import com.tmobile.pacman.api.admin.repository.RuleRepository; | ||
| import com.tmobile.pacman.api.admin.repository.model.JobExecutionManager; | ||
| import com.tmobile.pacman.api.admin.repository.model.Rule; | ||
| import com.tmobile.pacman.api.admin.service.AmazonClientBuilderService; | ||
|
|
||
| @Service | ||
| public class AdminService { | ||
|
|
||
| private static final Logger log = LoggerFactory.getLogger(AdminService.class); | ||
|
|
||
| @Autowired | ||
| private RuleRepository ruleRepository; | ||
|
|
||
| @Autowired | ||
| private JobExecutionManagerRepository jobRepository; | ||
|
|
||
| @Autowired | ||
| private AmazonClientBuilderService amazonClient; | ||
|
|
||
| @Autowired | ||
| private PacmanConfiguration config; | ||
|
|
||
| public String shutDownAlloperations(String operation, String job) throws PacManException { | ||
|
|
||
| String nextToken = null; | ||
| ListRulesResult listRulesResult ; | ||
| List<String> rules = new ArrayList<>(); | ||
| do{ | ||
| listRulesResult = amazonClient.getAmazonCloudWatchEvents(config.getRule().getLambda().getRegion()).listRules(new ListRulesRequest().withNextToken(nextToken)); | ||
| rules.addAll(listRulesResult.getRules().parallelStream().map(rule->rule.getName()).collect(Collectors.toList())); | ||
| nextToken = listRulesResult.getNextToken(); | ||
| }while(nextToken!=null); | ||
|
|
||
| if(operation.equals(AdminConstants.ENABLE)) { | ||
| if(job.equals(AdminConstants.RULE)) { | ||
| if(enableRules(rules)) { | ||
| return "All Rules has been sucessfully enabled"; | ||
| } | ||
| } else if(job.equals(AdminConstants.JOB)) { | ||
| if(enableJobs(rules)) { | ||
| return "All Jobs has been sucessfully enabled"; | ||
| } | ||
| } else { | ||
| if(enableRules(rules) && enableJobs(rules)) { | ||
| return "All Rules and Jobs has been sucessfully enabled"; | ||
| } | ||
| } | ||
| throw new PacManException("Enabling operation failed"); | ||
| } else { | ||
| if(job.equals(AdminConstants.RULE)) { | ||
| if(disableRules(rules)) { | ||
| return "All Rules has been sucessfully disabled"; | ||
| } | ||
| } else if(job.equals(AdminConstants.JOB)) { | ||
| if(disableJobs(rules)) { | ||
| return "All Jobs has been sucessfully disabled"; | ||
| } | ||
| } else { | ||
| if(disableRules(rules) && disableJobs(rules)) { | ||
| return "All Rules and Jobs has been sucessfully disabled"; | ||
| } | ||
| } | ||
| throw new PacManException("Disabling operation failed"); | ||
| } | ||
| } | ||
|
|
||
| private boolean disableRules(List<String> rules) { | ||
| List<Rule> ruleIds = ruleRepository.findAll(); | ||
| try { | ||
| for(Rule rule : ruleIds) { | ||
| if(rules.contains(rule.getRuleUUID())) { | ||
| amazonClient.getAmazonCloudWatchEvents(config.getRule().getLambda().getRegion()) | ||
| .disableRule(new DisableRuleRequest().withName(rule.getRuleUUID())); | ||
| rule.setStatus(RuleState.DISABLED.name()); | ||
| ruleRepository.save(rule); | ||
| } | ||
| } | ||
| return true; | ||
| } catch(Exception e) { | ||
| log.error("Error in disable rules",e); | ||
| return false; | ||
| } | ||
|
|
||
| } | ||
|
|
||
| private boolean disableJobs(List<String> rules) { | ||
| List<JobExecutionManager> jobs = jobRepository.findAll(); | ||
| try { | ||
| for(JobExecutionManager job : jobs) { | ||
| if(rules.contains(job.getJobUUID())) { | ||
| job.getJobUUID(); | ||
| amazonClient.getAmazonCloudWatchEvents(config.getRule().getLambda().getRegion()) | ||
| .disableRule(new DisableRuleRequest().withName(job.getJobUUID())); | ||
| job.setStatus(RuleState.DISABLED.name()); | ||
| jobRepository.save(job); | ||
| } | ||
| } | ||
| return true; | ||
| } catch(Exception e) { | ||
| log.error("Error in disable jobs",e); | ||
| return false; | ||
| } | ||
| } | ||
|
|
||
| private boolean enableRules(List<String> rules) { | ||
| List<Rule> ruleIds = ruleRepository.findAll(); | ||
| try { | ||
| for(Rule rule : ruleIds) { | ||
| if(rules.contains(rule.getRuleUUID())) { | ||
| amazonClient.getAmazonCloudWatchEvents(config.getRule().getLambda().getRegion()) | ||
| .enableRule(new EnableRuleRequest().withName(rule.getRuleUUID())); | ||
| rule.setStatus(RuleState.ENABLED.name()); | ||
| ruleRepository.save(rule); | ||
| } | ||
| } | ||
| return true; | ||
| } catch(Exception e) { | ||
| log.error("Error in enable rules",e); | ||
| return false; | ||
| } | ||
| } | ||
|
|
||
| private boolean enableJobs(List<String> rules) { | ||
| List<JobExecutionManager> jobs = jobRepository.findAll(); | ||
| try { | ||
| for(JobExecutionManager job : jobs) { | ||
| if(rules.contains(job.getJobUUID())) { | ||
| amazonClient.getAmazonCloudWatchEvents(config.getRule().getLambda().getRegion()) | ||
| .enableRule(new EnableRuleRequest().withName(job.getJobUUID())); | ||
| job.setStatus(RuleState.ENABLED.name()); | ||
| jobRepository.save(job); | ||
| } | ||
| } | ||
| return true; | ||
| } catch(Exception e) { | ||
| log.error("Error in enable jobs",e); | ||
| return false; | ||
| } | ||
| } | ||
|
|
||
| public Map<String,String> statusOfSystem() throws PacManException{ | ||
|
|
||
| Map<String,String> status = new HashMap<>(); | ||
| try { | ||
| List<Rule> rules = ruleRepository.findAll(); | ||
| List<JobExecutionManager> jobs = jobRepository.findAll(); | ||
|
|
||
| boolean rulesEnabled = false; | ||
| boolean jobsEnabled = false; | ||
|
|
||
| for(Rule rule : rules) { | ||
| if(rule.getStatus().equals(RuleState.ENABLED.name())) { | ||
| rulesEnabled = true; | ||
| break; | ||
| } | ||
| } | ||
|
|
||
| for(JobExecutionManager job : jobs) { | ||
| if(job.getStatus().equals(RuleState.ENABLED.name())) { | ||
| jobsEnabled = true; | ||
| break; | ||
| } | ||
| } | ||
|
|
||
| if(rulesEnabled) { | ||
| status.put("rule", RuleState.ENABLED.name()); | ||
| } else { | ||
| status.put("rule", RuleState.DISABLED.name()); | ||
| } | ||
|
|
||
| if(jobsEnabled) { | ||
| status.put("job", RuleState.ENABLED.name()); | ||
| } else { | ||
| status.put("job", RuleState.DISABLED.name()); | ||
| } | ||
| return status; | ||
| } catch(Exception e) { | ||
| log.error("Error in fetching status of system",e); | ||
| throw new PacManException("Error in fetching the status of system"); | ||
| } | ||
| } | ||
|
|
||
| } |
Oops, something went wrong.