Retrofit 2在拦截器中将POST请求添加到RequestBody

Question

我有以下内容：

OkHttpClient client = new OkHttpClient();
client.interceptors().add(new Interceptor() {
    @Override
    public com.squareup.okhttp.Response intercept(Chain chain) throws IOException {
        Request request = chain.request();
        HttpUrl url = request.httpUrl().newBuilder()
                     .addQueryParameter("platform", "android")
                     .addQueryParameter("app_version", com.package.BuildConfig.VERSION_NAME)
                     .build();
        Request newRequest = chain.request().newBuilder().url(url).build();
        return chain.proceed(newRequest);
    }
});

但是我也想将一个额外的帖子键值附加到请求体中，其中包含用户密钥。这看起来会像这样：

    RequestBody newBody = RequestBody.create(request.body().contentType(),request.body().content+ request.addPost("sUserKey","3254345kdskf");
...
...
 Request newRequest = chain.request()
.newBuilder()
.url(url)
.post(newBody)
.build();

- Nic Capdevila

---

什么问题？ - Blackbelt

你想对不同的终端执行POST请求吗？ - Blackbelt

不完全正确。我希望端点保持不变，但是添加额外的参数。我想要添加("sUserkey","345i34oj53i4s")。 - Nic Capdevila

你想在正文中添加些什么吗？ - Blackbelt

让我们在聊天中继续这个讨论。 - Nic Capdevila

Answer 1

向RequestBody追加内容并不是一件简单的事情。以下是一个自定义RequestBody的草图，它将添加一个帖子参数。需要注意的是——你需要添加一些错误检查，比如确保现有的请求体不为空。给定的代码还假定所有传递到该拦截器的调用都是POST。如果不是这种情况，则需要在应用新的请求体之前检查传入的请求类型。此外，由于此代码仅将新参数复制到请求体中，因此需要确保名称和主体已经进行了URL编码（如果需要的话）。

class AddPostParamRequestBody extends RequestBody {

    final RequestBody body;
    final String parameter;

    AddPostParamRequestBody(RequestBody body, String name, String value) {
        this.body = body;
        this.parameter = "&" + name + "=" + value;
    }

    @Override
    public long contentLength() throws IOException {
        return body.contentLength() + parameter.length();
    }

    @Override
    public MediaType contentType() {
        return body.contentType();
    }

    @Override
    public void writeTo(BufferedSink bufferedSink) throws IOException {
        body.writeTo(bufferedSink);
        bufferedSink.writeString(parameter, Charset.forName("UTF-8"));
    }

} 

然后你可以在你的拦截器中使用 --

client.interceptors().add(new Interceptor() {
    @Override
    public com.squareup.okhttp.Response intercept(Chain chain) throws IOException {
        Request request = chain.request();
        HttpUrl url = request.httpUrl().newBuilder().addQueryParameter("added", "param").build();
        AddPostParamRequestBody newBody = new AddPostParamRequestBody(request.body(), "sUserKey","3254345kdskf");
        Request newRequest = request.newBuilder().post(newBody).url(url).build();
        return chain.proceed(newRequest);
    }
});

您的另一个选择是在Retrofit定义中包含额外的Field注释，并在每次调用时传递它，但我认为您试图避免这种情况。

- iagreen

---

addQueryParameter("added", "param")查询中的added和param是什么意思？ - Rehan Sarwar

Answer 2

你可以不创建额外的类来完成它。

client.interceptors().add(new Interceptor() {
    @Override
    public com.squareup.okhttp.Response intercept(Chain chain) throws IOException {
        Request request = chain.request();
        String parameter = "&" + name + "=" + value;
        Request newRequest = interceptRequest(request, parameter)
        return chain.proceed(newRequest);
    }
});

这是一个创建新请求的简单方法。

 public static Request interceptRequest(@NotNull Request request, @NotNull String parameter)
            throws IOException {

        ByteArrayOutputStream baos = new ByteArrayOutputStream();

        Sink sink = Okio.sink(baos);
        BufferedSink bufferedSink = Okio.buffer(sink);

        /**
         * Write old params
         * */
        request.body().writeTo(bufferedSink);

        /**
         * write to buffer additional params
         * */
        bufferedSink.writeString(parameter, Charset.defaultCharset());

        RequestBody newRequestBody = RequestBody.create(
                request.body().contentType(),
                bufferedSink.buffer().readUtf8()
        );

        return request.newBuilder().post(newRequestBody).build();
    }

你也可以从Gist获取它。

- Silvestr