Search in sources :

Example 1 with DefaultTag

use of com.thinkbiganalytics.discovery.model.DefaultTag in project kylo by Teradata.

the class FeedModelTransform method domainToFeedMetadata.

/**
 * Transforms the specified Metadata feed to a Feed Manager feed.
 *
 * @param domain       the Metadata feed
 * @param userFieldMap cache map from category to user-defined fields, or {@code null}
 * @return the Feed Manager feed
 */
@Nonnull
private FeedMetadata domainToFeedMetadata(@Nonnull final Feed domain, @Nullable final Map<Category, Set<UserFieldDescriptor>> userFieldMap) {
    FeedMetadata feed = deserializeFeedMetadata(domain, false);
    feed.setId(domain.getId().toString());
    feed.setFeedId(domain.getId().toString());
    feed.setFeedName(domain.getDisplayName());
    feed.setSystemFeedName(domain.getName());
    feed.setDescription(domain.getDescription());
    feed.setAllowIndexing(domain.isAllowIndexing());
    feed.setHistoryReindexingStatus(domain.getCurrentHistoryReindexingStatus().getHistoryReindexingState().toString());
    feed.setOwner(domain.getOwner() != null ? new User(domain.getOwner().getName()) : null);
    if (domain.getCreatedTime() != null) {
        feed.setCreateDate(domain.getCreatedTime().toDate());
    }
    if (domain.getModifiedTime() != null) {
        feed.setUpdateDate(domain.getModifiedTime().toDate());
    }
    FeedManagerTemplate template = domain.getTemplate();
    if (template != null) {
        RegisteredTemplate registeredTemplate = templateModelTransform.DOMAIN_TO_REGISTERED_TEMPLATE.apply(template);
        feed.setRegisteredTemplate(registeredTemplate);
        feed.setTemplateId(registeredTemplate.getId());
        feed.setTemplateName(registeredTemplate.getTemplateName());
    }
    Category category = domain.getCategory();
    if (category != null) {
        feed.setCategory(categoryModelTransform.domainToFeedCategorySimple(category));
    }
    feed.setState(domain.getState() != null ? domain.getState().name() : null);
    feed.setVersionName(domain.getVersionName() != null ? domain.getVersionName() : null);
    // Set user-defined properties
    final Set<UserFieldDescriptor> userFields;
    if (userFieldMap == null) {
        userFields = getUserFields(category);
    } else if (userFieldMap.containsKey(category)) {
        userFields = userFieldMap.get(category);
    } else {
        userFields = getUserFields(category);
        userFieldMap.put(category, userFields);
    }
    @SuppressWarnings("unchecked") final Set<UserProperty> userProperties = UserPropertyTransform.toUserProperties(domain.getUserProperties(), userFields);
    feed.setUserProperties(userProperties);
    // Convert JCR securitygroup to DTO
    List<com.thinkbiganalytics.feedmgr.rest.model.HadoopSecurityGroup> restSecurityGroups = new ArrayList<>();
    if (domain.getSecurityGroups() != null && domain.getSecurityGroups().size() > 0) {
        for (Object group : domain.getSecurityGroups()) {
            HadoopSecurityGroup hadoopSecurityGroup = (HadoopSecurityGroup) group;
            com.thinkbiganalytics.feedmgr.rest.model.HadoopSecurityGroup restSecurityGroup = new com.thinkbiganalytics.feedmgr.rest.model.HadoopSecurityGroup();
            restSecurityGroup.setDescription(hadoopSecurityGroup.getDescription());
            restSecurityGroup.setId(hadoopSecurityGroup.getGroupId());
            restSecurityGroup.setName(hadoopSecurityGroup.getName());
            restSecurityGroups.add(restSecurityGroup);
        }
    }
    feed.setSecurityGroups(restSecurityGroups);
    feed.setTags(domain.getTags().stream().map(name -> new DefaultTag(name)).collect(Collectors.toList()));
    if (domain.getUsedByFeeds() != null) {
        final List<FeedSummary> usedByFeeds = domain.getUsedByFeeds().stream().map(this::domainToFeedSummary).collect(Collectors.toList());
        feed.setUsedByFeeds(usedByFeeds);
    }
    // add in access control items
    securityTransform.applyAccessControl(domain, feed);
    return feed;
}
Also used : User(com.thinkbiganalytics.security.rest.model.User) Category(com.thinkbiganalytics.metadata.api.category.Category) FeedCategory(com.thinkbiganalytics.feedmgr.rest.model.FeedCategory) UserFieldDescriptor(com.thinkbiganalytics.metadata.api.extension.UserFieldDescriptor) FeedMetadata(com.thinkbiganalytics.feedmgr.rest.model.FeedMetadata) ArrayList(java.util.ArrayList) HadoopSecurityGroup(com.thinkbiganalytics.metadata.api.security.HadoopSecurityGroup) JcrHadoopSecurityGroup(com.thinkbiganalytics.metadata.modeshape.security.JcrHadoopSecurityGroup) UserProperty(com.thinkbiganalytics.feedmgr.rest.model.UserProperty) RegisteredTemplate(com.thinkbiganalytics.feedmgr.rest.model.RegisteredTemplate) FeedManagerTemplate(com.thinkbiganalytics.metadata.api.template.FeedManagerTemplate) DefaultTag(com.thinkbiganalytics.discovery.model.DefaultTag) FeedSummary(com.thinkbiganalytics.feedmgr.rest.model.FeedSummary) Nonnull(javax.annotation.Nonnull)

