mirror of
https://github.com/nextcloud/server.git
synced 2026-05-28 04:32:30 -04:00
Add OC_Migrate::copyRows() method
This commit is contained in:
parent
c3dfcc5b21
commit
3ca76d24a9
2 changed files with 90 additions and 58 deletions
|
|
@ -2,73 +2,48 @@
|
|||
class OC_Migrate_Provider_Bookmarks extends OC_Migrate_Provider{
|
||||
|
||||
// Create the xml for the user supplied
|
||||
function export($uid){
|
||||
function export( $uid ){
|
||||
|
||||
$bookmarks = array();
|
||||
|
||||
$query = OC_DB::prepare("SELECT * FROM *PREFIX*bookmarks WHERE *PREFIX*bookmarks.user_id = ?");
|
||||
$bookmarksdata =& $query->execute(array($uid));
|
||||
// Foreach bookmark
|
||||
while ($row = $bookmarksdata->fetchRow()) {
|
||||
|
||||
// Get the tags
|
||||
$query = OC_DB::prepare("SELECT * FROM *PREFIX*bookmarks_tags WHERE *PREFIX*bookmarks_tags.bookmark_id = ?");
|
||||
$tagsdata =& $query->execute(array($row['id']));
|
||||
$options = array(
|
||||
'table'=>'bookmarks',
|
||||
'matchcol'=>'user_id',
|
||||
'matchval'=>$uid,
|
||||
'idcol'=>'id'
|
||||
);
|
||||
$ids = OC_Migrate::copyRows( $options );
|
||||
|
||||
$tags = array();
|
||||
// Foreach tag
|
||||
while ($row = $tagsdata->fetchRow()) {
|
||||
$tags[] = $row['tag'];
|
||||
}
|
||||
|
||||
$bookmarks[] = array(
|
||||
'url' => $row['url'],
|
||||
'title' => $row['title'],
|
||||
'public' => $row['public'],
|
||||
'added' => $row['added'],
|
||||
'lastmodified' => $row['lastmodified'],
|
||||
'clickcount' => $row['clickcount'],
|
||||
'tags' => $tags
|
||||
);
|
||||
|
||||
}
|
||||
$options = array(
|
||||
'table'=>'bookmarks_tags',
|
||||
'matchcol'=>'id',
|
||||
'matchval'=>$ids
|
||||
);
|
||||
|
||||
return array('bookmarks' => $bookmarks);
|
||||
// Export tags
|
||||
OC_Migrate::copyRows( $options );
|
||||
|
||||
}
|
||||
|
||||
// Import function for bookmarks
|
||||
function import($data,$uid){
|
||||
function import( $data, $uid ){
|
||||
|
||||
// Different import code for different versions of the app
|
||||
switch($data['info']['version']){
|
||||
default:
|
||||
// Foreach bookmark
|
||||
foreach($data['data']['bookmarks'] as $bookmark){
|
||||
|
||||
$query = OC_DB::prepare( "INSERT INTO `*PREFIX*bookmarks` ( `url`, `title`, `user_id`, `public`, `added`, `lastmodified`, `clickcount` ) VALUES( ?, ?, ?, ?, ?, ?, ? )" );
|
||||
$result = $query->execute( array(
|
||||
$bookmark['url'],
|
||||
$bookmark['title'],
|
||||
$uid,
|
||||
$bookmark['public'],
|
||||
$bookmark['added'],
|
||||
$bookmark['lastmodified'],
|
||||
$bookmark['clickcount']
|
||||
) );
|
||||
// Now add the tags
|
||||
$id = OC_DB::insertid();
|
||||
foreach($bookmark['tags'] as $tag){
|
||||
$query = OC_DB::prepare( "INSERT INTO `*PREFIX*bookmarks_tags` ( `id`, `tag` ) VALUES( ?, ? )" );
|
||||
$result = $query->execute( array( $id, $tag));
|
||||
}
|
||||
|
||||
}
|
||||
break;
|
||||
// new id mapping
|
||||
$newids = array();
|
||||
|
||||
// Import bookmarks
|
||||
foreach($data['bookmarks'] as $bookmark){
|
||||
$bookmark['user_id'] = $uid;
|
||||
// import to the db now
|
||||
$newids[$bookmark['id']] = OC_DB::insertid();
|
||||
}
|
||||
|
||||
// Import tags
|
||||
foreach($data['bookmarks_tags'] as $tag){
|
||||
// Map the new ids
|
||||
$tag['id'] = $newids[$tag['id']];
|
||||
// Import to the db now using OC_DB
|
||||
}
|
||||
// Finished import
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
new OC_Migrate_Provider_Bookmarks('bookmarks');
|
||||
new OC_Migrate_Provider_Bookmarks( 'bookmarks' );
|
||||
|
|
@ -134,7 +134,7 @@ class OC_Migrate{
|
|||
|
||||
// @breif prepares the db
|
||||
// @param $query the sql query to prepare
|
||||
public static function prepareDB( $query ){
|
||||
public static function prepare( $query ){
|
||||
|
||||
// Optimize the query
|
||||
$query = self::processQuery( $query );
|
||||
|
|
@ -174,12 +174,69 @@ class OC_Migrate{
|
|||
|
||||
}
|
||||
|
||||
// @brief copys rows to migration.db from the main database
|
||||
// @param $options array of options.
|
||||
// @return bool
|
||||
public static function copyRows( $options ){
|
||||
if( !array_key_exists( 'table', $options ) ){
|
||||
return false;
|
||||
}
|
||||
|
||||
// Need to include 'where' in the query?
|
||||
if( array_key_exists( 'matchval', $options ) && array_key_exists( 'matchcol', $options ) ){
|
||||
foreach( $options['matchval'] as $matchval ){
|
||||
// Run the query for this match value (where x = y value)
|
||||
$query = OC_DB::prepare( "SELECT * FROM *PREFIX*" . $options['table'] . " WHERE " . $options['matchcol'] . " LIKE ?" );
|
||||
$results = $query->execute( array( $matchval ) );
|
||||
self::insertData( $results, $options );
|
||||
|
||||
}
|
||||
|
||||
} else {
|
||||
// Just get everything
|
||||
$query = OC_DB::prepare( "SELECT * FROM *PREFIX*" . $options['table'] );
|
||||
$results = $query->execute();
|
||||
self::insertData( $results, $options );
|
||||
|
||||
}
|
||||
|
||||
return true;
|
||||
|
||||
}
|
||||
|
||||
// @breif saves a sql data set into migration.db
|
||||
// @param $data a sql data set returned from self::prepare()->query()
|
||||
// @param $options array of copyRows options
|
||||
// @return void
|
||||
private static function insertData( $data, $options ){
|
||||
while( $data = $result->fetchRow() ){
|
||||
// Now save all this to the migration.db
|
||||
foreach($row as $field=>$value){
|
||||
$fields[] = $field;
|
||||
$values[] = $value;
|
||||
}
|
||||
|
||||
// Generate some sql
|
||||
$sql = "INSERT INTO `*PREFIX*" . $options['table'] . '` ( `';
|
||||
$fieldssql = implode( '`, `', $fields );
|
||||
$sql .= $fieldssql . "` ) VALUES( ";
|
||||
$valuessql = substr( str_repeat( '?, ', count( $fields ) ),0,-1 );
|
||||
$sql .= $valuessql . " )";
|
||||
// Make the query
|
||||
$query = self::prepare( $sql );
|
||||
$query->execute( $values );
|
||||
}
|
||||
}
|
||||
|
||||
// @breif creates the tables in migration.db from an apps database.xml
|
||||
// @param $appid string id of the app
|
||||
// @return bool whether the operation was successful
|
||||
private static function createAppTables( $appid ){
|
||||
$file = OC::$SERVERROOT.'/apps/'.$appid.'appinfo/database.xml';
|
||||
if(file_exists( $file )){
|
||||
|
||||
self::connectScheme();
|
||||
|
||||
// There is a database.xml file
|
||||
$content = file_get_contents( $file );
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue