2010-10-24 03:12:37 +04:00
#!/usr/bin/php -Cq
< ? php
2010-10-25 16:22:22 +04:00
require_once ( dirname ( dirname ( __FILE__ )) . '/lib/init-cmd.php' );
2010-10-24 03:12:37 +04:00
ini_set ( 'memory_limit' , '800M' );
$aCMDOptions = array (
" Create and setup nominatim search system " ,
array ( 'help' , 'h' , 0 , 1 , 0 , 0 , false , 'Show Help' ),
array ( 'quiet' , 'q' , 0 , 1 , 0 , 0 , 'bool' , 'Quiet output' ),
array ( 'verbose' , 'v' , 0 , 1 , 0 , 0 , 'bool' , 'Verbose output' ),
2010-11-15 19:34:27 +03:00
array ( 'osm-file' , '' , 0 , 1 , 1 , 1 , 'realpath' , 'File to import' ),
array ( 'threads' , '' , 0 , 1 , 1 , 1 , 'int' , 'Number of threads (where possible)' ),
array ( 'all' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Do the complete process' ),
2010-10-27 18:05:42 +04:00
array ( 'create-db' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Create nominatim db' ),
array ( 'setup-db' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Build a blank nominatim db' ),
2010-11-15 19:34:27 +03:00
array ( 'import-data' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Import a osm file' ),
2012-04-12 01:39:07 +04:00
array ( 'osm2pgsql-cache' , '' , 0 , 1 , 1 , 1 , 'int' , 'Cache size used by osm2pgsql' ),
2010-10-27 18:05:42 +04:00
array ( 'create-functions' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Create functions' ),
2012-05-22 18:27:42 +04:00
array ( 'enable-diff-updates' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Turn on the code required to make diff updates work' ),
array ( 'enable-debug-statements' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Include debug warning statements in pgsql commands' ),
2014-05-14 01:23:31 +04:00
array ( 'ignore-errors' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Continue import even when errors in SQL are present (EXPERT)' ),
2011-04-06 22:19:40 +04:00
array ( 'create-minimal-tables' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Create minimal main tables' ),
2010-10-27 18:05:42 +04:00
array ( 'create-tables' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Create main tables' ),
2013-01-19 19:19:55 +04:00
array ( 'create-partition-tables' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Create required partition tables' ),
array ( 'create-partition-functions' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Create required partition triggers' ),
2013-07-13 23:38:09 +04:00
array ( 'no-partitions' , '' , 0 , 1 , 0 , 0 , 'bool' , " Do not partition search indices (speeds up import of single country extracts) " ),
2012-04-30 04:05:06 +04:00
array ( 'import-wikipedia-articles' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Import wikipedia article dump' ),
2010-10-27 18:05:42 +04:00
array ( 'load-data' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Copy data to live tables from import table' ),
2012-05-22 18:27:42 +04:00
array ( 'disable-token-precalc' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Disable name precalculation (EXPERT)' ),
2010-12-07 16:41:02 +03:00
array ( 'import-tiger-data' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Import tiger data (not included in \'all\')' ),
array ( 'calculate-postcodes' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Calculate postcode centroids' ),
2013-01-19 19:19:55 +04:00
array ( 'create-roads' , '' , 0 , 1 , 0 , 0 , 'bool' , '' ),
2011-01-05 17:07:26 +03:00
array ( 'osmosis-init' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Generate default osmosis configuration' ),
2010-12-07 16:41:02 +03:00
array ( 'index' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Index the data' ),
2012-07-10 17:36:17 +04:00
array ( 'index-noanalyse' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Do not perform analyse operations during index (EXPERT)' ),
2011-01-05 17:07:26 +03:00
array ( 'index-output' , '' , 0 , 1 , 1 , 1 , 'string' , 'File to dump index information to' ),
2012-02-11 01:01:35 +04:00
array ( 'create-search-indices' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Create additional indices required for search and update' ),
2011-04-06 22:19:40 +04:00
array ( 'create-website' , '' , 0 , 1 , 1 , 1 , 'realpath' , 'Create symlinks to setup web directory' ),
2010-10-24 03:12:37 +04:00
);
getCmdOpt ( $_SERVER [ 'argv' ], $aCMDOptions , $aCMDResult , true , true );
2010-10-26 19:22:41 +04:00
$bDidSomething = false ;
2012-08-24 19:55:19 +04:00
// Check if osm-file is set and points to a valid file if --all or --import-data is given
if ( $aCMDResult [ 'import-data' ] || $aCMDResult [ 'all' ])
{
if ( ! isset ( $aCMDResult [ 'osm-file' ]))
{
fail ( 'missing --osm-file for data import' );
}
if ( ! file_exists ( $aCMDResult [ 'osm-file' ]))
{
fail ( 'the path supplied to --osm-file does not exist' );
}
if ( ! is_readable ( $aCMDResult [ 'osm-file' ]))
{
fail ( 'osm-file "' . $aCMDResult [ 'osm-file' ] . '" not readable' );
}
}
2012-07-10 17:36:17 +04:00
// This is a pretty hard core default - the number of processors in the box - 1
2010-11-17 18:19:25 +03:00
$iInstances = isset ( $aCMDResult [ 'threads' ]) ? $aCMDResult [ 'threads' ] : ( getProcessorCount () - 1 );
if ( $iInstances < 1 )
{
$iInstances = 1 ;
echo " WARNING: resetting threads to $iInstances\n " ;
}
if ( $iInstances > getProcessorCount ())
{
$iInstances = getProcessorCount ();
echo " WARNING: resetting threads to $iInstances\n " ;
}
2012-04-26 04:35:06 +04:00
// Assume we can steal all the cache memory in the box (unless told otherwise)
$iCacheMemory = ( isset ( $aCMDResult [ 'osm2pgsql-cache' ]) ? $aCMDResult [ 'osm2pgsql-cache' ] : getCacheMemoryMB ());
if ( $iCacheMemory > getTotalMemoryMB ())
{
$iCacheMemory = getCacheMemoryMB ();
echo " WARNING: resetting cache memory to $iCacheMemory\n " ;
}
2011-04-06 22:19:40 +04:00
$aDSNInfo = DB :: parseDSN ( CONST_Database_DSN );
if ( ! isset ( $aDSNInfo [ 'port' ]) || ! $aDSNInfo [ 'port' ]) $aDSNInfo [ 'port' ] = 5432 ;
2010-11-17 18:19:25 +03:00
2014-10-03 23:55:18 +04:00
$fPostgisVersion = ( float ) CONST_Postgis_Version ;
2010-11-15 19:34:27 +03:00
if ( $aCMDResult [ 'create-db' ] || $aCMDResult [ 'all' ])
2010-10-27 18:05:42 +04:00
{
2011-01-05 17:07:26 +03:00
echo " Create DB \n " ;
2010-10-27 18:05:42 +04:00
$bDidSomething = true ;
$oDB =& DB :: connect ( CONST_Database_DSN , false );
if ( ! PEAR :: isError ( $oDB ))
{
2011-04-06 22:19:40 +04:00
fail ( 'database already exists (' . CONST_Database_DSN . ')' );
2010-10-27 18:05:42 +04:00
}
2013-01-15 01:48:52 +04:00
passthruCheckReturn ( 'createdb -E UTF-8 -p ' . $aDSNInfo [ 'port' ] . ' ' . $aDSNInfo [ 'database' ]);
2010-10-27 18:05:42 +04:00
}
2012-08-24 19:55:19 +04:00
if ( $aCMDResult [ 'setup-db' ] || $aCMDResult [ 'all' ])
2010-10-25 16:22:22 +04:00
{
2012-08-24 19:55:19 +04:00
echo " Setup DB \n " ;
2010-10-26 19:22:41 +04:00
$bDidSomething = true ;
2010-10-25 16:22:22 +04:00
// TODO: path detection, detection memory, etc.
2010-10-27 18:05:42 +04:00
$oDB =& getDB ();
2012-12-18 17:09:50 +04:00
$sVersionString = $oDB -> getOne ( 'select version()' );
2014-10-29 23:04:28 +03:00
preg_match ( '#PostgreSQL ([0-9]+)[.]([0-9]+)[^0-9]#' , $sVersionString , $aMatches );
2012-12-18 17:09:50 +04:00
if ( CONST_Postgresql_Version != $aMatches [ 1 ] . '.' . $aMatches [ 2 ])
{
echo " ERROR: PostgreSQL version is not correct. Expected " . CONST_Postgresql_Version . " found " . $aMatches [ 1 ] . '.' . $aMatches [ 2 ] . " \n " ;
exit ;
}
2013-01-15 01:48:52 +04:00
passthru ( 'createlang plpgsql -p ' . $aDSNInfo [ 'port' ] . ' ' . $aDSNInfo [ 'database' ]);
2012-04-26 04:35:06 +04:00
$pgver = ( float ) CONST_Postgresql_Version ;
2012-01-21 16:09:43 +04:00
if ( $pgver < 9.1 ) {
2012-01-21 14:21:42 +04:00
pgsqlRunScriptFile ( CONST_Path_Postgresql_Contrib . '/hstore.sql' );
2012-11-01 05:47:30 +04:00
pgsqlRunScriptFile ( CONST_BasePath . '/sql/hstore_compatability_9_0.sql' );
2012-01-21 14:21:42 +04:00
} else {
pgsqlRunScript ( 'CREATE EXTENSION hstore' );
}
2012-12-18 17:09:50 +04:00
2013-10-11 01:10:18 +04:00
if ( $fPostgisVersion < 2.0 ) {
pgsqlRunScriptFile ( CONST_Path_Postgresql_Postgis . '/postgis.sql' );
pgsqlRunScriptFile ( CONST_Path_Postgresql_Postgis . '/spatial_ref_sys.sql' );
} else {
pgsqlRunScript ( 'CREATE EXTENSION postgis' );
}
2014-02-06 11:04:42 +04:00
if ( $fPostgisVersion < 2.1 ) {
// Function was renamed in 2.1 and throws an annoying deprecation warning
pgsqlRunScript ( 'ALTER FUNCTION st_line_interpolate_point(geometry, double precision) RENAME TO ST_LineInterpolatePoint' );
}
2012-12-18 17:09:50 +04:00
$sVersionString = $oDB -> getOne ( 'select postgis_full_version()' );
2012-12-18 21:39:01 +04:00
preg_match ( '#POSTGIS="([0-9]+)[.]([0-9]+)[.]([0-9]+)( r([0-9]+))?"#' , $sVersionString , $aMatches );
2012-12-18 17:09:50 +04:00
if ( CONST_Postgis_Version != $aMatches [ 1 ] . '.' . $aMatches [ 2 ])
{
echo " ERROR: PostGIS version is not correct. Expected " . CONST_Postgis_Version . " found " . $aMatches [ 1 ] . '.' . $aMatches [ 2 ] . " \n " ;
exit ;
}
2012-05-23 21:25:43 +04:00
pgsqlRunScriptFile ( CONST_BasePath . '/data/country_name.sql' );
2010-12-07 16:41:02 +03:00
pgsqlRunScriptFile ( CONST_BasePath . '/data/country_naturalearthdata.sql' );
2012-05-23 21:25:43 +04:00
pgsqlRunScriptFile ( CONST_BasePath . '/data/country_osm_grid.sql' );
2014-02-22 02:07:41 +04:00
pgsqlRunScriptFile ( CONST_BasePath . '/data/gb_postcode_table.sql' );
if ( file_exists ( CONST_BasePath . '/data/gb_postcode_data.sql.gz' ))
{
pgsqlRunScriptFile ( CONST_BasePath . '/data/gb_postcode_data.sql.gz' );
}
else
{
echo " WARNING: external UK postcode table not found. \n " ;
}
2010-10-27 18:05:42 +04:00
pgsqlRunScriptFile ( CONST_BasePath . '/data/us_statecounty.sql' );
pgsqlRunScriptFile ( CONST_BasePath . '/data/us_state.sql' );
2010-12-07 16:41:02 +03:00
pgsqlRunScriptFile ( CONST_BasePath . '/data/us_postcode.sql' );
2013-07-13 23:38:09 +04:00
if ( $aCMDResult [ 'no-partitions' ])
{
pgsqlRunScript ( 'update country_name set partition = 0' );
}
2014-04-12 15:16:30 +04:00
// the following will be needed by create_functions later but
// is only defined in the subsequently called create_tables.
// Create dummies here that will be overwritten by the proper
// versions in create-tables.
pgsqlRunScript ( 'CREATE TABLE place_boundingbox ()' );
pgsqlRunScript ( 'create type wikipedia_article_match as ()' );
2010-10-27 18:05:42 +04:00
}
2010-11-15 19:34:27 +03:00
if ( $aCMDResult [ 'import-data' ] || $aCMDResult [ 'all' ])
2010-10-27 18:05:42 +04:00
{
2011-01-05 17:07:26 +03:00
echo " Import \n " ;
2010-10-27 18:05:42 +04:00
$bDidSomething = true ;
2011-01-05 17:07:26 +03:00
2012-01-21 14:21:42 +04:00
$osm2pgsql = CONST_Osm2pgsql_Binary ;
2012-05-24 02:26:16 +04:00
if ( ! file_exists ( $osm2pgsql ))
{
echo " Please download and build osm2pgsql. \n If it is already installed, check the path in your local settings (settings/local.php) file. \n " ;
fail ( " osm2pgsql not found in ' $osm2pgsql ' " );
}
2013-05-16 11:03:16 +04:00
2013-05-16 11:11:19 +04:00
if ( ! is_null ( CONST_Osm2pgsql_Flatnode_File ))
{
$osm2pgsql .= ' --flat-nodes ' . CONST_Osm2pgsql_Flatnode_File ;
}
2012-04-12 01:39:07 +04:00
$osm2pgsql .= ' -lsc -O gazetteer --hstore' ;
2012-04-26 04:35:06 +04:00
$osm2pgsql .= ' -C ' . $iCacheMemory ;
2013-01-15 01:48:52 +04:00
$osm2pgsql .= ' -P ' . $aDSNInfo [ 'port' ];
2012-04-12 01:39:07 +04:00
$osm2pgsql .= ' -d ' . $aDSNInfo [ 'database' ] . ' ' . $aCMDResult [ 'osm-file' ];
2012-07-31 02:31:38 +04:00
passthruCheckReturn ( $osm2pgsql );
2010-12-07 16:41:02 +03:00
$oDB =& getDB ();
$x = $oDB -> getRow ( 'select * from place limit 1' );
2012-04-26 15:30:49 +04:00
if ( PEAR :: isError ( $x )) {
fail ( $x -> getMessage ());
}
if ( ! $x ) fail ( 'No Data' );
2010-10-27 18:05:42 +04:00
}
2010-11-15 19:34:27 +03:00
if ( $aCMDResult [ 'create-functions' ] || $aCMDResult [ 'all' ])
2010-10-27 18:05:42 +04:00
{
2011-01-05 17:07:26 +03:00
echo " Functions \n " ;
2010-10-27 18:05:42 +04:00
$bDidSomething = true ;
2010-11-15 19:34:27 +03:00
if ( ! file_exists ( CONST_BasePath . '/module/nominatim.so' )) fail ( " nominatim module not built " );
2010-10-27 18:05:42 +04:00
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/functions.sql' );
2012-05-22 18:27:42 +04:00
$sTemplate = str_replace ( '{modulepath}' , CONST_BasePath . '/module' , $sTemplate );
if ( $aCMDResult [ 'enable-diff-updates' ]) $sTemplate = str_replace ( 'RETURN NEW; -- @DIFFUPDATES@' , '--' , $sTemplate );
if ( $aCMDResult [ 'enable-debug-statements' ]) $sTemplate = str_replace ( '--DEBUG:' , '' , $sTemplate );
2014-02-08 02:37:54 +04:00
if ( CONST_Limit_Reindexing ) $sTemplate = str_replace ( '--LIMIT INDEXING:' , '' , $sTemplate );
2010-10-27 18:05:42 +04:00
pgsqlRunScript ( $sTemplate );
2014-10-03 23:55:18 +04:00
if ( $fPostgisVersion < 2.0 ) {
echo " Helper functions for postgis < 2.0 \n " ;
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/postgis_15_aux.sql' );
} else {
echo " Helper functions for postgis >= 2.0 \n " ;
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/postgis_20_aux.sql' );
}
pgsqlRunScript ( $sTemplate );
2010-10-25 16:22:22 +04:00
}
2011-04-06 22:19:40 +04:00
if ( $aCMDResult [ 'create-minimal-tables' ])
{
echo " Minimal Tables \n " ;
$bDidSomething = true ;
pgsqlRunScriptFile ( CONST_BasePath . '/sql/tables-minimal.sql' );
$sScript = '' ;
// Backstop the import process - easliest possible import id
$sScript .= " insert into import_npi_log values (18022); \n " ;
$hFile = @ fopen ( CONST_BasePath . '/settings/partitionedtags.def' , " r " );
if ( ! $hFile ) fail ( 'unable to open list of partitions: ' . CONST_BasePath . '/settings/partitionedtags.def' );
while (( $sLine = fgets ( $hFile , 4096 )) !== false && $sLine && substr ( $sLine , 0 , 1 ) != '#' )
{
list ( $sClass , $sType ) = explode ( ' ' , trim ( $sLine ));
$sScript .= " create table place_classtype_ " . $sClass . " _ " . $sType . " as " ;
$sScript .= " select place_id as place_id,geometry as centroid from placex limit 0; \n " ;
$sScript .= " CREATE INDEX idx_place_classtype_ " . $sClass . " _ " . $sType . " _centroid " ;
$sScript .= " ON place_classtype_ " . $sClass . " _ " . $sType . " USING GIST (centroid); \n " ;
$sScript .= " CREATE INDEX idx_place_classtype_ " . $sClass . " _ " . $sType . " _place_id " ;
$sScript .= " ON place_classtype_ " . $sClass . " _ " . $sType . " USING btree(place_id); \n " ;
}
fclose ( $hFile );
pgsqlRunScript ( $sScript );
}
2010-11-15 19:34:27 +03:00
if ( $aCMDResult [ 'create-tables' ] || $aCMDResult [ 'all' ])
2010-10-25 16:22:22 +04:00
{
2011-01-05 17:07:26 +03:00
echo " Tables \n " ;
2010-10-26 19:22:41 +04:00
$bDidSomething = true ;
2010-10-27 18:05:42 +04:00
pgsqlRunScriptFile ( CONST_BasePath . '/sql/tables.sql' );
2010-11-01 18:09:10 +03:00
// re-run the functions
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/functions.sql' );
$sTemplate = str_replace ( '{modulepath}' , CONST_BasePath . '/module' , $sTemplate );
pgsqlRunScript ( $sTemplate );
2010-10-25 16:22:22 +04:00
}
2013-01-19 19:19:55 +04:00
if ( $aCMDResult [ 'create-partition-tables' ] || $aCMDResult [ 'all' ])
2010-10-24 03:12:37 +04:00
{
2013-01-19 19:19:55 +04:00
echo " Partition Tables \n " ;
2010-10-26 19:22:41 +04:00
$bDidSomething = true ;
2010-10-27 18:05:42 +04:00
$oDB =& getDB ();
2013-07-13 23:38:09 +04:00
$sSQL = 'select distinct partition from country_name' ;
2010-10-24 03:12:37 +04:00
$aPartitions = $oDB -> getCol ( $sSQL );
2010-10-25 16:22:22 +04:00
if ( PEAR :: isError ( $aPartitions ))
{
fail ( $aPartitions -> getMessage ());
}
2013-07-13 23:38:09 +04:00
if ( ! $aCMDResult [ 'no-partitions' ]) $aPartitions [] = 0 ;
2010-10-24 03:12:37 +04:00
2013-01-19 19:19:55 +04:00
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/partition-tables.src.sql' );
preg_match_all ( '#^-- start(.*?)^-- end#ms' , $sTemplate , $aMatches , PREG_SET_ORDER );
foreach ( $aMatches as $aMatch )
{
$sResult = '' ;
foreach ( $aPartitions as $sPartitionName )
{
$sResult .= str_replace ( '-partition-' , $sPartitionName , $aMatch [ 1 ]);
}
$sTemplate = str_replace ( $aMatch [ 0 ], $sResult , $sTemplate );
}
pgsqlRunScript ( $sTemplate );
}
if ( $aCMDResult [ 'create-partition-functions' ] || $aCMDResult [ 'all' ])
{
echo " Partition Functions \n " ;
$bDidSomething = true ;
$oDB =& getDB ();
2013-07-13 23:38:09 +04:00
$sSQL = 'select distinct partition from country_name' ;
2013-01-19 19:19:55 +04:00
$aPartitions = $oDB -> getCol ( $sSQL );
if ( PEAR :: isError ( $aPartitions ))
{
fail ( $aPartitions -> getMessage ());
}
2013-07-13 23:38:09 +04:00
if ( ! $aCMDResult [ 'no-partitions' ]) $aPartitions [] = 0 ;
2013-01-19 19:19:55 +04:00
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/partition-functions.src.sql' );
2010-10-24 03:12:37 +04:00
preg_match_all ( '#^-- start(.*?)^-- end#ms' , $sTemplate , $aMatches , PREG_SET_ORDER );
foreach ( $aMatches as $aMatch )
{
$sResult = '' ;
foreach ( $aPartitions as $sPartitionName )
{
$sResult .= str_replace ( '-partition-' , $sPartitionName , $aMatch [ 1 ]);
}
$sTemplate = str_replace ( $aMatch [ 0 ], $sResult , $sTemplate );
}
2010-12-07 16:41:02 +03:00
2010-10-27 18:05:42 +04:00
pgsqlRunScript ( $sTemplate );
}
2012-04-30 04:05:06 +04:00
if ( $aCMDResult [ 'import-wikipedia-articles' ] || $aCMDResult [ 'all' ])
{
$bDidSomething = true ;
$sWikiArticlesFile = CONST_BasePath . '/data/wikipedia_article.sql.bin' ;
$sWikiRedirectsFile = CONST_BasePath . '/data/wikipedia_redirect.sql.bin' ;
if ( file_exists ( $sWikiArticlesFile ))
{
echo " Importing wikipedia articles... " ;
2012-05-11 21:23:44 +04:00
pgsqlRunDropAndRestore ( $sWikiArticlesFile );
2012-04-30 04:05:06 +04:00
echo " ...done \n " ;
}
else
{
echo " WARNING: wikipedia article dump file not found - places will have default importance \n " ;
}
if ( file_exists ( $sWikiRedirectsFile ))
{
echo " Importing wikipedia redirects... " ;
2012-05-11 21:23:44 +04:00
pgsqlRunDropAndRestore ( $sWikiRedirectsFile );
2012-04-30 04:05:06 +04:00
echo " ...done \n " ;
}
else
{
echo " WARNING: wikipedia redirect dump file not found - some place importance values may be missing \n " ;
}
}
2010-11-15 19:34:27 +03:00
if ( $aCMDResult [ 'load-data' ] || $aCMDResult [ 'all' ])
2010-10-27 18:05:42 +04:00
{
2012-09-25 02:36:34 +04:00
echo " Drop old Data \n " ;
2010-10-27 18:05:42 +04:00
$bDidSomething = true ;
2010-11-01 18:09:10 +03:00
$oDB =& getDB ();
2010-11-09 13:19:36 +03:00
if ( ! pg_query ( $oDB -> connection , 'TRUNCATE word' )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
2010-11-01 18:09:10 +03:00
if ( ! pg_query ( $oDB -> connection , 'TRUNCATE placex' )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
if ( ! pg_query ( $oDB -> connection , 'TRUNCATE place_addressline' )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
2010-11-09 13:19:36 +03:00
if ( ! pg_query ( $oDB -> connection , 'TRUNCATE place_boundingbox' )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
2010-11-01 18:09:10 +03:00
if ( ! pg_query ( $oDB -> connection , 'TRUNCATE location_area' )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
2010-11-09 13:19:36 +03:00
if ( ! pg_query ( $oDB -> connection , 'TRUNCATE search_name' )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
if ( ! pg_query ( $oDB -> connection , 'TRUNCATE search_name_blank' )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
2010-11-01 18:09:10 +03:00
if ( ! pg_query ( $oDB -> connection , 'DROP SEQUENCE seq_place' )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
if ( ! pg_query ( $oDB -> connection , 'CREATE SEQUENCE seq_place start 100000' )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
2013-07-13 23:38:09 +04:00
$sSQL = 'select distinct partition from country_name' ;
2012-05-24 03:42:59 +04:00
$aPartitions = $oDB -> getCol ( $sSQL );
if ( PEAR :: isError ( $aPartitions ))
{
fail ( $aPartitions -> getMessage ());
}
2013-07-13 23:38:09 +04:00
if ( ! $aCMDResult [ 'no-partitions' ]) $aPartitions [] = 0 ;
2012-05-24 03:42:59 +04:00
foreach ( $aPartitions as $sPartition )
{
if ( ! pg_query ( $oDB -> connection , 'TRUNCATE location_road_' . $sPartition )) fail ( pg_last_error ( $oDB -> connection ));
echo '.' ;
}
2012-09-25 02:36:34 +04:00
// used by getorcreate_word_id to ignore frequent partial words
if ( ! pg_query ( $oDB -> connection , 'CREATE OR REPLACE FUNCTION get_maxwordfreq() RETURNS integer AS $$ SELECT ' . CONST_Max_Word_Frequency . ' as maxwordfreq; $$ LANGUAGE SQL IMMUTABLE' )) fail ( pg_last_error ( $oDB -> connection ));
echo " . \n " ;
2012-05-13 11:15:56 +04:00
// pre-create the word list
2012-05-22 18:27:42 +04:00
if ( ! $aCMDResult [ 'disable-token-precalc' ])
{
2012-09-25 02:36:34 +04:00
echo " Loading word list \n " ;
pgsqlRunScriptFile ( CONST_BasePath . '/data/words.sql' );
2012-05-22 18:27:42 +04:00
}
2012-05-13 11:15:56 +04:00
2012-09-25 02:36:34 +04:00
echo " Load Data \n " ;
2010-11-01 18:09:10 +03:00
$aDBInstances = array ();
for ( $i = 0 ; $i < $iInstances ; $i ++ )
{
$aDBInstances [ $i ] =& getDB ( true );
$sSQL = 'insert into placex (osm_type, osm_id, class, type, name, admin_level, ' ;
2013-04-27 02:57:18 +04:00
$sSQL .= 'housenumber, street, addr_place, isin, postcode, country_code, extratags, ' ;
2010-11-01 18:09:10 +03:00
$sSQL .= 'geometry) select * from place where osm_id % ' . $iInstances . ' = ' . $i ;
2010-11-15 19:34:27 +03:00
if ( $aCMDResult [ 'verbose' ]) echo " $sSQL\n " ;
2010-11-01 18:09:10 +03:00
if ( ! pg_send_query ( $aDBInstances [ $i ] -> connection , $sSQL )) fail ( pg_last_error ( $oDB -> connection ));
}
$bAnyBusy = true ;
while ( $bAnyBusy )
{
$bAnyBusy = false ;
for ( $i = 0 ; $i < $iInstances ; $i ++ )
{
if ( pg_connection_busy ( $aDBInstances [ $i ] -> connection )) $bAnyBusy = true ;
}
sleep ( 1 );
echo '.' ;
}
echo " \n " ;
2012-02-11 01:01:35 +04:00
echo " Reanalysing database... \n " ;
pgsqlRunScript ( 'ANALYSE' );
2010-10-24 03:12:37 +04:00
}
2011-01-05 17:07:26 +03:00
if ( $aCMDResult [ 'create-roads' ])
{
$bDidSomething = true ;
$oDB =& getDB ();
$aDBInstances = array ();
for ( $i = 0 ; $i < $iInstances ; $i ++ )
{
$aDBInstances [ $i ] =& getDB ( true );
if ( ! pg_query ( $aDBInstances [ $i ] -> connection , 'set enable_bitmapscan = off' )) fail ( pg_last_error ( $oDB -> connection ));
2012-10-10 00:05:37 +04:00
$sSQL = 'select count(*) from (select insertLocationRoad(partition, place_id, calculated_country_code, geometry) from ' ;
2011-01-05 17:07:26 +03:00
$sSQL .= 'placex where osm_id % ' . $iInstances . ' = ' . $i . ' and rank_search between 26 and 27 and class = \'highway\') as x ' ;
if ( $aCMDResult [ 'verbose' ]) echo " $sSQL\n " ;
if ( ! pg_send_query ( $aDBInstances [ $i ] -> connection , $sSQL )) fail ( pg_last_error ( $oDB -> connection ));
}
$bAnyBusy = true ;
while ( $bAnyBusy )
{
$bAnyBusy = false ;
for ( $i = 0 ; $i < $iInstances ; $i ++ )
{
if ( pg_connection_busy ( $aDBInstances [ $i ] -> connection )) $bAnyBusy = true ;
}
sleep ( 1 );
echo '.' ;
}
echo " \n " ;
}
2010-12-07 16:41:02 +03:00
if ( $aCMDResult [ 'import-tiger-data' ])
2010-11-17 18:19:25 +03:00
{
$bDidSomething = true ;
2012-08-18 12:22:35 +04:00
pgsqlRunScriptFile ( CONST_BasePath . '/sql/tiger_import_start.sql' );
2010-11-17 18:19:25 +03:00
$aDBInstances = array ();
for ( $i = 0 ; $i < $iInstances ; $i ++ )
{
$aDBInstances [ $i ] =& getDB ( true );
}
2012-02-18 00:34:28 +04:00
foreach ( glob ( CONST_BasePath . '/data/tiger2011/*.sql' ) as $sFile )
2010-11-17 18:19:25 +03:00
{
echo $sFile . ': ' ;
$hFile = fopen ( $sFile , " r " );
$sSQL = fgets ( $hFile , 100000 );
$iLines = 0 ;
while ( true )
{
for ( $i = 0 ; $i < $iInstances ; $i ++ )
{
if ( ! pg_connection_busy ( $aDBInstances [ $i ] -> connection ))
{
while ( pg_get_result ( $aDBInstances [ $i ] -> connection ));
$sSQL = fgets ( $hFile , 100000 );
if ( ! $sSQL ) break 2 ;
if ( ! pg_send_query ( $aDBInstances [ $i ] -> connection , $sSQL )) fail ( pg_last_error ( $oDB -> connection ));
$iLines ++ ;
if ( $iLines == 1000 )
{
echo " . " ;
$iLines = 0 ;
}
}
}
usleep ( 10 );
}
fclose ( $hFile );
2012-04-30 04:05:06 +04:00
2010-11-17 18:19:25 +03:00
$bAnyBusy = true ;
while ( $bAnyBusy )
{
$bAnyBusy = false ;
for ( $i = 0 ; $i < $iInstances ; $i ++ )
{
if ( pg_connection_busy ( $aDBInstances [ $i ] -> connection )) $bAnyBusy = true ;
}
usleep ( 10 );
}
echo " \n " ;
}
2012-08-18 12:22:35 +04:00
echo " Creating indexes \n " ;
pgsqlRunScriptFile ( CONST_BasePath . '/sql/tiger_import_finish.sql' );
2010-11-17 18:19:25 +03:00
}
2010-12-07 16:41:02 +03:00
if ( $aCMDResult [ 'calculate-postcodes' ] || $aCMDResult [ 'all' ])
{
2011-02-18 12:18:37 +03:00
$bDidSomething = true ;
2010-12-07 16:41:02 +03:00
$oDB =& getDB ();
if ( ! pg_query ( $oDB -> connection , 'DELETE from placex where osm_type=\'P\'' )) fail ( pg_last_error ( $oDB -> connection ));
2012-10-10 00:05:37 +04:00
$sSQL = " insert into placex (osm_type,osm_id,class,type,postcode,calculated_country_code,geometry) " ;
$sSQL .= " select 'P',nextval('seq_postcodes'),'place','postcode',postcode,calculated_country_code, " ;
$sSQL .= " ST_SetSRID(ST_Point(x,y),4326) as geometry from (select calculated_country_code,postcode, " ;
2010-12-07 16:41:02 +03:00
$sSQL .= " avg(st_x(st_centroid(geometry))) as x,avg(st_y(st_centroid(geometry))) as y " ;
2012-10-10 00:05:37 +04:00
$sSQL .= " from placex where postcode is not null group by calculated_country_code,postcode) as x " ;
2010-12-07 16:41:02 +03:00
if ( ! pg_query ( $oDB -> connection , $sSQL )) fail ( pg_last_error ( $oDB -> connection ));
2012-10-10 00:20:05 +04:00
$sSQL = " insert into placex (osm_type,osm_id,class,type,postcode,calculated_country_code,geometry) " ;
2010-12-07 16:41:02 +03:00
$sSQL .= " select 'P',nextval('seq_postcodes'),'place','postcode',postcode,'us', " ;
$sSQL .= " ST_SetSRID(ST_Point(x,y),4326) as geometry from us_postcode " ;
if ( ! pg_query ( $oDB -> connection , $sSQL )) fail ( pg_last_error ( $oDB -> connection ));
}
2012-10-22 06:09:38 +04:00
if ( $aCMDResult [ 'osmosis-init' ] || $aCMDResult [ 'all' ])
2011-01-05 17:07:26 +03:00
{
$bDidSomething = true ;
2012-04-15 19:45:01 +04:00
$oDB =& getDB ();
2011-01-05 17:07:26 +03:00
2013-04-03 20:03:39 +04:00
if ( ! file_exists ( CONST_Osmosis_Binary ))
{
echo " Please download osmosis. \n If it is already installed, check the path in your local settings (settings/local.php) file. \n " ;
2013-09-06 13:58:48 +04:00
if ( ! $aCMDResult [ 'all' ])
{
fail ( " osmosis not found in ' " . CONST_Osmosis_Binary . " ' " );
}
2012-10-22 06:09:38 +04:00
}
2012-09-24 00:03:28 +04:00
else
{
2013-09-06 13:58:48 +04:00
if ( file_exists ( CONST_BasePath . '/settings/configuration.txt' ))
{
echo " settings/configuration.txt already exists \n " ;
}
else
{
passthru ( CONST_Osmosis_Binary . ' --read-replication-interval-init ' . CONST_BasePath . '/settings' );
// update osmosis configuration.txt with our settings
passthru ( " sed -i 's!baseUrl=.*!baseUrl= " . CONST_Replication_Url . " !' " . CONST_BasePath . '/settings/configuration.txt' );
passthru ( " sed -i 's:maxInterval = .*:maxInterval = " . CONST_Replication_MaxInterval . " :' " . CONST_BasePath . '/settings/configuration.txt' );
}
2012-10-22 06:09:38 +04:00
2013-09-06 13:58:48 +04:00
// Find the last node in the DB
$iLastOSMID = $oDB -> getOne ( " select max(id) from planet_osm_nodes " );
// Lookup the timestamp that node was created (less 3 hours for margin for changsets to be closed)
$sLastNodeURL = 'http://www.openstreetmap.org/api/0.6/node/' . $iLastOSMID . " /1 " ;
$sLastNodeXML = file_get_contents ( $sLastNodeURL );
preg_match ( '#timestamp="(([0-9]{4})-([0-9]{2})-([0-9]{2})T([0-9]{2}):([0-9]{2}):([0-9]{2})Z)"#' , $sLastNodeXML , $aLastNodeDate );
$iLastNodeTimestamp = strtotime ( $aLastNodeDate [ 1 ]) - ( 3 * 60 * 60 );
// Search for the correct state file - uses file timestamps so need to sort by date descending
$sRepURL = CONST_Replication_Url . " / " ;
$sRep = file_get_contents ( $sRepURL . " ?C=M;O=D " );
// download.geofabrik.de: <a href="000/">000/</a></td><td align="right">26-Feb-2013 11:53 </td>
// planet.openstreetmap.org: <a href="273/">273/</a> 22-Mar-2013 07:41 -
preg_match_all ( '#<a href="[0-9]{3}/">([0-9]{3}/)</a>.*(([0-9]{2})-([A-z]{3})-([0-9]{4}) ([0-9]{2}):([0-9]{2}))#' , $sRep , $aRepMatches , PREG_SET_ORDER );
$aPrevRepMatch = false ;
foreach ( $aRepMatches as $aRepMatch )
{
if ( strtotime ( $aRepMatch [ 2 ]) < $iLastNodeTimestamp ) break ;
$aPrevRepMatch = $aRepMatch ;
}
if ( $aPrevRepMatch ) $aRepMatch = $aPrevRepMatch ;
2012-10-22 06:09:38 +04:00
2013-09-06 13:58:48 +04:00
$sRepURL .= $aRepMatch [ 1 ];
$sRep = file_get_contents ( $sRepURL . " ?C=M;O=D " );
preg_match_all ( '#<a href="[0-9]{3}/">([0-9]{3}/)</a>.*(([0-9]{2})-([A-z]{3})-([0-9]{4}) ([0-9]{2}):([0-9]{2}))#' , $sRep , $aRepMatches , PREG_SET_ORDER );
$aPrevRepMatch = false ;
foreach ( $aRepMatches as $aRepMatch )
{
if ( strtotime ( $aRepMatch [ 2 ]) < $iLastNodeTimestamp ) break ;
$aPrevRepMatch = $aRepMatch ;
}
if ( $aPrevRepMatch ) $aRepMatch = $aPrevRepMatch ;
2012-10-22 06:09:38 +04:00
2013-09-06 13:58:48 +04:00
$sRepURL .= $aRepMatch [ 1 ];
$sRep = file_get_contents ( $sRepURL . " ?C=M;O=D " );
preg_match_all ( '#<a href="[0-9]{3}.state.txt">([0-9]{3}).state.txt</a>.*(([0-9]{2})-([A-z]{3})-([0-9]{4}) ([0-9]{2}):([0-9]{2}))#' , $sRep , $aRepMatches , PREG_SET_ORDER );
$aPrevRepMatch = false ;
foreach ( $aRepMatches as $aRepMatch )
{
if ( strtotime ( $aRepMatch [ 2 ]) < $iLastNodeTimestamp ) break ;
$aPrevRepMatch = $aRepMatch ;
}
if ( $aPrevRepMatch ) $aRepMatch = $aPrevRepMatch ;
$sRepURL .= $aRepMatch [ 1 ] . '.state.txt' ;
echo " Getting state file: $sRepURL\n " ;
$sStateFile = file_get_contents ( $sRepURL );
if ( ! $sStateFile || strlen ( $sStateFile ) > 1000 ) fail ( " unable to obtain state file " );
file_put_contents ( CONST_BasePath . '/settings/state.txt' , $sStateFile );
echo " Updating DB status \n " ;
pg_query ( $oDB -> connection , 'TRUNCATE import_status' );
$sSQL = " INSERT INTO import_status VALUES(' " . $aRepMatch [ 2 ] . " ') " ;
pg_query ( $oDB -> connection , $sSQL );
2012-10-22 06:09:38 +04:00
}
2011-01-05 17:07:26 +03:00
}
2010-12-07 16:41:02 +03:00
if ( $aCMDResult [ 'index' ] || $aCMDResult [ 'all' ])
{
$bDidSomething = true ;
2011-01-05 17:07:26 +03:00
$sOutputFile = '' ;
if ( isset ( $aCMDResult [ 'index-output' ])) $sOutputFile = ' -F ' . $aCMDResult [ 'index-output' ];
2013-01-15 01:48:52 +04:00
$sBaseCmd = CONST_BasePath . '/nominatim/nominatim -i -d ' . $aDSNInfo [ 'database' ] . ' -P ' . $aDSNInfo [ 'port' ] . ' -t ' . $iInstances . $sOutputFile ;
2012-07-31 02:31:38 +04:00
passthruCheckReturn ( $sBaseCmd . ' -R 4' );
2012-05-22 18:27:42 +04:00
if ( ! $aCMDResult [ 'index-noanalyse' ]) pgsqlRunScript ( 'ANALYSE' );
2012-07-31 02:31:38 +04:00
passthruCheckReturn ( $sBaseCmd . ' -r 5 -R 25' );
2012-05-22 18:27:42 +04:00
if ( ! $aCMDResult [ 'index-noanalyse' ]) pgsqlRunScript ( 'ANALYSE' );
2012-07-31 02:31:38 +04:00
passthruCheckReturn ( $sBaseCmd . ' -r 26' );
2012-02-11 01:01:35 +04:00
}
if ( $aCMDResult [ 'create-search-indices' ] || $aCMDResult [ 'all' ])
{
echo " Search indices \n " ;
$bDidSomething = true ;
$oDB =& getDB ();
2013-07-13 23:38:09 +04:00
$sSQL = 'select distinct partition from country_name' ;
2012-02-11 01:01:35 +04:00
$aPartitions = $oDB -> getCol ( $sSQL );
if ( PEAR :: isError ( $aPartitions ))
{
fail ( $aPartitions -> getMessage ());
}
2013-07-13 23:38:09 +04:00
if ( ! $aCMDResult [ 'no-partitions' ]) $aPartitions [] = 0 ;
2012-02-11 01:01:35 +04:00
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/indices.src.sql' );
preg_match_all ( '#^-- start(.*?)^-- end#ms' , $sTemplate , $aMatches , PREG_SET_ORDER );
foreach ( $aMatches as $aMatch )
{
$sResult = '' ;
foreach ( $aPartitions as $sPartitionName )
{
$sResult .= str_replace ( '-partition-' , $sPartitionName , $aMatch [ 1 ]);
}
$sTemplate = str_replace ( $aMatch [ 0 ], $sResult , $sTemplate );
}
pgsqlRunScript ( $sTemplate );
2011-04-06 22:19:40 +04:00
}
if ( isset ( $aCMDResult [ 'create-website' ]))
{
$bDidSomething = true ;
$sTargetDir = $aCMDResult [ 'create-website' ];
2012-05-24 02:26:16 +04:00
if ( ! is_dir ( $sTargetDir ))
{
echo " You must create the website directory before calling this function. \n " ;
fail ( " Target directory does not exist. " );
}
2011-04-06 22:19:40 +04:00
@ symlink ( CONST_BasePath . '/website/details.php' , $sTargetDir . '/details.php' );
@ symlink ( CONST_BasePath . '/website/reverse.php' , $sTargetDir . '/reverse.php' );
@ symlink ( CONST_BasePath . '/website/search.php' , $sTargetDir . '/search.php' );
@ symlink ( CONST_BasePath . '/website/search.php' , $sTargetDir . '/index.php' );
2012-12-03 18:26:29 +04:00
@ symlink ( CONST_BasePath . '/website/deletable.php' , $sTargetDir . '/deletable.php' );
@ symlink ( CONST_BasePath . '/website/polygons.php' , $sTargetDir . '/polygons.php' );
2013-01-24 16:42:34 +04:00
@ symlink ( CONST_BasePath . '/website/status.php' , $sTargetDir . '/status.php' );
2011-04-06 22:19:40 +04:00
@ symlink ( CONST_BasePath . '/website/images' , $sTargetDir . '/images' );
@ symlink ( CONST_BasePath . '/website/js' , $sTargetDir . '/js' );
2012-11-19 20:00:19 +04:00
@ symlink ( CONST_BasePath . '/website/css' , $sTargetDir . '/css' );
2011-04-06 22:19:40 +04:00
echo " Symlinks created \n " ;
2013-06-13 21:07:19 +04:00
$sTestFile = @ file_get_contents ( CONST_Website_BaseURL . 'js/tiles.js' );
if ( ! $sTestFile )
{
echo " \n WARNING: Unable to access the website at " . CONST_Website_BaseURL . " \n " ;
echo " You may want to update settings/local.php with @define('CONST_Website_BaseURL', 'http://[HOST]/[PATH]/'); \n " ;
}
2010-12-07 16:41:02 +03:00
}
2010-10-26 19:22:41 +04:00
if ( ! $bDidSomething )
{
showUsage ( $aCMDOptions , true );
}
2014-09-29 21:31:37 +04:00
else
{
echo " Setup finished. \n " ;
}
2010-10-27 18:05:42 +04:00
function pgsqlRunScriptFile ( $sFilename )
{
if ( ! file_exists ( $sFilename )) fail ( 'unable to find ' . $sFilename );
2012-07-10 17:36:17 +04:00
// Convert database DSN to psql parameters
2010-10-27 18:05:42 +04:00
$aDSNInfo = DB :: parseDSN ( CONST_Database_DSN );
2012-04-30 04:05:06 +04:00
if ( ! isset ( $aDSNInfo [ 'port' ]) || ! $aDSNInfo [ 'port' ]) $aDSNInfo [ 'port' ] = 5432 ;
2014-02-22 02:07:41 +04:00
$sCMD = 'psql -p ' . $aDSNInfo [ 'port' ] . ' -d ' . $aDSNInfo [ 'database' ];
$ahGzipPipes = null ;
if ( preg_match ( '/\\.gz$/' , $sFilename ))
{
$aDescriptors = array (
0 => array ( 'pipe' , 'r' ),
1 => array ( 'pipe' , 'w' ),
2 => array ( 'file' , '/dev/null' , 'a' )
);
$hGzipProcess = proc_open ( 'zcat ' . $sFilename , $aDescriptors , $ahGzipPipes );
if ( ! is_resource ( $hGzipProcess )) fail ( 'unable to start zcat' );
$aReadPipe = $ahGzipPipes [ 1 ];
fclose ( $ahGzipPipes [ 0 ]);
}
else
{
$sCMD .= ' -f ' . $sFilename ;
$aReadPipe = array ( 'pipe' , 'r' );
}
2010-10-27 18:05:42 +04:00
$aDescriptors = array (
2014-02-22 02:07:41 +04:00
0 => $aReadPipe ,
2010-10-27 18:05:42 +04:00
1 => array ( 'pipe' , 'w' ),
2 => array ( 'file' , '/dev/null' , 'a' )
);
$ahPipes = null ;
$hProcess = proc_open ( $sCMD , $aDescriptors , $ahPipes );
if ( ! is_resource ( $hProcess )) fail ( 'unable to start pgsql' );
// TODO: error checking
while ( ! feof ( $ahPipes [ 1 ]))
{
echo fread ( $ahPipes [ 1 ], 4096 );
}
fclose ( $ahPipes [ 1 ]);
2014-04-12 15:16:30 +04:00
$iReturn = proc_close ( $hProcess );
if ( $iReturn > 0 )
{
fail ( " pgsql returned with error code ( $iReturn ) " );
}
2014-02-22 02:07:41 +04:00
if ( $ahGzipPipes )
{
fclose ( $ahGzipPipes [ 1 ]);
proc_close ( $hGzipProcess );
}
2010-10-27 18:05:42 +04:00
}
function pgsqlRunScript ( $sScript )
{
2014-05-14 01:23:31 +04:00
global $aCMDResult ;
2012-07-10 17:36:17 +04:00
// Convert database DSN to psql parameters
2010-10-27 18:05:42 +04:00
$aDSNInfo = DB :: parseDSN ( CONST_Database_DSN );
2011-02-07 14:13:18 +03:00
if ( ! isset ( $aDSNInfo [ 'port' ]) || ! $aDSNInfo [ 'port' ]) $aDSNInfo [ 'port' ] = 5432 ;
2012-04-30 04:05:06 +04:00
$sCMD = 'psql -p ' . $aDSNInfo [ 'port' ] . ' -d ' . $aDSNInfo [ 'database' ];
2014-05-14 01:23:31 +04:00
if ( ! $aCMDResult [ 'ignore-errors' ])
$sCMD .= ' -v ON_ERROR_STOP=1' ;
2010-10-27 18:05:42 +04:00
$aDescriptors = array (
0 => array ( 'pipe' , 'r' ),
2011-02-21 03:02:17 +03:00
1 => STDOUT ,
2 => STDERR
2010-10-27 18:05:42 +04:00
);
$ahPipes = null ;
2011-06-14 17:42:46 +04:00
$hProcess = @ proc_open ( $sCMD , $aDescriptors , $ahPipes );
2010-10-27 18:05:42 +04:00
if ( ! is_resource ( $hProcess )) fail ( 'unable to start pgsql' );
2011-02-21 03:02:17 +03:00
while ( strlen ( $sScript ))
2010-10-27 18:05:42 +04:00
{
2011-02-21 03:02:17 +03:00
$written = fwrite ( $ahPipes [ 0 ], $sScript );
2014-05-14 01:17:05 +04:00
if ( $written <= 0 ) break ;
2011-02-21 03:02:17 +03:00
$sScript = substr ( $sScript , $written );
2010-10-27 18:05:42 +04:00
}
2011-02-21 03:02:17 +03:00
fclose ( $ahPipes [ 0 ]);
2014-04-12 15:16:30 +04:00
$iReturn = proc_close ( $hProcess );
if ( $iReturn > 0 )
{
fail ( " pgsql returned with error code ( $iReturn ) " );
}
2010-10-27 18:05:42 +04:00
}
2012-04-30 04:05:06 +04:00
function pgsqlRunRestoreData ( $sDumpFile )
{
2012-07-10 17:36:17 +04:00
// Convert database DSN to psql parameters
2012-04-30 04:05:06 +04:00
$aDSNInfo = DB :: parseDSN ( CONST_Database_DSN );
if ( ! isset ( $aDSNInfo [ 'port' ]) || ! $aDSNInfo [ 'port' ]) $aDSNInfo [ 'port' ] = 5432 ;
$sCMD = 'pg_restore -p ' . $aDSNInfo [ 'port' ] . ' -d ' . $aDSNInfo [ 'database' ] . ' -Fc -a ' . $sDumpFile ;
$aDescriptors = array (
0 => array ( 'pipe' , 'r' ),
1 => array ( 'pipe' , 'w' ),
2 => array ( 'file' , '/dev/null' , 'a' )
);
$ahPipes = null ;
$hProcess = proc_open ( $sCMD , $aDescriptors , $ahPipes );
if ( ! is_resource ( $hProcess )) fail ( 'unable to start pg_restore' );
fclose ( $ahPipes [ 0 ]);
// TODO: error checking
while ( ! feof ( $ahPipes [ 1 ]))
{
echo fread ( $ahPipes [ 1 ], 4096 );
}
fclose ( $ahPipes [ 1 ]);
2014-04-12 15:16:30 +04:00
$iReturn = proc_close ( $hProcess );
2012-04-30 04:05:06 +04:00
}
2012-05-11 21:23:44 +04:00
function pgsqlRunDropAndRestore ( $sDumpFile )
{
2012-07-10 17:36:17 +04:00
// Convert database DSN to psql parameters
2012-05-11 21:23:44 +04:00
$aDSNInfo = DB :: parseDSN ( CONST_Database_DSN );
if ( ! isset ( $aDSNInfo [ 'port' ]) || ! $aDSNInfo [ 'port' ]) $aDSNInfo [ 'port' ] = 5432 ;
$sCMD = 'pg_restore -p ' . $aDSNInfo [ 'port' ] . ' -d ' . $aDSNInfo [ 'database' ] . ' -Fc --clean ' . $sDumpFile ;
$aDescriptors = array (
0 => array ( 'pipe' , 'r' ),
1 => array ( 'pipe' , 'w' ),
2 => array ( 'file' , '/dev/null' , 'a' )
);
$ahPipes = null ;
$hProcess = proc_open ( $sCMD , $aDescriptors , $ahPipes );
if ( ! is_resource ( $hProcess )) fail ( 'unable to start pg_restore' );
fclose ( $ahPipes [ 0 ]);
// TODO: error checking
while ( ! feof ( $ahPipes [ 1 ]))
{
echo fread ( $ahPipes [ 1 ], 4096 );
}
fclose ( $ahPipes [ 1 ]);
2014-04-12 15:16:30 +04:00
$iReturn = proc_close ( $hProcess );
2012-05-11 21:23:44 +04:00
}
2012-07-31 02:31:38 +04:00
function passthruCheckReturn ( $cmd )
{
$result = - 1 ;
passthru ( $cmd , $result );
if ( $result != 0 ) fail ( 'Error executing external command: ' . $cmd );
}