-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
4b9a445
commit 0a8e501
Showing
21 changed files
with
1,249 additions
and
163 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
209 changes: 209 additions & 0 deletions
209
src/main/java/org/phoebus/olog/LogTemplateRepository.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,209 @@ | ||
| /* | ||
| * Copyright (c) 2010-2020 Brookhaven National Laboratory | ||
| * Copyright (c) 2010-2020 Helmholtz-Zentrum Berlin für Materialien und Energie GmbH | ||
| * All rights reserved. Use is subject to license terms and conditions. | ||
| */ | ||
| package org.phoebus.olog; | ||
|
|
||
| import co.elastic.clients.elasticsearch.ElasticsearchClient; | ||
| import co.elastic.clients.elasticsearch._types.FieldSort; | ||
| import co.elastic.clients.elasticsearch._types.Refresh; | ||
| import co.elastic.clients.elasticsearch._types.Result; | ||
| import co.elastic.clients.elasticsearch._types.SortOptions; | ||
| import co.elastic.clients.elasticsearch._types.query_dsl.MatchAllQuery; | ||
| import co.elastic.clients.elasticsearch.core.DeleteRequest; | ||
| import co.elastic.clients.elasticsearch.core.DeleteResponse; | ||
| import co.elastic.clients.elasticsearch.core.GetRequest; | ||
| import co.elastic.clients.elasticsearch.core.GetResponse; | ||
| import co.elastic.clients.elasticsearch.core.IndexRequest; | ||
| import co.elastic.clients.elasticsearch.core.IndexResponse; | ||
| import co.elastic.clients.elasticsearch.core.SearchRequest; | ||
| import co.elastic.clients.elasticsearch.core.SearchResponse; | ||
| import co.elastic.clients.elasticsearch.core.search.Hit; | ||
| import org.phoebus.olog.entity.Log; | ||
| import org.phoebus.olog.entity.LogTemplate; | ||
| import org.springframework.beans.factory.annotation.Autowired; | ||
| import org.springframework.beans.factory.annotation.Qualifier; | ||
| import org.springframework.data.repository.CrudRepository; | ||
| import org.springframework.http.HttpStatus; | ||
| import org.springframework.lang.NonNull; | ||
| import org.springframework.stereotype.Repository; | ||
| import org.springframework.web.server.ResponseStatusException; | ||
|
|
||
| import java.io.IOException; | ||
| import java.text.MessageFormat; | ||
| import java.time.Instant; | ||
| import java.util.Optional; | ||
| import java.util.logging.Level; | ||
| import java.util.logging.Logger; | ||
| import java.util.stream.Collectors; | ||
|
|
||
|
|
||
| @Repository | ||
| public class LogTemplateRepository implements CrudRepository<LogTemplate, String> { | ||
|
|
||
| private static final Logger logger = Logger.getLogger(LogTemplateRepository.class.getName()); | ||
|
|
||
| @SuppressWarnings("unused") | ||
| @Autowired | ||
| @Qualifier("client") | ||
| ElasticsearchClient client; | ||
|
|
||
| @Autowired | ||
| SequenceGenerator generator; | ||
|
|
||
| @Override | ||
| public <S extends LogTemplate> S save(S logTemplate) { | ||
|
|
||
| try { | ||
| Long id = generator.getID(); | ||
| logTemplate.setCreatedDate(Instant.now()); | ||
| logTemplate.setId(id); | ||
| IndexRequest<Object> indexRequest = | ||
| IndexRequest.of(i -> | ||
| i.index(ElasticConfig.ES_LOG_TEMPLATE_INDEX) | ||
| .id(String.valueOf(id)) | ||
| .document(logTemplate) | ||
| .refresh(Refresh.True)); | ||
| IndexResponse response = client.index(indexRequest); | ||
|
|
||
| if (response.result().equals(Result.Created)) { | ||
| GetRequest getRequest = | ||
| GetRequest.of(g -> | ||
| g.index(ElasticConfig.ES_LOG_TEMPLATE_INDEX).id(response.id())); | ||
| GetResponse<LogTemplate> resp = | ||
| client.get(getRequest, LogTemplate.class); | ||
| return (S) resp.source(); | ||
| } | ||
| } catch (Exception e) { | ||
| String message = MessageFormat.format(TextUtil.LOG_TEMPLATE_NOT_SAVED, logTemplate.getName()); | ||
| logger.log(Level.SEVERE, message, e); | ||
| throw new ResponseStatusException(HttpStatus.INTERNAL_SERVER_ERROR, message); | ||
| } | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public <S extends LogTemplate> Iterable<S> saveAll(Iterable<S> logTemplates) { | ||
| throw new ResponseStatusException(HttpStatus.NOT_FOUND, TextUtil.LOG_TEMPLATES_SAVE_ALL); | ||
| } | ||
|
|
||
| public LogTemplate update(LogTemplate logTemplate) { | ||
| try { | ||
| logTemplate.setModifyDate(Instant.now()); | ||
| IndexRequest<Log> indexRequest = | ||
| IndexRequest.of(i -> | ||
| i.index(ElasticConfig.ES_LOG_TEMPLATE_INDEX) | ||
| .id(String.valueOf(logTemplate.getId())) | ||
| .refresh(Refresh.True) | ||
| .document(logTemplate)); | ||
|
|
||
| IndexResponse response = client.index(indexRequest); | ||
|
|
||
| if (response.result().equals(Result.Updated)) { | ||
| GetRequest getRequest = | ||
| GetRequest.of(g -> | ||
| g.index(ElasticConfig.ES_LOG_TEMPLATE_INDEX).id(response.id())); | ||
| GetResponse<LogTemplate> resp = | ||
| client.get(getRequest, LogTemplate.class); | ||
| return resp.source(); | ||
| } | ||
| } catch (Exception e) { | ||
| String message = MessageFormat.format(TextUtil.LOG_TEMPLATE_NOT_UPDATED, logTemplate.getName()); | ||
| logger.log(Level.SEVERE, message, e); | ||
| throw new ResponseStatusException(HttpStatus.INTERNAL_SERVER_ERROR, message); | ||
| } | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public Optional<LogTemplate> findById(String id) { | ||
| try { | ||
| GetRequest getRequest = | ||
| GetRequest.of(g -> | ||
| g.index(ElasticConfig.ES_LOG_TEMPLATE_INDEX).id(id)); | ||
| GetResponse<LogTemplate> resp = | ||
| client.get(getRequest, LogTemplate.class); | ||
|
|
||
| if (!resp.found()) { | ||
| throw new ResponseStatusException(HttpStatus.NOT_FOUND, MessageFormat.format(TextUtil.LOG_TEMPLATE_NOT_FOUND, id)); | ||
| } | ||
| return resp.source() != null ? Optional.of(resp.source()) : Optional.empty(); | ||
| } catch (Exception e) { | ||
| // https://www.baeldung.com/exception-handling-for-rest-with-spring#controlleradvice | ||
| String message = MessageFormat.format(TextUtil.LOG_TEMPLATE_NOT_RETRIEVED, id); | ||
| logger.log(Level.SEVERE, message, e); | ||
| throw new ResponseStatusException(HttpStatus.NOT_FOUND, message); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public boolean existsById(String logId) { | ||
| throw new ResponseStatusException(HttpStatus.NOT_FOUND, TextUtil.LOG_TEMPLATE_EXISTS_UNSUPPORTED); | ||
| } | ||
|
|
||
| @Override | ||
| public Iterable<LogTemplate> findAll() { | ||
| try { | ||
| SearchRequest searchRequest = | ||
| SearchRequest.of(s -> | ||
| s.index(ElasticConfig.ES_LOG_TEMPLATE_INDEX) | ||
| .query(new MatchAllQuery.Builder().build()._toQuery()) | ||
| .sort(SortOptions.of(so -> so.field(FieldSort.of(f -> f.field("name")))))); | ||
|
|
||
| SearchResponse<LogTemplate> response = | ||
| client.search(searchRequest, LogTemplate.class); | ||
| return response.hits().hits().stream().map(Hit::source).collect(Collectors.toList()); | ||
| } catch (Exception e) { | ||
| logger.log(Level.SEVERE, TextUtil.LOG_TEMPLATES_NOT_RETRIEVED, e); | ||
| throw new ResponseStatusException(HttpStatus.INTERNAL_SERVER_ERROR, TextUtil.LOG_TEMPLATES_NOT_RETRIEVED); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public Iterable<LogTemplate> findAllById(Iterable<String> logIds) { | ||
| throw new ResponseStatusException(HttpStatus.NOT_FOUND, TextUtil.LOG_TEMPLATES_FIND_ALL_BY_ID); | ||
| } | ||
|
|
||
| @Override | ||
| public long count() { | ||
| return 0; | ||
| } | ||
|
|
||
| @Override | ||
| public void deleteById(String id) { | ||
| try { | ||
| DeleteRequest deleteRequest = DeleteRequest.of(d -> | ||
| d.index(ElasticConfig.ES_LOG_TEMPLATE_INDEX).id(id).refresh(Refresh.True)); | ||
| DeleteResponse deleteResponse = client.delete(deleteRequest); | ||
| if (deleteResponse.result().equals(Result.Deleted)) { | ||
| logger.log(Level.WARNING, MessageFormat.format(TextUtil.LOG_TEMPLATE_DELETED, id)); | ||
| } else { | ||
| logger.log(Level.WARNING, MessageFormat.format(TextUtil.LOG_TEMPLATE_NOT_DELETED, id)); | ||
| } | ||
| } catch (IOException e) { | ||
| logger.log(Level.SEVERE, MessageFormat.format(TextUtil.LOG_TEMPLATE_NOT_DELETED, id), e); | ||
| throw new RuntimeException(e); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void delete(LogTemplate logTemplate) { | ||
| deleteById(String.valueOf(logTemplate.getId())); | ||
| } | ||
|
|
||
| @Override | ||
| public void deleteAll(Iterable<? extends LogTemplate> entities) { | ||
| throw new ResponseStatusException(HttpStatus.NOT_FOUND, TextUtil.LOG_TEMPLATE_DELETE_ALL_NOT_SUPPORTED); | ||
| } | ||
|
|
||
| @Override | ||
| public void deleteAll() { | ||
| throw new ResponseStatusException(HttpStatus.FORBIDDEN, TextUtil.LOG_TEMPLATE_DELETE_ALL_NOT_SUPPORTED); | ||
| } | ||
|
|
||
| @Override | ||
| public void deleteAllById(Iterable ids) { | ||
| throw new ResponseStatusException(HttpStatus.NOT_FOUND, TextUtil.LOG_TEMPLATE_DELETE_ALL_NOT_SUPPORTED); | ||
| } | ||
| } |
Oops, something went wrong.