Elgato_dark/zeppelin
1
0
Fork 0
mirror of https://github.com/apache/zeppelin synced 2026-05-24 09:38:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

filter reload using predicates

Browse source
This commit is contained in:
Khalid Huseynov 2016-09-17 17:22:48 +09:00
parent 92f37f50c3
commit d9c3bc98cb
1 changed files with 11 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
zeppelin-zengine/src/main/java/org/apache/zeppelin/notebook/Notebook.java
View file

@ -485,47 +485,24 @@ public class Notebook implements NoteEventListener {
}
List<NoteInfo> noteInfos = notebookRepo.list(subject);
noteInfos = getAuthorizedNoteInfos(noteInfos, subject);
noteInfos = filterByUser(noteInfos, subject);
for (NoteInfo info : noteInfos) {
loadNoteFromRepo(info.getId(), subject);
}
}
private List<NoteInfo> getAuthorizedNoteInfos(List<NoteInfo> notes, AuthenticationInfo subject) {
Set<String> allIds = Sets.newHashSet();
if (subject == null) {
logger.warn("Subject for retrieving notes is null");
return notes;
private List<NoteInfo> filterByUser(List<NoteInfo> notes, AuthenticationInfo subject) {
final Set<String> entities = Sets.newHashSet();
if (subject != null) {
entities.add(subject.getUser());
}
if ("anonymous".equals(subject.getUser())) {
return notes;
}
for (NoteInfo note: notes) {
allIds.add(note.getId());
}
Set<String> filteredIds = applyAuthorizationFilter(allIds, subject.getUser());
List<NoteInfo> filteredNotes = Lists.newArrayList();
for (NoteInfo note: notes) {
if (filteredIds.contains(note.getId())){
filteredNotes.add(note);
return FluentIterable.from(notes).filter(new Predicate<NoteInfo>() {
@Override
public boolean apply(NoteInfo input) {
return input != null && notebookAuthorization.isReader(input.getId(), entities);
}
}
return filteredNotes;
}
public Set<String> applyAuthorizationFilter(Set<String> ids, String user) {
logger.info("applying filter for {}", user);
Set<String> filteredIds = Sets.newHashSet();
Set<String> userEntity = Sets.newHashSet((Arrays.asList(user)));
NotebookAuthorization auth = getNotebookAuthorization();
for (String id: ids) {
if (auth.isOwner(id, userEntity) || auth.isReader(id, userEntity)
|| auth.isWriter(id, userEntity)) {
filteredIds.add(id);
}
}
return filteredIds;
}).toList();
}
private class SnapshotAngularObject {