Example 2 with DefaultTag

use of com.thinkbiganalytics.discovery.model.DefaultTag in project kylo by Teradata.

the class FeedModelTransform method domainToFeedMetadata.

/**
 * Transforms the specified Metadata feed to a Feed Manager feed.
 *
 * @param domain       the Metadata feed
 * @param userFieldMap cache map from category to user-defined fields, or {@code null}
 * @return the Feed Manager feed
 */
@Nonnull
private FeedMetadata domainToFeedMetadata(@Nonnull final Feed domain, @Nullable final Map<Category, Set<UserFieldDescriptor>> userFieldMap, @Nullable ActionGroup actionGroup) {
    FeedMetadata feed = deserializeFeedMetadata(domain, false);
    feed.setId(domain.getId().toString());
    feed.setFeedId(domain.getId().toString());
    feed.setFeedName(domain.getDisplayName());
    feed.setSystemFeedName(domain.getName());
    feed.setDescription(domain.getDescription());
    feed.setAllowIndexing(domain.isAllowIndexing());
    feed.setHistoryReindexingStatus(domain.getCurrentHistoryReindexingStatus().getHistoryReindexingState().toString());
    feed.setOwner(domain.getOwner() != null ? new User(domain.getOwner().getName()) : null);
    if (domain.getCreatedTime() != null) {
        feed.setCreateDate(domain.getCreatedTime().toDate());
    }
    if (domain.getModifiedTime() != null) {
        feed.setUpdateDate(domain.getModifiedTime().toDate());
    }
    setFeedMetadataRegisteredTemplate(domain, feed);
    Category category = domain.getCategory();
    if (category != null) {
        feed.setCategory(categoryModelTransform.domainToFeedCategorySimple(category));
    }
    feed.setState(domain.getState() != null ? domain.getState().name() : null);
    // Set user-defined properties
    final Set<UserFieldDescriptor> userFields;
    if (userFieldMap == null) {
        userFields = getUserFields(category);
    } else if (userFieldMap.containsKey(category)) {
        userFields = userFieldMap.get(category);
    } else {
        userFields = getUserFields(category);
        userFieldMap.put(category, userFields);
    }
    @SuppressWarnings("unchecked") final Set<UserProperty> userProperties = UserPropertyTransform.toUserProperties(domain.getUserProperties(), userFields);
    feed.setUserProperties(userProperties);
    // Convert JCR securitygroup to DTO
    List<com.thinkbiganalytics.feedmgr.rest.model.HadoopSecurityGroup> restSecurityGroups = new ArrayList<>();
    if (domain.getSecurityGroups() != null && domain.getSecurityGroups().size() > 0) {
        for (Object group : domain.getSecurityGroups()) {
            HadoopSecurityGroup hadoopSecurityGroup = (HadoopSecurityGroup) group;
            com.thinkbiganalytics.feedmgr.rest.model.HadoopSecurityGroup restSecurityGroup = new com.thinkbiganalytics.feedmgr.rest.model.HadoopSecurityGroup();
            restSecurityGroup.setDescription(hadoopSecurityGroup.getDescription());
            restSecurityGroup.setId(hadoopSecurityGroup.getGroupId());
            restSecurityGroup.setName(hadoopSecurityGroup.getName());
            restSecurityGroups.add(restSecurityGroup);
        }
    }
    feed.setSecurityGroups(restSecurityGroups);
    feed.setTags(domain.getTags().stream().map(name -> new DefaultTag(name)).collect(Collectors.toList()));
    if (domain.getUsedByFeeds() != null) {
        final List<FeedSummary> usedByFeeds = domain.getUsedByFeeds().stream().map(this::domainToFeedSummary).collect(Collectors.toList());
        feed.setUsedByFeeds(usedByFeeds);
    }
    List<DataSet> srcDataSets = domain.getSources().stream().map(FeedConnection::getDataSet).filter(Optional::isPresent).map(Optional::get).map(dataSet -> {
        try {
            return catalogModelTransform.dataSetToRestModel().apply(dataSet);
        } catch (final AccessControlException e) {
            log.debug("Denied access to data set: {}: {}", domain.getId(), e);
            final DataSource dataSource = new DataSource();
            dataSource.setAllowedActions(new ActionGroup());
            dataSource.getAllowedActions().setActions(Collections.emptyList());
            final DataSet model = new DataSet();
            model.setId(domain.getId().toString());
            model.setDataSource(dataSource);
            return model;
        }
    }).collect(Collectors.toList());
    feed.setSourceDataSets(srcDataSets);
    if (actionGroup == null) {
        // add in access control items
        securityTransform.applyAccessControl(domain, feed);
    } else {
        feed.setAllowedActions(actionGroup);
    }
    return feed;
}
Also used : Category(com.thinkbiganalytics.metadata.api.category.Category) DataSourceProvider(com.thinkbiganalytics.metadata.api.catalog.DataSourceProvider) LoggerFactory(org.slf4j.LoggerFactory) StringUtils(org.apache.commons.lang3.StringUtils) ObjectReader(com.fasterxml.jackson.databind.ObjectReader) ObjectMapperSerializer(com.thinkbiganalytics.json.ObjectMapperSerializer) FeedProvider(com.thinkbiganalytics.metadata.api.feed.FeedProvider) DataSetProvider(com.thinkbiganalytics.metadata.api.catalog.DataSetProvider) HiveService(com.thinkbiganalytics.hive.service.HiveService) DataSource(com.thinkbiganalytics.kylo.catalog.rest.model.DataSource) Map(java.util.Map) AccessController(com.thinkbiganalytics.security.AccessController) JsonNode(com.fasterxml.jackson.databind.JsonNode) Pageable(org.springframework.data.domain.Pageable) Tag(com.thinkbiganalytics.discovery.schema.Tag) CatalogModelTransform(com.thinkbiganalytics.kylo.catalog.rest.model.CatalogModelTransform) FeedManagerTemplateProvider(com.thinkbiganalytics.metadata.api.template.FeedManagerTemplateProvider) DataSet(com.thinkbiganalytics.kylo.catalog.rest.model.DataSet) FeedVersions(com.thinkbiganalytics.feedmgr.rest.model.FeedVersions) ObjectWriter(com.fasterxml.jackson.databind.ObjectWriter) Collection(java.util.Collection) ActionGroup(com.thinkbiganalytics.security.rest.model.ActionGroup) Set(java.util.Set) DefaultTag(com.thinkbiganalytics.discovery.model.DefaultTag) Page(org.springframework.data.domain.Page) EntityVersionDifference(com.thinkbiganalytics.feedmgr.rest.model.EntityVersionDifference) Collectors(java.util.stream.Collectors) Sets(com.google.common.collect.Sets) FeedSummary(com.thinkbiganalytics.feedmgr.rest.model.FeedSummary) List(java.util.List) CategoryProvider(com.thinkbiganalytics.metadata.api.category.CategoryProvider) AccessControlException(java.security.AccessControlException) Optional(java.util.Optional) PageImpl(org.springframework.data.domain.PageImpl) JsonDiff(com.flipkart.zjsonpatch.JsonDiff) SecurityModelTransform(com.thinkbiganalytics.security.rest.controller.SecurityModelTransform) SearchResultImpl(com.thinkbiganalytics.rest.model.search.SearchResultImpl) Feed(com.thinkbiganalytics.metadata.api.feed.Feed) HistoryReindexingStatus(com.thinkbiganalytics.metadata.api.feed.reindex.HistoryReindexingStatus) FeedMetadata(com.thinkbiganalytics.feedmgr.rest.model.FeedMetadata) EntityVersion(com.thinkbiganalytics.metadata.api.versioning.EntityVersion) UserProperty(com.thinkbiganalytics.feedmgr.rest.model.UserProperty) ArrayList(java.util.ArrayList) FeedCategory(com.thinkbiganalytics.feedmgr.rest.model.FeedCategory) Inject(javax.inject.Inject) FeedConnection(com.thinkbiganalytics.metadata.api.feed.FeedConnection) UIFeed(com.thinkbiganalytics.feedmgr.rest.model.UIFeed) TemplateModelTransform(com.thinkbiganalytics.feedmgr.service.template.TemplateModelTransform) User(com.thinkbiganalytics.security.rest.model.User) RegisteredTemplate(com.thinkbiganalytics.feedmgr.rest.model.RegisteredTemplate) Nonnull(javax.annotation.Nonnull) Nullable(javax.annotation.Nullable) Logger(org.slf4j.Logger) ObjectMapper(com.fasterxml.jackson.databind.ObjectMapper) IOException(java.io.IOException) UserFieldDescriptor(com.thinkbiganalytics.metadata.api.extension.UserFieldDescriptor) Maps(com.google.common.collect.Maps) SearchResult(com.thinkbiganalytics.rest.model.search.SearchResult) FeedManagerTemplate(com.thinkbiganalytics.metadata.api.template.FeedManagerTemplate) HadoopSecurityGroup(com.thinkbiganalytics.metadata.api.security.HadoopSecurityGroup) EncryptionService(com.thinkbiganalytics.security.core.encrypt.EncryptionService) CategoryModelTransform(com.thinkbiganalytics.feedmgr.service.category.CategoryModelTransform) UserPropertyTransform(com.thinkbiganalytics.feedmgr.service.UserPropertyTransform) DatasourceAccessControl(com.thinkbiganalytics.metadata.api.datasource.security.DatasourceAccessControl) JcrHadoopSecurityGroup(com.thinkbiganalytics.metadata.modeshape.security.JcrHadoopSecurityGroup) HadoopSecurityGroupProvider(com.thinkbiganalytics.metadata.api.security.HadoopSecurityGroupProvider) Collections(java.util.Collections) User(com.thinkbiganalytics.security.rest.model.User) Category(com.thinkbiganalytics.metadata.api.category.Category) FeedCategory(com.thinkbiganalytics.feedmgr.rest.model.FeedCategory) DataSet(com.thinkbiganalytics.kylo.catalog.rest.model.DataSet) FeedMetadata(com.thinkbiganalytics.feedmgr.rest.model.FeedMetadata) ArrayList(java.util.ArrayList) HadoopSecurityGroup(com.thinkbiganalytics.metadata.api.security.HadoopSecurityGroup) JcrHadoopSecurityGroup(com.thinkbiganalytics.metadata.modeshape.security.JcrHadoopSecurityGroup) Optional(java.util.Optional) UserFieldDescriptor(com.thinkbiganalytics.metadata.api.extension.UserFieldDescriptor) FeedConnection(com.thinkbiganalytics.metadata.api.feed.FeedConnection) AccessControlException(java.security.AccessControlException) DataSource(com.thinkbiganalytics.kylo.catalog.rest.model.DataSource) UserProperty(com.thinkbiganalytics.feedmgr.rest.model.UserProperty) ActionGroup(com.thinkbiganalytics.security.rest.model.ActionGroup) DefaultTag(com.thinkbiganalytics.discovery.model.DefaultTag) FeedSummary(com.thinkbiganalytics.feedmgr.rest.model.FeedSummary) Nonnull(javax.annotation.Nonnull)

