FilesExpand file tree

 master

/

DirectivesUtil.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

122 lines (98 loc) · 4.75 KB

 master

/

DirectivesUtil.java

Copy path

Top

File metadata and controls

• Code
• Blame

122 lines (98 loc) · 4.75 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

package graphql;

import com.google.common.collect.ImmutableList;

import com.google.common.collect.ImmutableMap;

import graphql.schema.GraphQLArgument;

import graphql.schema.GraphQLDirective;

import graphql.util.FpKit;

import java.util.Collection;

import java.util.List;

import java.util.Map;

import java.util.Optional;

import java.util.stream.Collectors;

import static graphql.Assert.assertNotNull;

import static graphql.collect.ImmutableKit.emptyList;

@Internal

public class DirectivesUtil {

public static Map<String, GraphQLDirective> nonRepeatableDirectivesByName(List<GraphQLDirective> directives) {

// filter the repeatable directives

List<GraphQLDirective> singletonDirectives = directives.stream()

.filter(d -> !d.isRepeatable()).collect(Collectors.toList());

return FpKit.getByName(singletonDirectives, GraphQLDirective::getName);

}

public static Map<String, ImmutableList<GraphQLDirective>> allDirectivesByName(List<GraphQLDirective> directives) {

return ImmutableMap.copyOf(FpKit.groupingBy(directives, GraphQLDirective::getName));

}

public static Optional<GraphQLArgument> directiveWithArg(List<GraphQLDirective> directives, String directiveName, String argumentName) {

GraphQLDirective directive = nonRepeatableDirectivesByName(directives).get(directiveName);

GraphQLArgument argument = null;

if (directive != null) {

argument = directive.getArgument(argumentName);

}

return Optional.ofNullable(argument);

}

public static boolean isAllNonRepeatable(List<GraphQLDirective> directives) {

if (directives == null || directives.isEmpty()) {

return false;

}

for (GraphQLDirective graphQLDirective : directives) {

if (graphQLDirective.isRepeatable()) {

return false;

}

}

return true;

}

public static List<GraphQLDirective> add(List<GraphQLDirective> targetList, GraphQLDirective newDirective) {

assertNotNull(targetList, () -> "directive list can't be null");

assertNotNull(newDirective, () -> "directive can't be null");

targetList.add(newDirective);

return targetList;

}

public static List<GraphQLDirective> addAll(List<GraphQLDirective> targetList, List<GraphQLDirective> newDirectives) {

assertNotNull(targetList, () -> "directive list can't be null");

assertNotNull(newDirectives, () -> "directive list can't be null");

targetList.addAll(newDirectives);

return targetList;

}

public static GraphQLDirective getFirstDirective(String name, Map<String, List<GraphQLDirective>> allDirectivesByName) {

List<GraphQLDirective> directives = allDirectivesByName.getOrDefault(name, emptyList());

if (directives.isEmpty()) {

return null;

}

return directives.get(0);

}

/**

* A holder class that breaks a list of directives into maps to be more easily accessible in using classes

*/

public static class DirectivesHolder {

private final ImmutableMap<String, List<GraphQLDirective>> allDirectivesByName;

private final ImmutableMap<String, GraphQLDirective> nonRepeatableDirectivesByName;

private final List<GraphQLDirective> allDirectives;

public DirectivesHolder(Collection<GraphQLDirective> allDirectives) {

this.allDirectives = ImmutableList.copyOf(allDirectives);

this.allDirectivesByName = ImmutableMap.copyOf(FpKit.groupingBy(allDirectives, GraphQLDirective::getName));

// filter out the repeatable directives

List<GraphQLDirective> nonRepeatableDirectives = allDirectives.stream()

.filter(d -> !d.isRepeatable()).collect(Collectors.toList());

this.nonRepeatableDirectivesByName = ImmutableMap.copyOf(FpKit.getByName(nonRepeatableDirectives, GraphQLDirective::getName));

}

public ImmutableMap<String, List<GraphQLDirective>> getAllDirectivesByName() {

return allDirectivesByName;

}

public ImmutableMap<String, GraphQLDirective> getDirectivesByName() {

return nonRepeatableDirectivesByName;

}

public List<GraphQLDirective> getDirectives() {

return allDirectives;

}

public GraphQLDirective getDirective(String directiveName) {

List<GraphQLDirective> directiveList = allDirectivesByName.get(directiveName);

if (directiveList == null || directiveList.isEmpty()) {

return null;

}

return directiveList.get(0);

}

public List<GraphQLDirective> getDirectives(String directiveName) {

return allDirectivesByName.getOrDefault(directiveName, emptyList());

}

}

}