gvSIG projects pool

/*

 * gvSIG. Desktop Geographic Information System.

 * 

 * Copyright (C) 2007-2020 gvSIG Association.

 * 

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 * 

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 * 

 * You should have received a copy of the GNU General Public License 

 * along with this program. If not, see <https://www.gnu.org/licenses/>. 

 * 

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.vcsgis.lib.server.handlers;

import java.io.IOException;

import java.io.InputStream;

import java.io.Writer;

import java.util.Collections;

import java.util.Map;

import javax.json.JsonException;

import javax.json.JsonObject;

import javax.json.JsonReader;

import javax.json.stream.JsonGenerator;

import javax.json.stream.JsonGeneratorFactory;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.lang3.mutable.MutableObject;

import org.gvsig.json.Json;

import org.gvsig.json.JsonObjectBuilder;

import org.gvsig.vcsgis.lib.VCSGisUtils;

import static org.gvsig.vcsgis.lib.VCSGisUtils.ENTITY_AUTHORIZATIONS;

import static org.gvsig.vcsgis.lib.VCSGisUtils.ENTITY_FEATURETYPEASJSON;

import org.gvsig.vcsgis.lib.repository.VCSGisRepository;

import org.gvsig.vcsgis.lib.repository.VCSGisRepositoryData;

import org.gvsig.vcsgis.lib.repository.requests.VCSGisRequest;

import org.gvsig.vcsgis.lib.repository.requests.VCSGisUpdateRequest;

import static org.gvsig.vcsgis.lib.server.handlers.AbstractVCSGisServertHandler.LOGGER;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class UpdateHandler extends AbstractVCSGisServertHandler {

    private static final String REQUEST_NAME = "Update";

    public UpdateHandler(VCSGisRepository repository) {

        super(repository, REQUEST_NAME);

    }

    @Override

    protected void requestProducer(MutableObject<VCSGisRequest>req,Map<String,String> params, InputStream request_contents) throws IOException {

        LOGGER.debug("===: ["+this.getName()+"] requestProducer 1");

        JsonObject jsonParameters = null;

        try {

            JsonReader reader = Json.createReader(request_contents);

            JsonObject jsonRequest = reader.readObject();

            jsonParameters = jsonRequest.getJsonObject("Parameters");

        } catch (JsonException e) {

            //Do nothing

        }

        String entityName = params.get("EntityName");;

        if( jsonParameters != null ) {

            entityName = jsonParameters.getString("EntityName", entityName);

        }

        VCSGisUpdateRequest request = this.getRepository().createUpdateRequest(entityName);

        request.setAuthenticationTokenAndUser(params);

        if( jsonParameters == null ) {

            request.setLocalRevisionCode(params.get("LocalRevisionCode"));

        } else {

            request.setLocalRevisionCode(jsonParameters.getString("LocalRevisionCode", params.get("LocalRevisionCode")));

        }

        req.setValue(request);

//        notifyRequestConsumers();

        LOGGER.debug("===: ["+this.getName()+"] requestProducer 2 notifyRequestConsumers");

        notifyRequestConsumers();

        LOGGER.debug("===: ["+this.getName()+"] requestProducer 3 return");

    }

    @Override

    protected void responseProducer(VCSGisRequest req, Writer response_contents) throws IOException {

        LOGGER.debug("===: ["+this.getName()+"] responseProducer 1");

        final VCSGisUpdateRequest request = (VCSGisUpdateRequest) req;

        JsonGenerator gen = null;

        try {

            JsonGeneratorFactory genFactory = Json.createGeneratorFactory(

                Collections.singletonMap(JsonGenerator.PRETTY_PRINTING, true)

            );

            gen = genFactory.createGenerator(response_contents);

            LOGGER.debug("===: ["+this.getName()+"] responseProducer 2");

            gen.writeStartObject();

            gen.writeStartObject("Parameters");

            gen.write("StatusCode", request.getLastErrorCode());

            LOGGER.debug("===: ["+this.getName()+"] responseProducer 3");

            notifyResponseConsumers();

            gen.flush();

            if( StringUtils.isBlank(request.getLastErrorMessage()) ) {

                gen.writeNull("StatusMessage");

            } else {

                gen.write("StatusMessage", request.getLastErrorMessage());

            }

            JsonObjectBuilder builder = VCSGisUtils.toJsonBuilder(request.getEntity(),

                    (String t) -> StringUtils.equalsIgnoreCase(t, ENTITY_FEATURETYPEASJSON)

                    || StringUtils.equalsIgnoreCase(t, ENTITY_AUTHORIZATIONS));

            gen.write("Entity", builder.build());

            gen.writeEnd(); // Paramaters

            LOGGER.debug("===: ["+this.getName()+"] responseProducer 4");

            gen.writeStartArray("Data");

            for (VCSGisRepositoryData data : request.getData()) {

                gen.write(data.toJson());

                gen.flush();

            }

            LOGGER.debug("===: ["+this.getName()+"] responseProducer 5");

            gen.writeEnd(); // Data

            gen.writeEnd(); 

            LOGGER.debug("===: ["+this.getName()+"] responseProducer 6 gen.flush()");

            gen.flush();

            LOGGER.debug("===: ["+this.getName()+"] responseProducer 7 response_contents.flush()");

            response_contents.flush();

            LOGGER.debug("===: ["+this.getName()+"] responseProducer 8 end");

        } catch (Exception ex) {

            LOGGER.warn("Can't produce Json data for "+this.getName()+" response.",ex);

        } finally {

            LOGGER.debug("===: ["+this.getName()+"] responseProducer finally");

            IOUtils.closeQuietly(gen);

//            IOUtils.closeQuietly(response_contents);

        }

    }

}