Example 3 with DefaultTag

use of com.thinkbiganalytics.discovery.model.DefaultTag in project kylo by Teradata.

the class AvroFeedIT method getTags.

protected List<Tag> getTags() {
    List<Tag> tags = new ArrayList<>();
    tags.add(new DefaultTag("users"));
    tags.add(new DefaultTag("registrations"));
    return tags;
}
Also used : ArrayList(java.util.ArrayList) Tag(com.thinkbiganalytics.discovery.schema.Tag) DefaultTag(com.thinkbiganalytics.discovery.model.DefaultTag) DefaultTag(com.thinkbiganalytics.discovery.model.DefaultTag)

Example 4 with DefaultTag

use of com.thinkbiganalytics.discovery.model.DefaultTag in project kylo by Teradata.

the class JsonFeedIT method getTags.

protected List<Tag> getTags() {
    List<Tag> tags = new ArrayList<>();
    tags.add(new DefaultTag("books"));
    return tags;
}
Also used : DefaultTag(com.thinkbiganalytics.discovery.model.DefaultTag) Tag(com.thinkbiganalytics.discovery.schema.Tag) DefaultTag(com.thinkbiganalytics.discovery.model.DefaultTag)

Example 5 with DefaultTag

use of com.thinkbiganalytics.discovery.model.DefaultTag in project kylo by Teradata.

