-
Notifications
You must be signed in to change notification settings - Fork 1k
/
SearchDbService.java
143 lines (121 loc) · 5.41 KB
/
SearchDbService.java
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
package org.graylog.plugins.enterprise.search.db;
import com.google.common.collect.Streams;
import com.mongodb.BasicDBObject;
import org.bson.types.ObjectId;
import org.graylog.plugins.enterprise.search.Search;
import org.graylog.plugins.enterprise.search.SearchRequirements;
import org.graylog.plugins.enterprise.search.views.ViewService;
import org.graylog.plugins.enterprise.search.views.sharing.IsViewSharedForUser;
import org.graylog.plugins.enterprise.search.views.sharing.ViewSharingService;
import org.graylog2.bindings.providers.MongoJackObjectMapperProvider;
import org.graylog2.database.MongoConnection;
import org.graylog2.database.PaginatedList;
import org.graylog2.plugin.database.users.User;
import org.mongojack.DBCursor;
import org.mongojack.DBQuery;
import org.mongojack.DBSort;
import org.mongojack.JacksonDBCollection;
import org.mongojack.WriteResult;
import javax.inject.Inject;
import java.util.Collection;
import java.util.Optional;
import java.util.Set;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import java.util.stream.Stream;
/**
* This class is a helper to implement a basic Mongojack-based database service that allows CRUD operations on a single DTO type.
*
* <p>
* Subclasses can add more sophisticated search methods by access the protected "db" property.<br/>
* Indices can be added in the constructor.
* </p>
*/
public class SearchDbService {
protected final JacksonDBCollection<Search, ObjectId> db;
private final ViewService viewService;
private final ViewSharingService viewSharingService;
private final IsViewSharedForUser isViewSharedForUser;
private final SearchRequirements.Factory searchRequirementsFactory;
@Inject
protected SearchDbService(MongoConnection mongoConnection,
MongoJackObjectMapperProvider mapper,
ViewService viewService,
ViewSharingService viewSharingService,
IsViewSharedForUser isViewSharedForUser,
SearchRequirements.Factory searchRequirementsFactory) {
this.viewService = viewService;
this.viewSharingService = viewSharingService;
this.isViewSharedForUser = isViewSharedForUser;
this.searchRequirementsFactory = searchRequirementsFactory;
db = JacksonDBCollection.wrap(mongoConnection.getDatabase().getCollection("searches"),
Search.class,
ObjectId.class,
mapper.get());
db.createIndex(new BasicDBObject("created_at", 1), new BasicDBObject("unique", false));
}
public Optional<Search> get(String id) {
return Optional.ofNullable(db.findOneById(new ObjectId(id)))
.map(this::requirementsForSearch);
}
public Optional<Search> getForUser(String id, User user, Predicate<String> permissionChecker) {
final Optional<Search> search = get(id).map(this::requirementsForSearch);
if (!search.isPresent()) {
return Optional.empty();
}
if (search.map(s -> s.owner().map(owner -> owner.equals(user.getName())).orElse(false)).orElse(false)) {
return search;
}
if (viewService.forSearch(id).stream()
.map(view -> viewSharingService.forView(view.id()))
.anyMatch(viewSharing -> viewSharing.map(sharing -> isViewSharedForUser.isAllowedToSee(user, sharing)).orElse(false))) {
return search;
}
if (viewService.forSearch(id).stream()
.anyMatch(view -> permissionChecker.test(view.id()))) {
return search;
}
return Optional.empty();
}
public Search save(Search search) {
final Search searchToSave = requirementsForSearch(search);
if (searchToSave.id() != null) {
db.update(
DBQuery.is("_id", search.id()),
searchToSave,
true,
false
);
return searchToSave;
}
final WriteResult<Search, ObjectId> save = db.insert(searchToSave);
return save.getSavedObject();
}
public PaginatedList<Search> findPaginated(DBQuery.Query search, DBSort.SortBuilder sort, int page, int perPage) {
final DBCursor<Search> cursor = db.find(search)
.sort(sort)
.limit(perPage)
.skip(perPage * Math.max(0, page - 1));
return new PaginatedList<>(
Streams.stream((Iterable<Search>) cursor).map(this::requirementsForSearch).collect(Collectors.toList()),
cursor.count(),
page,
perPage
);
}
public void delete(String id) {
db.removeById(new ObjectId(id));
}
public Collection<Search> findByIds(Set<String> idSet) {
return Streams.stream((Iterable<Search>) db.find(DBQuery.in("_id", idSet.stream().map(ObjectId::new).collect(Collectors.toList()))))
.map(this::requirementsForSearch)
.collect(Collectors.toList());
}
public Stream<Search> streamAll() {
return Streams.stream((Iterable<Search>) db.find()).map(this::requirementsForSearch);
}
private Search requirementsForSearch(Search search) {
return searchRequirementsFactory.create(search)
.rebuildRequirements(Search::requires, (s, newRequirements) -> s.toBuilder().requires(newRequirements).build());
}
}