82 lines
3.0 KiB
Java
82 lines
3.0 KiB
Java
package eu.openaire.urls_worker.controllers;
|
|
|
|
import eu.openaire.urls_worker.UrlsWorkerApplication;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.http.HttpStatus;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|
import org.springframework.web.bind.annotation.PathVariable;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
|
|
@RestController
|
|
@RequestMapping("")
|
|
public class GeneralController {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(GeneralController.class);
|
|
|
|
|
|
public GeneralController() {}
|
|
|
|
|
|
@GetMapping("isAlive")
|
|
public ResponseEntity<?> isWorkerAlive() {
|
|
logger.info("Received an \"isAlive\" request.");
|
|
return ResponseEntity.ok().build();
|
|
}
|
|
|
|
|
|
public static boolean shouldShutdownWorker = false;
|
|
|
|
@GetMapping("shutdownWorker/{shutdownCode}")
|
|
public ResponseEntity<?> shutdownWorkerGracefully(@PathVariable String shutdownCode)
|
|
{
|
|
String initMsg = "Received a \"shutdownWorker\" request.";
|
|
if ( shutdownCode.equals(UrlsWorkerApplication.shutdownOrCancelCode) ) {
|
|
shouldShutdownWorker = true;
|
|
logger.info(initMsg + " The worker will shutdown, after finishing current work.");
|
|
return ResponseEntity.ok().build();
|
|
} else {
|
|
String errorMsg = initMsg + " But, it has an invalid \"shutdownCode\": " + shutdownCode;
|
|
logger.error(errorMsg);
|
|
return ResponseEntity.status(HttpStatus.FORBIDDEN).body(errorMsg);
|
|
}
|
|
}
|
|
|
|
|
|
@GetMapping("cancelShutdownWorker/{cancelCode}")
|
|
public ResponseEntity<?> cancelShutdownWorkerGracefully(@PathVariable String cancelCode)
|
|
{
|
|
String initMsg = "Received a \"cancelShutdownWorker\" request.";
|
|
if ( cancelCode.equals(UrlsWorkerApplication.shutdownOrCancelCode) ) {
|
|
shouldShutdownWorker = false;
|
|
logger.info(initMsg + " Any previous \"shutdownWorker\"-request is canceled. The \"maxAssignmentsBatchesToHandleBeforeShutdown\" will still be honored (if it's set).");
|
|
return ResponseEntity.ok().build();
|
|
} else {
|
|
String errorMsg = initMsg + " But, it has an invalid \"cancelCode\": " + cancelCode;
|
|
logger.error(errorMsg);
|
|
return ResponseEntity.status(HttpStatus.FORBIDDEN).body(errorMsg);
|
|
}
|
|
}
|
|
|
|
|
|
@GetMapping("getHandledAssignmentsCounts")
|
|
public ResponseEntity<?> getHandledAssignmentsCounts()
|
|
{
|
|
List<Long> handledAssignmentsCounts = new ArrayList<>(FullTextsController.assignmentsNumsHandledAndLocallyDeleted.size()/2);
|
|
for ( Map.Entry<Long,Boolean> entry : FullTextsController.assignmentsNumsHandledAndLocallyDeleted.entrySet() )
|
|
{
|
|
if ( entry.getValue().equals(true) )
|
|
handledAssignmentsCounts.add(entry.getKey());
|
|
}
|
|
return ResponseEntity.ok(handledAssignmentsCounts);
|
|
}
|
|
|
|
}
|