AWS API网关 - 集成响应主体映射

时间:2017-11-18 20:01:38

标签: amazon-web-services aws-api-gateway

在AWS API网关集成响应正文映射中,我有以下代码:

#set($inputRoot = $input.path('$.Item'))
[
#foreach($elem in $inputRoot.Events)
 {
  "id": $elem.id,
  "from" : $elem.from,
  "to" : $elem.to,
  "spent" : $elem.spent,
  #if("$!elem.comment" != "")
    "comment": $elem.comment,    
  #end
  "project" : {
    "id" : $elem.project.id,
    "number" : $elem.project.number,
    "name" : $elem.project.name
  }
  }
#if($foreach.hasNext),#end
#end
]

数据来自查询DynamoDB表的lambda函数

API网关返回如下数据:

[
 {
  "id": 123443214,
  "from" : 19:34,
  "to" : 22:30,
  "spent" : 02:56,
    "project" : {
    "id" : 4321,
    "number" : CIB,
    "name" : Backend
  }
  }
, {
  "id": 12341234,
  "from" : 19:34,
  "to" : 22:30,
  "spent" : 02:56,
    "project" : {
    "id" : 12341234,
    "number" : CIB,
    "name" : Backend
  }
  }
]

所以它已经格式化了。如何让APi Gateway返回未格式化的响应?所以它只是纯粹的json,没有断行,缩进等。

提前致谢!

1 个答案:

答案 0 :(得分:0)

(小的初步评论:你遗漏了一些JSON字符串值的引号)。

可以使用##删除换行符并使用#**#缩进,如下所示,但模板看起来有点难看:

#set($inputRoot = $input.path('$.Item'))##
[##
#foreach($elem in $inputRoot.Events)##
{##
#**#"id":$elem.id,##
#**#"from": $elem.from,##
#**#"to":$elem.to,##
#**#"spent":$elem.spent,##
#if("$!elem.comment" != "")##
#*  *#"comment":$elem.comment,##
#end##
#**#"project":{##
#**#"id":$elem.project.id,##
#**#"number":"$elem.project.number",##
#**#"name":"$elem.project.name"##
}##
}##
#if($foreach.hasNext),#end##
#end##
]##

由于缩进首先出现在这里的唯一原因是模板的可读性,我会转向另一个方向。

例如,您可以使用org.json在View servlet中添加后处理整理格式化程序:

import org.json.JSONObject;
  ....
Writer writer = new StringWriter();
getVelocityView().merge(template, context, writer);
String compactJSON = new JSONObject(writer.toString()).toString();
response.getWriter().write(compactJSON);

但是这只适用于小型JSON文件,因为响应被缓冲到内存中,所以让我们继续寻找更优雅的解决方案。

使用自定义ResouceLoader,可以预处理您的模板。

<强> CompactJSONResourceLoader.java

package my.custom.loader;

import java.io.InputStream;
import java.io.IOException;
import org.apache.commons.collections.ExtendedProperties;
import org.apache.velocity.exception.ResourceNotFoundException;
import org.apache.velocity.runtime.resource.Resource;
import org.apache.velocity.runtime.resource.loader.ResourceLoader;
import org.apache.velocity.runtime.resource.loader.ResourceLoaderFactory;

public class CompactJSONResourceLoader extends ResourceLoader
{
    protected ResourceLoader innerLoader = null;

    @Override
    public void init(ExtendedProperties configuration)
    {
        try
        {
            String innerLoaderID = configuration.getString("innerLoader") + ".resource.loader";
            String innerLoaderClass = rsvc.getConfiguration().getString(innerLoaderID + ".class");
            innerLoader = ResourceLoaderFactory.getLoader(rsvc, innerLoaderClass);
            ExtendedProperties innerConfiguration = rsvc.getConfiguration().subset(innerLoaderID);
            innerLoader.commonInit(rsvc, innerConfiguration);
            innerLoader.init(innerConfiguration);
        }
        catch (Exception e)
        {
            log.error("could not initialize CompactJSONResourceLoader inner loader", e);
        }
    }

    protected class CompactJSONInputStream extends InputStream
    {
        InputStream innerStream = null;
        boolean insideQuotes = false;

        public CompactJSONInputStream(InputStream innerStream)
        {
            this.innerStream = innerStream;
        }

        @Override
        public int read() throws IOException
        {
            int ch;
            do
            {
                ch = innerStream.read();
                if (insideQuotes)
                {
                    if (ch == '"') insideQuotes = false;
                    break;
                }
                else if (!Character.isWhitespace(ch))
                {
                    if (ch == '"') insideQuotes = true;
                    break;
                }
            }
            while (ch != -1);
            return ch;
        }
    }

    @Override
    public InputStream getResourceStream(String source) throws ResourceNotFoundException
    {
        return new CompactJSONInputStream(innerLoader.getResourceStream(source));
    }

    @Override
    public boolean isSourceModified(Resource resource)
    {
        return innerLoader.isSourceModified(resource);
    }

    @Override
    public long getLastModified(Resource resource)
    {
        return innerLoader.getLastModified(resource);
    }
}

然后您需要使用以下属性配置Velocity:

resource.loader = compact
compact.resource.loader.class = my.custom.loader.CompactJSONResourceLoader

compact.resource.loader.innerLoader = file

(当然,您可以用当前使用的资源加载器替换file。)