ITQC/gls_docs
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use new query structure.

Browse Source
This commit is contained in:
Fabian Märki
2015-07-14 15:11:43 +02:00
parent 7518010d9b
commit 11e931c8e3
4 changed files with 70 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/main/java/ch/psi/daq/queryrest/config/QueryRestConfig.java
View File

@@ -1,7 +1,6 @@
package ch.psi.daq.queryrest.config;
import java.util.Arrays;
import java.util.List;
import java.util.LinkedHashSet;
import java.util.Set;
import org.springframework.beans.factory.annotation.Autowired;
@@ -15,7 +14,6 @@ import org.springframework.util.StringUtils;
import com.fasterxml.jackson.annotation.JsonInclude.Include;
import com.fasterxml.jackson.core.JsonFactory;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.common.collect.Sets;
import ch.psi.daq.common.statistic.StorelessStatistics;
import ch.psi.daq.domain.cassandra.ChannelEvent;
@@ -64,12 +62,14 @@ public class QueryRestConfig {
}
@Bean
private Set<String> defaultResponseFields() {
List<String> defaultFields = Arrays.asList(
StringUtils.commaDelimitedListToStringArray(
env.getProperty("queryrest.default.response.fields")
));
Set<String> defaultResponseFields = Sets.newHashSet(defaultFields.iterator());
public Set<String> defaultResponseFields() {
String[] responseFields = StringUtils.commaDelimitedListToStringArray(env.getProperty("queryrest.default.response.fields"));
// preserve order
LinkedHashSet<String> defaultResponseFields = new LinkedHashSet<>(responseFields.length);
for (String field : defaultResponseFields) {
defaultResponseFields.add(field);
}
return defaultResponseFields;
}
}

21
src/main/java/ch/psi/daq/queryrest/response/ResponseStreamWriter.java
View File

@@ -1,11 +1,8 @@
/**
*
*/
package ch.psi.daq.queryrest.response;
import java.io.IOException;
import java.util.LinkedHashSet;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import javax.servlet.ServletResponse;
@@ -24,6 +21,7 @@ import com.fasterxml.jackson.databind.ser.impl.SimpleFilterProvider;
import ch.psi.daq.domain.cassandra.DataEvent;
import ch.psi.daq.query.model.AbstractQuery;
import ch.psi.daq.query.model.AggregationEnum;
/**
* Takes a Java 8 stream and writes it to the output stream provided by the {@link ServletResponse}
@@ -41,6 +39,7 @@ public class ResponseStreamWriter {
@Autowired
private Set<String> defaultResponseFields;
/**
* Responding with the the contents of the stream by writing into the output stream of the
* {@link ServletResponse}.
@@ -52,17 +51,13 @@ public class ResponseStreamWriter {
*/
public void respond(Stream<DataEvent> stream, AbstractQuery query, ServletResponse response) throws IOException {
Set<String> includedFields = query.getFields();
if (includedFields == null) {
includedFields = defaultResponseFields;
}
Set<String> includedFields = query.getFieldsOrDefault(defaultResponseFields);
if (query.getAggregations() != null) {
includedFields.addAll(query.getAggregations()
.stream()
.map(a -> {
return a.getType().toString();
}).collect(Collectors.toSet()));
includedFields = new LinkedHashSet<String>(includedFields);
for (AggregationEnum aggregation : query.getAggregations()) {
includedFields.add(aggregation.name());
}
}
ObjectWriter writer = configureWriter(includedFields);
respondInternal(stream, response, writer);