use of io.grafeas.v1.DiscoveryOccurrence.AnalysisStatus in project java-docs-samples by GoogleCloudPlatform.
the class PollDiscoveryOccurrenceFinished method pollDiscoveryOccurrenceFinished.
// Repeatedly query the Container Analysis API for the latest discovery occurrence until it is
// either in a terminal state, or the timeout value has been exceeded
public static Occurrence pollDiscoveryOccurrenceFinished(String resourceUrl, String projectId, long timeoutSeconds) throws IOException, TimeoutException, InterruptedException {
// String resourceUrl = "https://gcr.io/project/image@sha256:123";
// String projectId = "my-project-id";
// long timeoutSeconds = 30;
final String projectName = ProjectName.format(projectId);
long deadline = System.currentTimeMillis() + timeoutSeconds * 1000;
// Initialize client that will be used to send requests. After completing all of your requests,
// call the "close" method on the client to safely clean up any remaining background resources.
GrafeasClient client = ContainerAnalysisClient.create().getGrafeasClient();
// find the discovery occurrence using a filter string
Occurrence discoveryOccurrence = null;
// vulbnerability discovery occurrences are always associated with the
// PACKAGE_VULNERABILITY note in the "goog-analysis" GCP project
String filter = String.format("resourceUrl=\"%s\" AND noteProjectId=\"%s\" AND noteId=\"%s\"", resourceUrl, "goog-analysis", "PACKAGE_VULNERABILITY");
// [END containeranalysis_poll_discovery_occurrence_finished]
// the above filter isn't testable, since it looks for occurrences in a locked down project
// fall back to a more permissive filter for testing
filter = String.format("kind=\"DISCOVERY\" AND resourceUrl=\"%s\"", resourceUrl);
// [START containeranalysis_poll_discovery_occurrence_finished]
while (discoveryOccurrence == null) {
for (Occurrence o : client.listOccurrences(projectName, filter).iterateAll()) {
if (o.getDiscovery() != null) {
// there should be only one valid discovery occurrence returned by the given filter
discoveryOccurrence = o;
}
}
TimeUnit.SECONDS.sleep(1);
// check for timeout
if (System.currentTimeMillis() > deadline) {
throw new TimeoutException("discovery occurrence not found");
}
}
// wait for discovery occurrence to enter a terminal state
AnalysisStatus status = AnalysisStatus.PENDING;
while (status != AnalysisStatus.FINISHED_SUCCESS && status != AnalysisStatus.FINISHED_FAILED && status != AnalysisStatus.FINISHED_UNSUPPORTED) {
// update the occurrence state
discoveryOccurrence = client.getOccurrence(discoveryOccurrence.getName());
status = discoveryOccurrence.getDiscovery().getAnalysisStatus();
TimeUnit.SECONDS.sleep(1);
// check for timeout
if (System.currentTimeMillis() > deadline) {
throw new TimeoutException("discovery occurrence not in terminal state");
}
}
return discoveryOccurrence;
}
use of io.grafeas.v1.DiscoveryOccurrence.AnalysisStatus in project java-docs-samples by GoogleCloudPlatform.
the class SamplesTest method testPollDiscoveryOccurrenceFinished.
@Test
public void testPollDiscoveryOccurrenceFinished() throws Exception {
try {
// expect fail on first try
PollDiscoveryOccurrenceFinished.pollDiscoveryOccurrenceFinished(imageUrl, PROJECT_ID, 5);
Assert.fail("found unexpected discovery occurrence");
} catch (TimeoutException e) {
// test passes
}
// create discovery note
Note newNote = Note.newBuilder().setDiscovery(DiscoveryNote.newBuilder().setAnalysisKind(NoteKind.DISCOVERY)).build();
String discNoteId = "discovery-note-" + (new Date()).getTime();
NoteName noteName = NoteName.of(PROJECT_ID, discNoteId);
GrafeasClient client = ContainerAnalysisClient.create().getGrafeasClient();
client.createNote(ProjectName.format(PROJECT_ID), discNoteId, newNote);
// create discovery occurrence
Occurrence newOcc = Occurrence.newBuilder().setNoteName(noteName.toString()).setResourceUri(imageUrl).setDiscovery(DiscoveryOccurrence.newBuilder().setAnalysisStatus(AnalysisStatus.FINISHED_SUCCESS)).build();
Occurrence result = client.createOccurrence(ProjectName.format(PROJECT_ID), newOcc);
// poll again
Occurrence found = PollDiscoveryOccurrenceFinished.pollDiscoveryOccurrenceFinished(imageUrl, PROJECT_ID, 5);
AnalysisStatus foundStatus = found.getDiscovery().getAnalysisStatus();
assertEquals(foundStatus, AnalysisStatus.FINISHED_SUCCESS);
// clean up
String[] nameArr = found.getName().split("/");
String occId = nameArr[nameArr.length - 1];
DeleteOccurrence.deleteOccurrence(occId, PROJECT_ID);
DeleteNote.deleteNote(discNoteId, PROJECT_ID);
}
Aggregations