com.structurizr.dsl.RelationshipsDslContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of structurizr-dsl Show documentation

Structurizr for Java

The newest version!

package com.structurizr.dsl;

import com.structurizr.model.ModelItem;
import com.structurizr.model.Relationship;

import java.util.Set;
import java.util.stream.Collectors;

class RelationshipsDslContext extends ModelItemsDslContext {

    private final Set relationships;

    RelationshipsDslContext(DslContext parentDslContext, Set relationships) {
        super(parentDslContext);
        this.relationships = relationships;
    }

    Set getRelationships() {
        return relationships;
    }

    @Override
    Set getModelItems() {
        return relationships.stream().map(e -> (ModelItem)e).collect(Collectors.toSet());
    }

    @Override
    protected String[] getPermittedTokens() {
        return new String[] {
                StructurizrDslTokens.TAG_TOKEN,
                StructurizrDslTokens.TAGS_TOKEN,
                StructurizrDslTokens.URL_TOKEN,
                StructurizrDslTokens.PROPERTIES_TOKEN,
                StructurizrDslTokens.PERSPECTIVES_TOKEN
        };
    }

}