/*
* 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.system;
import com.fasterxml.jackson.databind.node.JsonNodeFactory;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.fasterxml.jackson.jaxrs.json.annotation.JSONP;
import com.google.common.base.Preconditions;
import com.google.inject.Injector;
import org.apache.usergrid.persistence.core.migration.data.DataMigrationManager;
import org.apache.usergrid.persistence.core.migration.schema.MigrationManager;
import org.apache.usergrid.rest.AbstractContextResource;
import org.apache.usergrid.rest.ApiResponse;
import org.apache.usergrid.rest.security.annotations.RequireSystemAccess;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;
import javax.ws.rs.*;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.UriInfo;
import java.util.Map;
import java.util.Set;
@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" );
}
@Autowired
private Injector guiceInjector;
@RequireSystemAccess
@PUT
@Path( "run" )
@JSONP
@Produces({MediaType.APPLICATION_JSON, "application/javascript"})
public ApiResponse 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( "Migrating Schema" );
try {
getMigrationManager().migrate(false);
}
catch ( Exception e ) {
logger.error( "Unable to migrate data", e );
}
logger.info( "Migrating Data" );
try {
getDataMigrationManager().migrate();
}
catch ( Exception e ) {
logger.error( "Unable to migrate data", e );
}
}
};
migrate.setName( "Index migrate data formats" );
migrate.setDaemon( true );
migrate.start();
response.setSuccess();
return response;
}
@RequireSystemAccess
@PUT
@Path( "run/{pluginName}" )
@JSONP
@Produces({MediaType.APPLICATION_JSON, "application/javascript"})
public ApiResponse migrateData(@PathParam("pluginName") String pluginName , @Context UriInfo ui,
@QueryParam( "callback" ) @DefaultValue( "callback" ) String callback )
throws Exception {
if(!getDataMigrationManager().pluginExists(pluginName)){
throw new IllegalArgumentException("Plugin doesn't exist: "+pluginName);
}
ApiResponse response = createApiResponse();
response.setAction( "Migrate Data: "+ pluginName );
//TODO make this use the task scheduler
final Thread migrate = new Thread() {
@Override
public void run() {
logger.info( "Migrating Data for plugin: {}", pluginName );
try {
getDataMigrationManager().migrate(pluginName);
}
catch ( Exception e ) {
logger.error( "Unable to migrate data for plugin: {}", pluginName, e );
}
}
};
migrate.setName( "Index migrate data formats: "+pluginName );
migrate.setDaemon( true );
migrate.start();
response.setSuccess();
return response;
}
@RequireSystemAccess
@PUT
@Path( "set" )
@JSONP
@Produces({MediaType.APPLICATION_JSON, "application/javascript"})
public ApiResponse setMigrationVersion(
@Context UriInfo ui, Map<String, Object> json,
@QueryParam( "callback" ) @DefaultValue( "" ) String callback )
throws Exception {
if (logger.isDebugEnabled()) {
logger.debug("setMigrationVersion");
}
Preconditions.checkNotNull( json, "You must provide a json body" );
Preconditions.checkArgument( json.keySet().size() > 0, "You must specify at least one module and version" );
ApiResponse response = createApiResponse();
response.setAction("Set Migration Versions");
ObjectNode node = JsonNodeFactory.instance.objectNode();
final DataMigrationManager dataMigrationManager = getDataMigrationManager();
final Set<String> plugins = dataMigrationManager.getPluginNames();
/**
* Set the migration version for the plugins specified
*/
for ( final String key : json.keySet() ) {
int version = ( int ) json.get( key );
dataMigrationManager.resetToVersion(key, version);
}
/**
* Echo back a response of the current versions for all plugins
*/
for(final String pluginName: plugins){
node.put(pluginName, dataMigrationManager.getCurrentVersion(pluginName));
}
response.setData( node );
response.setSuccess();
return response;
}
@RequireSystemAccess
@GET
@Path( "status" )
@JSONP
@Produces({MediaType.APPLICATION_JSON, "application/javascript"})
public ApiResponse 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();
final DataMigrationManager dataMigrationManager = getDataMigrationManager();
final Set<String> plugins = dataMigrationManager.getPluginNames();
for(final String pluginName: plugins){
node.put( pluginName, dataMigrationManager.getCurrentVersion( pluginName ) );
}
response.setData( node );
response.setSuccess();
return response;
}
@RequireSystemAccess
@GET
@Path( "count" )
@JSONP
@Produces({MediaType.APPLICATION_JSON, "application/javascript"})
public ApiResponse 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 response;
}
/**
* Get the schema migration manager
*/
private MigrationManager getMigrationManager() {
return guiceInjector.getInstance( MigrationManager.class );
}
/**
* Get the Data migration manager
*/
private DataMigrationManager getDataMigrationManager() {
return guiceInjector.getInstance( DataMigrationManager.class );
}
}