Examples with GraphRecordStore au.gov.asd.tac.constellation.graph.processing.GraphRecordStore used on opensource projects

Example 1 with GraphRecordStore

use of au.gov.asd.tac.constellation.graph.processing.GraphRecordStore in project constellation by constellation-app.

the class TestParametersPlugin method query.

@Override
protected RecordStore query(final RecordStore query, final PluginInteraction interaction, final PluginParameters parameters) throws PluginException {
    final int sleep = parameters.getParameters().get(SLEEP_PARAMETER_ID).getIntegerValue();
    for (int i = 0; i < sleep; i++) {
        LOGGER.log(Level.INFO, "sleep {0}/{1}", new Object[] { i, sleep });
        try {
            Thread.sleep(1000);
        } catch (InterruptedException ex) {
            Exceptions.printStackTrace(ex);
            Thread.currentThread().interrupt();
        }
    }
    LOGGER.log(Level.INFO, "slept for {0} seconds", sleep);
    LOGGER.log(Level.INFO, "parameters: {0}", parameters);
    LOGGER.log(Level.INFO, "GraphElementType: {0}", parameters.getSingleChoice(ELEMENT_TYPE_PARAMETER_ID));
    final LocalDate localDate = parameters.getLocalDateValue(LOCAL_DATE_PARAMETER_ID);
    LOGGER.log(Level.INFO, "localdate: {0} ", localDate);
    if (localDate != null) {
        final Calendar cal = LocalDateParameterType.toCalendar(localDate);
        LOGGER.log(Level.INFO, String.format("toDate: [%s] [%04d-%02d-%02d]", cal, cal.get(Calendar.YEAR), cal.get(Calendar.MONTH), cal.get(Calendar.DAY_OF_MONTH)));
        LOGGER.log(Level.INFO, String.format("fields: [%04d-%02d-%02d]", localDate.get(ChronoField.YEAR), localDate.get(ChronoField.MONTH_OF_YEAR), localDate.get(ChronoField.DAY_OF_MONTH)));
    }
    final MultiChoiceParameterValue planets = parameters.getMultiChoiceValue(PLANETS_PARAMETER_ID);
    planets.getChoices().stream().forEach(planet -> LOGGER.log(Level.INFO, "Planet: {0}", planet));
    LOGGER.log(Level.INFO, "==== begin string values");
    parameters.getParameters().values().stream().forEach(param -> LOGGER.log(Level.INFO, "String {0}: \"{1}\"", new Object[] { param.getName(), param.getStringValue() }));
    LOGGER.log(Level.INFO, "==== end string values");
    final File outputDir = DataAccessPreferenceUtilities.getDataAccessResultsDir();
    if (outputDir != null) {
        final String fnam = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMdd'T'HHmmss")) + "-testChainer.txt";
        final File fout = new File(outputDir, fnam);
        try (final PrintWriter writer = new PrintWriter(fout, StandardCharsets.UTF_8.name())) {
            parameters.getParameters().values().stream().forEach(param -> writer.printf("%s: '%s'", param.getName(), param.getStringValue()));
        } catch (final FileNotFoundException | UnsupportedEncodingException ex) {
            Exceptions.printStackTrace(ex);
        }
    }
    final List<String> keys = query.keys();
    while (query.next()) {
        keys.stream().forEach(key -> LOGGER.log(Level.INFO, String.format("%-20s: %s", key, query.get(key))));
        LOGGER.log(Level.INFO, "--");
    }
    final boolean crash = parameters.getBooleanValue(CRASH_PARAMETER_ID);
    if (crash) {
        throw new RuntimeException("Simulated plugin failure");
    }
    try {
        interaction.setProgress(1, 0, String.format("Pretended to add %d node(s), modify %d node(s)", r.nextInt(100) + 1, r.nextInt(100) + 1), false);
    } catch (InterruptedException ex) {
        Exceptions.printStackTrace(ex);
        Thread.currentThread().interrupt();
    }
    final String queryName = parameters.getStringValue(CoreGlobalParameters.QUERY_NAME_PARAMETER_ID);
    LOGGER.log(Level.INFO, "query name: {0}", queryName);
    final DateTimeRange dtr = parameters.getDateTimeRangeValue(CoreGlobalParameters.DATETIME_RANGE_PARAMETER_ID);
    final ZonedDateTime[] dtrStartEnd = dtr.getZonedStartEnd();
    LOGGER.log(Level.INFO, "range: (zdt) {0} .. {1}", new Object[] { dtrStartEnd[0], dtrStartEnd[1] });
    LOGGER.log(Level.INFO, "range: (zdt) {0} .. {1}", new Object[] { DateTimeFormatter.ISO_INSTANT.format(dtrStartEnd[0]), DateTimeFormatter.ISO_INSTANT.format(dtrStartEnd[1]) });
    final String interactionLevel = parameters.getParameters().get(INTERACTION_PARAMETER_ID).getStringValue();
    final PluginNotificationLevel pnInteractionLevel;
    if (interactionLevel != null) {
        switch(interactionLevel) {
            case DEBUG:
                pnInteractionLevel = PluginNotificationLevel.DEBUG;
                break;
            case INFO:
                pnInteractionLevel = PluginNotificationLevel.INFO;
                break;
            case WARNING:
                pnInteractionLevel = PluginNotificationLevel.WARNING;
                break;
            case ERROR:
                pnInteractionLevel = PluginNotificationLevel.ERROR;
                break;
            case FATAL:
                pnInteractionLevel = PluginNotificationLevel.FATAL;
                break;
            default:
                pnInteractionLevel = null;
                break;
        }
        if (pnInteractionLevel != null) {
            interaction.notify(pnInteractionLevel, "Interaction from plugin");
        }
    }
    final String exceptionLevel = parameters.getParameters().get(LEVEL_PARAMETER_ID).getStringValue();
    final PluginNotificationLevel pnExceptionLevel;
    if (exceptionLevel != null) {
        switch(exceptionLevel) {
            case DEBUG:
                pnExceptionLevel = PluginNotificationLevel.DEBUG;
                break;
            case INFO:
                pnExceptionLevel = PluginNotificationLevel.INFO;
                break;
            case WARNING:
                pnExceptionLevel = PluginNotificationLevel.WARNING;
                break;
            case ERROR:
                pnExceptionLevel = PluginNotificationLevel.ERROR;
                break;
            case FATAL:
                pnExceptionLevel = PluginNotificationLevel.FATAL;
                break;
            default:
                pnExceptionLevel = null;
                break;
        }
        if (pnExceptionLevel != null) {
            throw new PluginException(pnExceptionLevel, "Exception thrown from plugin");
        }
    }
    // Add nodes containing global query parameters
    final RecordStore results = new GraphRecordStore();
    results.add();
    results.set(GraphRecordStoreUtilities.SOURCE + AnalyticConcept.VertexAttribute.RAW, "name1@domain1.com");
    results.set(GraphRecordStoreUtilities.SOURCE + AnalyticConcept.VertexAttribute.TYPE, "Email");
    results.set(GraphRecordStoreUtilities.SOURCE + AnalyticConcept.VertexAttribute.COMMENT, queryName);
    results.set(GraphRecordStoreUtilities.SOURCE + TemporalConcept.VertexAttribute.LAST_SEEN, DateTimeFormatter.ISO_INSTANT.format(dtrStartEnd[0]).replace("Z", ".000Z"));
    results.set(GraphRecordStoreUtilities.DESTINATION + AnalyticConcept.VertexAttribute.RAW, "name2@domain2.com");
    results.set(GraphRecordStoreUtilities.DESTINATION + AnalyticConcept.VertexAttribute.TYPE, "Email");
    results.set(GraphRecordStoreUtilities.DESTINATION + AnalyticConcept.VertexAttribute.COMMENT, queryName);
    results.set(GraphRecordStoreUtilities.DESTINATION + TemporalConcept.VertexAttribute.LAST_SEEN, DateTimeFormatter.ISO_INSTANT.format(dtrStartEnd[1]).replace("Z", ".000Z"));
    results.set(GraphRecordStoreUtilities.TRANSACTION + AnalyticConcept.TransactionAttribute.COMMENT, parameters.toString());
    return results;
}

