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.sql.Timestamp;

import java.util.Map;

import javax.json.JsonObject;

import javax.json.JsonString;

import javax.json.JsonValue;

import javax.json.stream.JsonParser;

import static javax.json.stream.JsonParser.Event.END_ARRAY;

import static javax.json.stream.JsonParser.Event.END_OBJECT;

import static javax.json.stream.JsonParser.Event.START_ARRAY;

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.JsonArrayBuilder;

import org.gvsig.json.JsonObjectBuilder;

import org.gvsig.vcsgis.lib.ChangeImpl;

import org.gvsig.vcsgis.lib.PipedIterator;

import org.gvsig.vcsgis.lib.SAJParserImpl;

import org.gvsig.vcsgis.lib.VCSGisChange;

import org.gvsig.vcsgis.lib.VCSGisEntityEditable;

import org.gvsig.vcsgis.lib.VCSGisUtils;

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

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

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

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

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspaceEntity;

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspaceEntityImpl;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class CommitHandler extends AbstractVCSGisServertHandler {

    public CommitHandler(VCSGisRepository repository) {

        super(repository, "CommitHandler");

    }

    @Override

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

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

        final PipedIterator<VCSGisChange> changesIterator = new PipedIterator<>();                

        final VCSGisCommitRequest request = this.getRepository().createCommitRequest();

        request.setAuthenticationTokenAndUser(params);

        request.add(changesIterator);

        req.setValue(request);

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

        SAJParserImpl parser = new SAJParserImpl(

                request_contents, 

                new SAJParserImpl.SAJParserHandler() {

                    @Override

                    public void handle(SAJParserImpl.SAJParserContext context, JsonParser.Event e, Object value) {

                        if( e==null ) {

                            return;

                        }

                        switch (e) {

                            case END_OBJECT:

                                switch(context.getPathName()) {

                                    case "/Parameters":

                                        JsonObject jsonParams = ((javax.json.JsonObjectBuilder)value).build();

                                        if( jsonParams.isNull("EfectiveDate") ) {

                                            request.setEfectiveDate(null);

                                        } else {

                                            request.setEfectiveDate(Timestamp.valueOf(jsonParams.getString("EfectiveDate")));

                                        }

                                        if( jsonParams.isNull("RevisionDate") ) {

                                            request.setRevisionDate(null);

                                        } else {

                                            request.setRevisionDate(Timestamp.valueOf(jsonParams.getString("RevisionDate")));

                                        }

                                        if( jsonParams.isNull("Comment") ) {

                                            request.setComment(null);

                                        } else {

                                            request.setComment(jsonParams.getString("Comment"));

                                        }

                                        if( !jsonParams.isNull("Entities") ) {

                                            for (JsonValue jsonvEntity : jsonParams.getJsonArray("Entities")) {

                                                VCSGisWorkspaceEntity entity = new VCSGisWorkspaceEntityImpl((JsonObject) jsonvEntity);

                                                request.add(entity);

                                            }

                                        }

                                        if( !jsonParams.isNull("NewEntityCodes") ) {

                                            for (JsonValue jsonvEntityCode : jsonParams.getJsonArray("NewEntityCodes")) {

                                                request.markAsNew(((JsonString)jsonvEntityCode).getString());

                                            }

                                        }

                                        notifyRequestConsumers();

                                        break;

                                    case "/Data/?":

                                        // Hemos terminado de generar un elemento del array

                                        // de datos, construimos un Change y lo adicionamos a la 

                                        // cola de cambios.

                                        JsonObject jsonChange = ((javax.json.JsonObjectBuilder)value).build();

                                        VCSGisChange change = new ChangeImpl(jsonChange);

                                        changesIterator.put(change);

                                        break;

                                }

                                break;

                            case END_ARRAY:

                                if( "/Data".equals(context.getPathName()) ) {

                                    // Marcamos como que ya hemos llegado al final del

                                    // array de datos.

                                    changesIterator.close();

                                }

                                break;

                            case START_ARRAY:

                                switch(context.getPathName()) {

                                    case "/Data":

                                        // Forzamos que no se cree el array en memoria.

                                        context.setArrayBuilder(null);

                                        break;

                                }

                                break;

                            case START_OBJECT:

                                switch(context.getPathName()) {

                                    case "/Data/?":

                                        break;

                                }

                                break;

                            default:

                                break;

                        }

                    }

                }

        );

        parser.parse();

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

        notifyRequestConsumers();

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

    }

    @Override

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

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

        final VCSGisCommitRequest request = (VCSGisCommitRequest) req;

        try {

            JsonObjectBuilder builder = Json.createObjectBuilder();

            JsonObjectBuilder jsonParams = Json.createObjectBuilder();

            jsonParams.add("StatusCode", request.getLastErrorCode());

            jsonParams.add("StatusMessage", request.getLastErrorMessage());

            JsonArrayBuilder jsonEntities = Json.createArrayBuilder();

            for (VCSGisEntityEditable entity : request.getChangedLocalEntities()) {

//                jsonEntities.add(entity.toJsonBuilder());

                JsonObject jsonEntity = VCSGisUtils.toJsonBuilder(

                        entity, 

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

                    ).build();

                jsonEntities.add(jsonEntity);

            }

            jsonParams.add("Entities", jsonEntities);

            builder.add("Parameters", jsonParams);

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

            notifyResponseConsumers();

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

            IOUtils.write(builder.toString(), response_contents);

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

            response_contents.flush();

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

        } catch (Exception ex) {

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

        } finally {

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

            IOUtils.closeQuietly(response_contents);

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

        }

    }

}