Unit test for PostDataTransactionUtil class
[netconf.git] / restconf / sal-rest-connector / src / main / java / org / opendaylight / restconf / restful / utils / ResponseFactory.java
1 /*
2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.restconf.restful.utils;
9
10 import java.net.URI;
11 import javax.ws.rs.core.Response;
12 import javax.ws.rs.core.Response.ResponseBuilder;
13 import javax.ws.rs.core.Response.Status;
14 import org.apache.commons.lang3.builder.Builder;
15 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
16
17 final class ResponseFactory extends FutureDataFactory<Void> implements Builder<Response> {
18
19     private ResponseBuilder responseBuilder;
20     ResponseFactory(final NormalizedNode<?, ?> readData) {
21         final Status status = prepareStatus(readData);
22         this.responseBuilder = Response.status(status);
23     }
24
25     ResponseFactory(final NormalizedNode<?, ?> readData, final URI location) {
26         final Status status = prepareStatus(readData);
27         this.responseBuilder = Response.status(status);
28         this.responseBuilder.location(location);
29     }
30
31     ResponseFactory() {
32         this.responseBuilder = Response.status(Status.OK);
33     }
34
35     @Override
36     public Response build() {
37         if (getFailureStatus()) {
38             responseBuilder = responseBuilder.status(Response.Status.INTERNAL_SERVER_ERROR);
39         }
40         return this.responseBuilder.build();
41     }
42
43     private Status prepareStatus(final NormalizedNode<?, ?> readData) {
44         return readData != null ? Status.OK : Status.CREATED;
45     }
46 }