Example 2 with GraphRecordStore

use of au.gov.asd.tac.constellation.graph.processing.GraphRecordStore in project constellation by constellation-app.

the class TestParametersPluginNGTest method testQueryException5.

/**
 * Test of query method, of class TestParametersPlugin. Tests throwing of
 * fatal pluginException
 */
@Test(expectedExceptions = PluginException.class)
public void testQueryException5() throws Exception {
    System.out.println("throw pluginexception5");
    final TestParametersPlugin instance = new TestParametersPlugin();
    final PluginParameters result = instance.createParameters();
    final GraphRecordStore recordStore = new GraphRecordStore();
    final DefaultPluginInteraction interaction = new DefaultPluginInteraction(null, null);
    // Set plugin query name here before execution
    result.getParameters().get(CoreGlobalParameters.QUERY_NAME_PARAMETER_ID).setStringValue("TESTPARAMETERSPLUGIN");
    // Set plugin parameters here before execution
    result.getParameters().get(TestParametersPlugin.LEVEL_PARAMETER_ID).setStringValue("Fatal");
    try {
        instance.query(recordStore, interaction, result);
    } catch (final PluginException ex) {
        assertEquals(ex.getNotificationLevel(), PluginNotificationLevel.FATAL);
        throw ex;
    }
}

