mirror of
https://github.com/alibaba/fastjson.git
synced 2026-10-22 09:34:13 +08:00
Merge pull request #114 from mz0827/master
add "extractValueFromSingleKeyModel" property to FastJsonJsonView just l...
This commit is contained in:
高铁
@@ -1,5 +1,13 @@
|
||||
package com.alibaba.fastjson.support.spring;
|
||||
|
||||
import com.alibaba.fastjson.JSON;
|
||||
import com.alibaba.fastjson.serializer.SerializerFeature;
|
||||
import org.springframework.util.CollectionUtils;
|
||||
import org.springframework.validation.BindingResult;
|
||||
import org.springframework.web.servlet.view.AbstractView;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.OutputStream;
|
||||
import java.nio.charset.Charset;
|
||||
@@ -7,130 +15,137 @@ import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
|
||||
import org.springframework.util.CollectionUtils;
|
||||
import org.springframework.validation.BindingResult;
|
||||
import org.springframework.web.servlet.view.AbstractView;
|
||||
|
||||
import com.alibaba.fastjson.JSON;
|
||||
import com.alibaba.fastjson.serializer.SerializerFeature;
|
||||
|
||||
/**
|
||||
* @author libinsong1204@gmail.com
|
||||
*/
|
||||
public class FastJsonJsonView extends AbstractView {
|
||||
|
||||
public static final String DEFAULT_CONTENT_TYPE = "application/json";
|
||||
public static final String DEFAULT_CONTENT_TYPE = "application/json";
|
||||
|
||||
public final static Charset UTF8 = Charset.forName("UTF-8");
|
||||
public final static Charset UTF8 = Charset.forName("UTF-8");
|
||||
|
||||
private Charset charset = UTF8;
|
||||
private Charset charset = UTF8;
|
||||
|
||||
private SerializerFeature[] serializerFeatures = new SerializerFeature[0];
|
||||
private SerializerFeature[] serializerFeatures = new SerializerFeature[0];
|
||||
|
||||
private Set<String> renderedAttributes;
|
||||
private Set<String> renderedAttributes;
|
||||
|
||||
private boolean disableCaching = true;
|
||||
private boolean disableCaching = true;
|
||||
|
||||
private boolean updateContentLength = false;
|
||||
private boolean updateContentLength = false;
|
||||
|
||||
public FastJsonJsonView(){
|
||||
setContentType(DEFAULT_CONTENT_TYPE);
|
||||
setExposePathVariables(false);
|
||||
}
|
||||
private boolean extractValueFromSingleKeyModel = false;
|
||||
|
||||
public void setRenderedAttributes(Set<String> renderedAttributes) {
|
||||
this.renderedAttributes = renderedAttributes;
|
||||
}
|
||||
public FastJsonJsonView(){
|
||||
setContentType(DEFAULT_CONTENT_TYPE);
|
||||
setExposePathVariables(false);
|
||||
}
|
||||
|
||||
@Deprecated
|
||||
public void setSerializerFeature(SerializerFeature... features) {
|
||||
this.setFeatures(features);
|
||||
}
|
||||
public void setRenderedAttributes(Set<String> renderedAttributes) {
|
||||
this.renderedAttributes = renderedAttributes;
|
||||
}
|
||||
|
||||
public Charset getCharset() {
|
||||
return this.charset;
|
||||
}
|
||||
@Deprecated
|
||||
public void setSerializerFeature(SerializerFeature... features) {
|
||||
this.setFeatures(features);
|
||||
}
|
||||
|
||||
public void setCharset(Charset charset) {
|
||||
this.charset = charset;
|
||||
}
|
||||
public Charset getCharset() {
|
||||
return this.charset;
|
||||
}
|
||||
|
||||
public SerializerFeature[] getFeatures() {
|
||||
return serializerFeatures;
|
||||
}
|
||||
public void setCharset(Charset charset) {
|
||||
this.charset = charset;
|
||||
}
|
||||
|
||||
public void setFeatures(SerializerFeature... features) {
|
||||
this.serializerFeatures = features;
|
||||
}
|
||||
public SerializerFeature[] getFeatures() {
|
||||
return serializerFeatures;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void renderMergedOutputModel(Map<String, Object> model, HttpServletRequest request,
|
||||
HttpServletResponse response) throws Exception {
|
||||
Object value = filterModel(model);
|
||||
public void setFeatures(SerializerFeature... features) {
|
||||
this.serializerFeatures = features;
|
||||
}
|
||||
|
||||
String text = JSON.toJSONString(value, serializerFeatures);
|
||||
byte[] bytes = text.getBytes(charset);
|
||||
public boolean isExtractValueFromSingleKeyModel() {
|
||||
return extractValueFromSingleKeyModel;
|
||||
}
|
||||
|
||||
OutputStream stream = this.updateContentLength ? createTemporaryOutputStream() : response.getOutputStream();
|
||||
stream.write(bytes);
|
||||
public void setExtractValueFromSingleKeyModel(boolean extractValueFromSingleKeyModel) {
|
||||
this.extractValueFromSingleKeyModel = extractValueFromSingleKeyModel;
|
||||
}
|
||||
|
||||
if (this.updateContentLength) {
|
||||
writeToResponse(response, (ByteArrayOutputStream) stream);
|
||||
}
|
||||
}
|
||||
@Override
|
||||
protected void renderMergedOutputModel(Map<String, Object> model, HttpServletRequest request,
|
||||
HttpServletResponse response) throws Exception {
|
||||
Object value = filterModel(model);
|
||||
|
||||
@Override
|
||||
protected void prepareResponse(HttpServletRequest request, HttpServletResponse response) {
|
||||
setResponseContentType(request, response);
|
||||
response.setCharacterEncoding(UTF8.name());
|
||||
if (this.disableCaching) {
|
||||
response.addHeader("Pragma", "no-cache");
|
||||
response.addHeader("Cache-Control", "no-cache, no-store, max-age=0");
|
||||
response.addDateHeader("Expires", 1L);
|
||||
}
|
||||
}
|
||||
String text = JSON.toJSONString(value, serializerFeatures);
|
||||
byte[] bytes = text.getBytes(charset);
|
||||
|
||||
/**
|
||||
* Disables caching of the generated JSON.
|
||||
* <p>
|
||||
* Default is {@code true}, which will prevent the client from caching the generated JSON.
|
||||
*/
|
||||
public void setDisableCaching(boolean disableCaching) {
|
||||
this.disableCaching = disableCaching;
|
||||
}
|
||||
OutputStream stream = this.updateContentLength ? createTemporaryOutputStream() : response.getOutputStream();
|
||||
stream.write(bytes);
|
||||
|
||||
/**
|
||||
* Whether to update the 'Content-Length' header of the response. When set to {@code true}, the response is buffered
|
||||
* in order to determine the content length and set the 'Content-Length' header of the response.
|
||||
* <p>
|
||||
* The default setting is {@code false}.
|
||||
*/
|
||||
public void setUpdateContentLength(boolean updateContentLength) {
|
||||
this.updateContentLength = updateContentLength;
|
||||
}
|
||||
if (this.updateContentLength) {
|
||||
writeToResponse(response, (ByteArrayOutputStream) stream);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Filters out undesired attributes from the given model. The return value can be either another {@link Map}, or a
|
||||
* single value object.
|
||||
* <p>
|
||||
* Default implementation removes {@link BindingResult} instances and entries not included in the
|
||||
* {@link #setRenderedAttributes(Set) renderedAttributes} property.
|
||||
*
|
||||
* @param model the model, as passed on to {@link #renderMergedOutputModel}
|
||||
* @return the object to be rendered
|
||||
*/
|
||||
protected Object filterModel(Map<String, Object> model) {
|
||||
Map<String, Object> result = new HashMap<String, Object>(model.size());
|
||||
Set<String> renderedAttributes = !CollectionUtils.isEmpty(this.renderedAttributes) ? this.renderedAttributes : model.keySet();
|
||||
for (Map.Entry<String, Object> entry : model.entrySet()) {
|
||||
if (!(entry.getValue() instanceof BindingResult) && renderedAttributes.contains(entry.getKey())) {
|
||||
result.put(entry.getKey(), entry.getValue());
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
@Override
|
||||
protected void prepareResponse(HttpServletRequest request, HttpServletResponse response) {
|
||||
setResponseContentType(request, response);
|
||||
response.setCharacterEncoding(UTF8.name());
|
||||
if (this.disableCaching) {
|
||||
response.addHeader("Pragma", "no-cache");
|
||||
response.addHeader("Cache-Control", "no-cache, no-store, max-age=0");
|
||||
response.addDateHeader("Expires", 1L);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
/**
|
||||
* Disables caching of the generated JSON.
|
||||
* <p>
|
||||
* Default is {@code true}, which will prevent the client from caching the generated JSON.
|
||||
*/
|
||||
public void setDisableCaching(boolean disableCaching) {
|
||||
this.disableCaching = disableCaching;
|
||||
}
|
||||
|
||||
/**
|
||||
* Whether to update the 'Content-Length' header of the response. When set to {@code true}, the response is buffered
|
||||
* in order to determine the content length and set the 'Content-Length' header of the response.
|
||||
* <p>
|
||||
* The default setting is {@code false}.
|
||||
*/
|
||||
public void setUpdateContentLength(boolean updateContentLength) {
|
||||
this.updateContentLength = updateContentLength;
|
||||
}
|
||||
|
||||
/**
|
||||
* Filters out undesired attributes from the given model. The return value can be either another {@link Map}, or a
|
||||
* single value object.
|
||||
* <p>
|
||||
* Default implementation removes {@link BindingResult} instances and entries not included in the
|
||||
* {@link #setRenderedAttributes(Set) renderedAttributes} property.
|
||||
*
|
||||
* @param model the model, as passed on to {@link #renderMergedOutputModel}
|
||||
* @return the object to be rendered
|
||||
*/
|
||||
protected Object filterModel(Map<String, Object> model) {
|
||||
Map<String, Object> result = new HashMap<String, Object>(model.size());
|
||||
Set<String> renderedAttributes = !CollectionUtils.isEmpty(this.renderedAttributes) ? this.renderedAttributes : model.keySet();
|
||||
for (Map.Entry<String, Object> entry : model.entrySet()) {
|
||||
if (!(entry.getValue() instanceof BindingResult) && renderedAttributes.contains(entry.getKey())) {
|
||||
result.put(entry.getKey(), entry.getValue());
|
||||
}
|
||||
}
|
||||
if(extractValueFromSingleKeyModel){
|
||||
if(result.size() == 1){
|
||||
for(Map.Entry<String, Object> entry : result.entrySet()){
|
||||
return entry.getValue();
|
||||
}
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user