the class DomainTypeIT method testCreateAndUpdateDatasource.

@Test
public void testCreateAndUpdateDatasource() {
    FieldStandardizationRule toUpperCase = new FieldStandardizationRule();
    FieldValidationRule email = new FieldValidationRule();
    toUpperCase.setName("Uppercase");
    toUpperCase.setDisplayName("Uppercase");
    toUpperCase.setDescription("Convert string to uppercase");
    toUpperCase.setObjectClassType("com.thinkbiganalytics.policy.standardization.UppercaseStandardizer");
    toUpperCase.setObjectShortClassType("UppercaseStandardizer");
    email.setName("email");
    email.setDisplayName("Email");
    email.setDescription("Valid email address");
    email.setObjectClassType("com.thinkbiganalytics.policy.validation.EmailValidator");
    email.setObjectShortClassType("EmailValidator");
    DomainType[] initialDomainTypes = getDomainTypes();
    // create new domain type
    DomainType dt = new DomainType();
    dt.setTitle("Domain Type 1");
    dt.setDescription("domain type created by integration tests");
    DomainType response = createDomainType(dt);
    assertEquals(dt.getTitle(), response.getTitle());
    assertEquals(dt.getDescription(), response.getDescription());
    assertEquals(null, response.getIcon());
    assertEquals(null, response.getIconColor());
    // assert new domain type was added
    DomainType[] currentDomainTypes = getDomainTypes();
    assertEquals(initialDomainTypes.length + 1, currentDomainTypes.length);
    // update existing domain type
    dt = getDomainType(response.getId());
    dt.setTitle("Domain Type 1 with updated title");
    dt.setDescription("domain type description updated by integration tests");
    dt.setIcon("stars");
    dt.setIconColor("green");
    DefaultField field = new DefaultField();
    Tag tag1 = new DefaultTag("tag1");
    Tag tag2 = new DefaultTag("tag2");
    field.setTags(Arrays.asList(tag1, tag2));
    field.setName("field-name");
    field.setDerivedDataType("decimal");
    field.setPrecisionScale("9, 3");
    dt.setField(field);
    dt.setFieldNamePattern("field-name-pattern");
    dt.setFieldPolicy(newPolicyBuilder(null).withStandardisation(toUpperCase).withValidation(email).toPolicy());
    dt.setRegexPattern("regex-pattern");
    DomainType updated = createDomainType(dt);
    assertEquals(dt.getTitle(), updated.getTitle());
    assertEquals(dt.getDescription(), updated.getDescription());
    assertEquals("stars", updated.getIcon());
    assertEquals("green", updated.getIconColor());
    Field updatedField = updated.getField();
    assertEquals(field.getName(), updatedField.getName());
    assertEquals(field.getDerivedDataType(), updatedField.getDerivedDataType());
    assertEquals(field.getPrecisionScale(), updatedField.getPrecisionScale());
    assertEquals(field.getTags().size(), updatedField.getTags().size());
    assertEquals(tag1.getName(), updatedField.getTags().get(0).getName());
    assertEquals(tag2.getName(), updatedField.getTags().get(1).getName());
    assertEquals(dt.getFieldNamePattern(), updated.getFieldNamePattern());
    assertEquals(dt.getRegexPattern(), updated.getRegexPattern());
    FieldStandardizationRule updatedStandardisation = updated.getFieldPolicy().getStandardization().get(0);
    assertEquals(toUpperCase.getName(), updatedStandardisation.getName());
    assertEquals(toUpperCase.getObjectShortClassType(), updatedStandardisation.getObjectShortClassType());
    FieldValidationRule updatedValidation = updated.getFieldPolicy().getValidation().get(0);
    assertEquals(email.getName(), updatedValidation.getName());
    assertEquals(email.getObjectShortClassType(), updatedValidation.getObjectShortClassType());
    // assert domain type was updated, rather than added
    currentDomainTypes = getDomainTypes();
    assertEquals(initialDomainTypes.length + 1, currentDomainTypes.length);
    // delete domain type
    deleteDomainType(dt.getId());
    currentDomainTypes = getDomainTypes();
    assertEquals(initialDomainTypes.length, currentDomainTypes.length);
    // assert domain type was removed
    getDomainTypeExpectingStatus(dt.getId(), HTTP_NOT_FOUND);
}
Also used : DefaultField(com.thinkbiganalytics.discovery.model.DefaultField) Field(com.thinkbiganalytics.discovery.schema.Field) DomainType(com.thinkbiganalytics.feedmgr.rest.model.DomainType) FieldValidationRule(com.thinkbiganalytics.policy.rest.model.FieldValidationRule) DefaultField(com.thinkbiganalytics.discovery.model.DefaultField) DefaultTag(com.thinkbiganalytics.discovery.model.DefaultTag) Tag(com.thinkbiganalytics.discovery.schema.Tag) DefaultTag(com.thinkbiganalytics.discovery.model.DefaultTag) FieldStandardizationRule(com.thinkbiganalytics.policy.rest.model.FieldStandardizationRule) Test(org.junit.Test)

