stack/rest/src/main/java/org/apache/usergrid/rest/MigrateResource.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.usergrid.rest;


import java.util.Map;

import javax.ws.rs.DefaultValue;
import javax.ws.rs.GET;
import javax.ws.rs.PUT;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.UriInfo;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;

import org.apache.usergrid.rest.security.annotations.RequireSystemAccess;

import com.fasterxml.jackson.databind.node.JsonNodeFactory;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.google.common.base.Preconditions;
import com.sun.jersey.api.json.JSONWithPadding;


@Component
@Scope( "singleton" )
@Produces( {
        MediaType.APPLICATION_JSON, "application/javascript", "application/x-javascript", "text/ecmascript",
        "application/ecmascript", "text/jscript"
} )
public class MigrateResource extends AbstractContextResource {

    private static final Logger logger = LoggerFactory.getLogger( MigrateResource.class );


    public MigrateResource() {
        logger.info( "SystemResource initialized" );
    }



    @RequireSystemAccess
    @PUT
    @Path( "run" )
    public JSONWithPadding migrateData( @Context UriInfo ui,
                                        @QueryParam( "callback" ) @DefaultValue( "callback" ) String callback )
            throws Exception {

        ApiResponse response = createApiResponse();
        response.setAction( "Migrate Data" );
        //TODO make this use the task scheduler


        final Thread migrate = new Thread() {

            @Override
            public void run() {
                logger.info( "Rebuilding all indexes" );

                try {
                    emf.migrateData();
                }
                catch ( Exception e ) {
                    logger.error( "Unable to rebuild indexes", e );
                }
            }
        };

        migrate.setName( "Index migrate data formats" );
        migrate.setDaemon( true );
        migrate.start();


        response.setSuccess();

        return new JSONWithPadding( response, callback );
    }


    @RequireSystemAccess
    @PUT
    @Path( "set" )
    public JSONWithPadding setMigrationVersion( @Context UriInfo ui, Map<String, Object> json,
                                                @QueryParam( "callback" ) @DefaultValue( "" ) String callback )
            throws Exception {

        logger.debug( "newOrganization" );

        Preconditions.checkNotNull( json, "You must provide a json body" );

        String version = ( String ) json.get( "version" );

        Preconditions.checkArgument( version != null && version.length() > 0,
                "You must specify a version field in your json" );


        int intVersion = Integer.parseInt( version );

        emf.setMigrationVersion( intVersion );


        return migrateStatus( ui, callback );
    }


    @RequireSystemAccess
    @GET
    @Path( "status" )
    public JSONWithPadding migrateStatus( @Context UriInfo ui,
                                          @QueryParam( "callback" ) @DefaultValue( "callback" ) String callback )
            throws Exception {

        ApiResponse response = createApiResponse();
        response.setAction( "Migrate Schema indexes" );

        ObjectNode node = JsonNodeFactory.instance.objectNode();
        node.put( "currentVersion", emf.getMigrateDataVersion() );
        node.put( "lastMessage", emf.getMigrateDataStatus() );
        response.setProperty( "status", node );

        response.setSuccess();

        return new JSONWithPadding( response, callback );
    }

    @RequireSystemAccess
       @GET
       @Path( "count" )
       public JSONWithPadding migrateCount( @Context UriInfo ui,
                                             @QueryParam( "callback" ) @DefaultValue( "callback" ) String callback )
               throws Exception {

           ApiResponse response = createApiResponse();
           response.setAction( "Current entity count in system" );

           response.setProperty( "count", emf.performEntityCount()  );

           response.setSuccess();

           return new JSONWithPadding( response, callback );
       }

}