Example 3 with GraphRecordStore

use of au.gov.asd.tac.constellation.graph.processing.GraphRecordStore in project constellation by constellation-app.

the class TestParametersPluginNGTest method testQueryException6.

/**
 * Test of query method, of class TestParametersPlugin. Tests not throwing
 * of pluginException
 */
@Test
public void testQueryException6() throws Exception {
    System.out.println("throw pluginexception6");
    final TestParametersPlugin instance = new TestParametersPlugin();
    final PluginParameters result = instance.createParameters();
    final GraphRecordStore recordStore = new GraphRecordStore();
    final DefaultPluginInteraction interaction = new DefaultPluginInteraction(null, null);
    // Set plugin query name here before execution
    result.getParameters().get(CoreGlobalParameters.QUERY_NAME_PARAMETER_ID).setStringValue("TESTPARAMETERSPLUGIN");
    // Set plugin parameters here before execution
    result.getParameters().get(TestParametersPlugin.LEVEL_PARAMETER_ID).setStringValue("None");
    instance.query(recordStore, interaction, result);
}

Example 4 with GraphRecordStore

use of au.gov.asd.tac.constellation.graph.processing.GraphRecordStore in project constellation by constellation-app.

the class TestParametersPluginNGTest method testQueryException4.

/**
 * Test of query method, of class TestParametersPlugin. Tests throwing of
 * error pluginException
 */
@Test(expectedExceptions = PluginException.class)
public void testQueryException4() throws Exception {
    System.out.println("throw pluginexception4");
    final TestParametersPlugin instance = new TestParametersPlugin();
    final PluginParameters result = instance.createParameters();
    final GraphRecordStore recordStore = new GraphRecordStore();
    final DefaultPluginInteraction interaction = new DefaultPluginInteraction(null, null);
    // Set plugin query name here before execution
    result.getParameters().get(CoreGlobalParameters.QUERY_NAME_PARAMETER_ID).setStringValue("TESTPARAMETERSPLUGIN");
    // Set plugin parameters here before execution
    result.getParameters().get(TestParametersPlugin.LEVEL_PARAMETER_ID).setStringValue("Error");
    try {
        instance.query(recordStore, interaction, result);
    } catch (final PluginException ex) {
        assertEquals(ex.getNotificationLevel(), PluginNotificationLevel.ERROR);
        throw ex;
    }
}

Example 5 with GraphRecordStore

use of au.gov.asd.tac.constellation.graph.processing.GraphRecordStore in project constellation by constellation-app.

the class ExtractTypesFromTextPluginNGTest method testRegexQuery.

/**
 * Test of query method in class ExtractTypesFromTextPlugin using a string
 * where a detection regex is picked up
 *
 * @throws Exception
 */
@Test
public void testRegexQuery() throws Exception {
    RecordStore query = new GraphRecordStore();
    ExtractTypesFromTextPlugin instance = new ExtractTypesFromTextPlugin();
    PluginInteraction interaction = new TextPluginInteraction();
    PluginParameters parameters = instance.createParameters();
    parameters.getParameters().get(ExtractTypesFromTextPlugin.TEXT_PARAMETER_ID).setStringValue("abc@def.ghi");
    RecordStore result = instance.query(query, interaction, parameters);
    RecordStore expResult = new GraphRecordStore();
    expResult.add();
    expResult.set(GraphRecordStoreUtilities.SOURCE + VisualConcept.VertexAttribute.IDENTIFIER, "abc@def.ghi");
    expResult.set(GraphRecordStoreUtilities.SOURCE + AnalyticConcept.VertexAttribute.TYPE, AnalyticConcept.VertexType.EMAIL_ADDRESS);
    expResult.set(GraphRecordStoreUtilities.SOURCE + AnalyticConcept.VertexAttribute.SEED, "true");
    assertEquals(result, expResult);
}