org.zalando.riptide.SerializationPlugin Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of riptide-core Show documentation

Client side response routing

There is a newer version: 4.0.0

package org.zalando.riptide;

import lombok.AllArgsConstructor;
import org.springframework.http.HttpOutputMessage;
import org.zalando.riptide.RequestArguments.Entity;

import java.io.IOException;

@AllArgsConstructor
final class SerializationPlugin implements Plugin {

    private final MessageWriter writer;

    @Override
    public RequestExecution aroundSerialization(final RequestExecution execution) {
        return arguments -> execution.execute(arguments.getEntity() == null ?
                arguments.withEntity(new SerializingEntity(arguments)) :
                arguments);
    }

    @AllArgsConstructor
    private class SerializingEntity implements Entity {

        private final RequestArguments arguments;

        @Override
        public void writeTo(final HttpOutputMessage message) throws IOException {
            writer.write(message, arguments);
        }

        @Override
        public boolean isEmpty() {
            return arguments.getBody() == null;
        }

    }

}