Aggregations

DefaultTag (com.thinkbiganalytics.discovery.model.DefaultTag)11 Tag (com.thinkbiganalytics.discovery.schema.Tag)8 FeedMetadata (com.thinkbiganalytics.feedmgr.rest.model.FeedMetadata)7 ArrayList (java.util.ArrayList)6 User (com.thinkbiganalytics.security.rest.model.User)5 FeedCategory (com.thinkbiganalytics.feedmgr.rest.model.FeedCategory)4 NifiProperty (com.thinkbiganalytics.nifi.rest.model.NifiProperty)4 JsonNode (com.fasterxml.jackson.databind.JsonNode)3 EntityVersionDifference (com.thinkbiganalytics.feedmgr.rest.model.EntityVersionDifference)3 FeedSchedule (com.thinkbiganalytics.feedmgr.rest.model.FeedSchedule)3 FeedVersions (com.thinkbiganalytics.feedmgr.rest.model.FeedVersions)3 FieldPolicy (com.thinkbiganalytics.policy.rest.model.FieldPolicy)3 HashSet (java.util.HashSet)3 ArrayNode (com.fasterxml.jackson.databind.node.ArrayNode)2 Field (com.thinkbiganalytics.discovery.schema.Field)2 EntityDifference (com.thinkbiganalytics.feedmgr.rest.model.EntityDifference)2 EntityVersion (com.thinkbiganalytics.feedmgr.rest.model.EntityVersion)2 FeedSummary (com.thinkbiganalytics.feedmgr.rest.model.FeedSummary)2 RegisteredTemplate (com.thinkbiganalytics.feedmgr.rest.model.RegisteredTemplate)2 UserProperty (com.thinkbiganalytics.feedmgr.rest.model.UserProperty)2