2010-10-24 03:12:37 +04:00
#!/usr/bin/php -Cq
< ? php
2016-03-09 02:07:46 +03:00
require_once ( dirname ( dirname ( __FILE__ )) . '/settings/settings.php' );
require_once ( CONST_BasePath . '/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)' ),
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' ),
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)' ),
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' ),
2015-09-27 12:00:45 +03:00
array ( 'drop' , '' , 0 , 1 , 0 , 0 , 'bool' , 'Drop tables needed for updates, making the database readonly (EXPERIMENTAL)' ),
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)
2015-08-08 19:15:14 +03:00
if ( isset ( $aCMDResult [ 'osm2pgsql-cache' ]))
{
$iCacheMemory = $aCMDResult [ 'osm2pgsql-cache' ];
}
else
2012-04-26 04:35:06 +04:00
{
$iCacheMemory = getCacheMemoryMB ();
}
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
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
2016-03-30 23:48:18 +03:00
$fPostgresVersion = getPostgresVersion ( $oDB );
echo 'Postgres version found: ' . $fPostgresVersion . " \n " ;
if ( $fPostgresVersion < 9.1 )
2012-12-18 17:09:50 +04:00
{
2016-03-30 23:48:18 +03:00
fail ( " Minimum supported version of Postgresql is 9.1. " );
2012-12-18 17:09:50 +04:00
}
2016-03-30 23:48:18 +03:00
pgsqlRunScript ( 'CREATE EXTENSION IF NOT EXISTS hstore' );
pgsqlRunScript ( 'CREATE EXTENSION IF NOT EXISTS postgis' );
2012-12-18 17:09:50 +04:00
2016-03-30 23:48:18 +03:00
// For extratags and namedetails the hstore_to_json converter is
// needed which is only available from Postgresql 9.3+. For older
// versions add a dummy function that returns nothing.
$iNumFunc = $oDB -> getOne ( " select count(*) from pg_proc where proname = 'hstore_to_json' " );
if ( PEAR :: isError ( $iNumFunc ))
{
fail ( " Cannot query stored procedures. " , $iNumFunc );
2013-10-11 01:10:18 +04:00
}
2016-03-30 23:48:18 +03:00
if ( $iNumFunc == 0 )
{
pgsqlRunScript ( " create function hstore_to_json(dummy hstore) returns text AS 'select null::text' language sql immutable " );
echo " WARNING: Postgresql is too old. extratags and namedetails API not available. " ;
2014-02-06 11:04:42 +04:00
}
2016-03-30 23:48:18 +03:00
$fPostgisVersion = getPostgisVersion ( $oDB );
echo 'Postgis version found: ' . $fPostgisVersion . " \n " ;
if ( $fPostgisVersion < 2.1 )
2012-12-18 17:09:50 +04:00
{
2016-03-30 23:48:18 +03:00
// 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
}
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 " ;
}
2016-03-31 00:21:13 +03:00
if ( CONST_Use_Extra_US_Postcodes )
{
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 ;
}
2014-11-27 23:54:09 +03:00
if ( CONST_Tablespace_Osm2pgsql_Data )
$osm2pgsql .= ' --tablespace-slim-data ' . CONST_Tablespace_Osm2pgsql_Data ;
if ( CONST_Tablespace_Osm2pgsql_Index )
$osm2pgsql .= ' --tablespace-slim-index ' . CONST_Tablespace_Osm2pgsql_Index ;
if ( CONST_Tablespace_Place_Data )
$osm2pgsql .= ' --tablespace-main-data ' . CONST_Tablespace_Place_Data ;
if ( CONST_Tablespace_Place_Index )
$osm2pgsql .= ' --tablespace-main-index ' . CONST_Tablespace_Place_Index ;
2015-08-29 14:14:11 +03:00
$osm2pgsql .= ' -lsc -O gazetteer --hstore --number-processes 1' ;
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 ;
2016-02-28 18:42:24 +03:00
if ( ! file_exists ( CONST_InstallPath . '/module/nominatim.so' )) fail ( " nominatim module not built " );
2016-04-23 12:07:12 +03:00
create_sql_functions ( $aCMDResult );
2010-10-25 16:22:22 +04:00
}
2010-11-15 19:34:27 +03:00
if ( $aCMDResult [ 'create-tables' ] || $aCMDResult [ 'all' ])
2010-10-25 16:22:22 +04:00
{
2010-10-26 19:22:41 +04:00
$bDidSomething = true ;
2014-11-27 23:54:09 +03:00
echo " Tables \n " ;
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/tables.sql' );
$sTemplate = str_replace ( '{www-user}' , CONST_Database_Web_User , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:address-data}' ,
CONST_Tablespace_Address_Data , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:address-index}' ,
CONST_Tablespace_Address_Index , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:search-data}' ,
CONST_Tablespace_Search_Data , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:search-index}' ,
CONST_Tablespace_Search_Index , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:aux-data}' ,
CONST_Tablespace_Aux_Data , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:aux-index}' ,
CONST_Tablespace_Aux_Index , $sTemplate );
pgsqlRunScript ( $sTemplate , false );
2010-11-01 18:09:10 +03:00
// re-run the functions
2014-11-27 23:54:09 +03:00
echo " Functions \n " ;
2016-04-23 12:07:12 +03:00
create_sql_functions ( $aCMDResult );
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' );
2014-11-27 23:54:09 +03:00
$sTemplate = replace_tablespace ( '{ts:address-data}' ,
CONST_Tablespace_Address_Data , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:address-index}' ,
CONST_Tablespace_Address_Index , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:search-data}' ,
CONST_Tablespace_Search_Data , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:search-index}' ,
CONST_Tablespace_Search_Index , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:aux-data}' ,
CONST_Tablespace_Aux_Data , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:aux-index}' ,
CONST_Tablespace_Aux_Index , $sTemplate );
2013-01-19 19:19:55 +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 );
}
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
}
2010-12-07 16:41:02 +03:00
if ( $aCMDResult [ 'import-tiger-data' ])
2010-11-17 18:19:25 +03:00
{
$bDidSomething = true ;
2015-01-11 15:44:39 +03:00
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/tiger_import_start.sql' );
$sTemplate = str_replace ( '{www-user}' , CONST_Database_Web_User , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:aux-data}' ,
CONST_Tablespace_Aux_Data , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:aux-index}' ,
CONST_Tablespace_Aux_Index , $sTemplate );
pgsqlRunScript ( $sTemplate , false );
2012-08-18 12:22:35 +04:00
2010-11-17 18:19:25 +03:00
$aDBInstances = array ();
for ( $i = 0 ; $i < $iInstances ; $i ++ )
{
$aDBInstances [ $i ] =& getDB ( true );
}
2015-05-08 20:48:27 +03:00
foreach ( glob ( CONST_Tiger_Data_Path . '/*.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 " ;
2015-01-11 15:44:39 +03:00
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/tiger_import_finish.sql' );
$sTemplate = str_replace ( '{www-user}' , CONST_Database_Web_User , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:aux-data}' ,
CONST_Tablespace_Aux_Data , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:aux-index}' ,
CONST_Tablespace_Aux_Index , $sTemplate );
pgsqlRunScript ( $sTemplate , false );
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 ));
2016-03-31 00:21:13 +03:00
if ( CONST_Use_Extra_US_Postcodes )
{
$sSQL = " insert into placex (osm_type,osm_id,class,type,postcode,calculated_country_code,geometry) " ;
$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 ));
}
2010-12-07 16:41:02 +03:00
}
2015-09-26 02:41:19 +03:00
if ( $aCMDResult [ 'osmosis-init' ] || ( $aCMDResult [ 'all' ] && ! $aCMDResult [ 'drop' ])) // no use doing osmosis-init when dropping update tables
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
{
2016-02-28 18:42:24 +03:00
if ( file_exists ( CONST_InstallPath . '/settings/configuration.txt' ))
2013-09-06 13:58:48 +04:00
{
echo " settings/configuration.txt already exists \n " ;
}
else
{
2016-02-28 18:42:24 +03:00
passthru ( CONST_Osmosis_Binary . ' --read-replication-interval-init ' . CONST_InstallPath . '/settings' );
2013-09-06 13:58:48 +04:00
// update osmosis configuration.txt with our settings
2016-02-28 18:42:24 +03:00
passthru ( " sed -i 's!baseUrl=.*!baseUrl= " . CONST_Replication_Url . " !' " . CONST_InstallPath . '/settings/configuration.txt' );
passthru ( " sed -i 's:maxInterval = .*:maxInterval = " . CONST_Replication_MaxInterval . " :' " . CONST_InstallPath . '/settings/configuration.txt' );
2013-09-06 13:58:48 +04:00
}
2012-10-22 06:09:38 +04:00
2013-09-06 13:58:48 +04:00
// Find the last node in the DB
2015-03-09 23:33:02 +03:00
$iLastOSMID = $oDB -> getOne ( " select max(osm_id) from place where osm_type = 'N' " );
2013-09-06 13:58:48 +04:00
// 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 . " / " ;
2014-11-28 00:23:04 +03:00
$sRep = file_get_contents ( $sRepURL . " ?C=M;O=D;F=1 " );
2013-09-06 13:58:48 +04:00
// download.geofabrik.de: <a href="000/">000/</a></td><td align="right">26-Feb-2013 11:53 </td>
2014-11-28 00:23:04 +03:00
// planet.openstreetmap.org: <a href="273/">273/</a> 2013-03-11 07:41 -
preg_match_all ( '#<a href="[0-9]{3}/">([0-9]{3}/)</a>\s*([-0-9a-zA-Z]+ [0-9]{2}:[0-9]{2})#' , $sRep , $aRepMatches , PREG_SET_ORDER );
if ( $aRepMatches )
2013-09-06 13:58:48 +04:00
{
2014-11-28 00:23:04 +03:00
$aPrevRepMatch = false ;
foreach ( $aRepMatches as $aRepMatch )
{
if ( strtotime ( $aRepMatch [ 2 ]) < $iLastNodeTimestamp ) break ;
$aPrevRepMatch = $aRepMatch ;
}
if ( $aPrevRepMatch ) $aRepMatch = $aPrevRepMatch ;
$sRepURL .= $aRepMatch [ 1 ];
$sRep = file_get_contents ( $sRepURL . " ?C=M;O=D;F=1 " );
preg_match_all ( '#<a href="[0-9]{3}/">([0-9]{3}/)</a>\s*([-0-9a-zA-Z]+ [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 ];
$sRep = file_get_contents ( $sRepURL . " ?C=M;O=D;F=1 " );
preg_match_all ( '#<a href="[0-9]{3}.state.txt">([0-9]{3}).state.txt</a>\s*([-0-9a-zA-Z]+ [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 " );
2016-02-28 18:42:24 +03:00
file_put_contents ( CONST_InstallPath . '/settings/state.txt' , $sStateFile );
2014-11-28 00:23:04 +03:00
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 );
2013-09-06 13:58:48 +04:00
}
2014-11-28 00:23:04 +03:00
else
2013-09-06 13:58:48 +04:00
{
2014-11-28 00:23:04 +03:00
if ( ! $aCMDResult [ 'all' ])
{
fail ( " Cannot read state file directory. " );
}
2013-09-06 13:58:48 +04:00
}
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 = '' ;
2016-02-28 18:42:24 +03:00
$sBaseCmd = CONST_InstallPath . '/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' );
2014-11-27 23:54:09 +03:00
$sTemplate = replace_tablespace ( '{ts:address-index}' ,
CONST_Tablespace_Address_Index , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:search-index}' ,
CONST_Tablespace_Search_Index , $sTemplate );
$sTemplate = replace_tablespace ( '{ts:aux-index}' ,
CONST_Tablespace_Aux_Index , $sTemplate );
2012-02-11 01:01:35 +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 );
}
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. " );
}
2016-03-16 22:02:09 +03:00
@ symlink ( CONST_InstallPath . '/website/details.php' , $sTargetDir . '/details.php' );
@ symlink ( CONST_InstallPath . '/website/reverse.php' , $sTargetDir . '/reverse.php' );
@ symlink ( CONST_InstallPath . '/website/search.php' , $sTargetDir . '/search.php' );
@ symlink ( CONST_InstallPath . '/website/search.php' , $sTargetDir . '/index.php' );
@ symlink ( CONST_InstallPath . '/website/lookup.php' , $sTargetDir . '/lookup.php' );
@ symlink ( CONST_InstallPath . '/website/deletable.php' , $sTargetDir . '/deletable.php' );
@ symlink ( CONST_InstallPath . '/website/polygons.php' , $sTargetDir . '/polygons.php' );
@ symlink ( CONST_InstallPath . '/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
}
2015-09-27 12:09:55 +03:00
if ( $aCMDResult [ 'drop' ])
2015-09-26 02:41:19 +03:00
{
2015-09-27 12:00:45 +03:00
// The implementation is potentially a bit dangerous because it uses
// a positive selection of tables to keep, and deletes everything else.
// Including any tables that the unsuspecting user might have manually
// created. USE AT YOUR OWN PERIL.
2015-09-26 02:41:19 +03:00
$bDidSomething = true ;
// tables we want to keep. everything else goes.
$aKeepTables = array (
" *columns " ,
" import_polygon_* " ,
" import_status " ,
" place_addressline " ,
" location_property* " ,
" placex " ,
" search_name " ,
" seq_* " ,
" word " ,
" query_log " ,
" new_query_log " ,
" gb_postcode " ,
2015-09-27 12:00:45 +03:00
" spatial_ref_sys " ,
" country_name " ,
" place_classtype_* "
2015-09-26 02:41:19 +03:00
);
$oDB =& getDB ();
$aDropTables = array ();
$aHaveTables = $oDB -> getCol ( " SELECT tablename FROM pg_tables WHERE schemaname='public' " );
if ( PEAR :: isError ( $aHaveTables ))
{
fail ( $aPartitions -> getMessage ());
}
foreach ( $aHaveTables as $sTable )
{
$bFound = false ;
foreach ( $aKeepTables as $sKeep )
{
if ( fnmatch ( $sKeep , $sTable ))
{
$bFound = true ;
break ;
}
}
if ( ! $bFound ) array_push ( $aDropTables , $sTable );
}
foreach ( $aDropTables as $sDrop )
{
if ( $aCMDResult [ 'verbose' ]) echo " dropping table $sDrop\n " ;
@ pg_query ( $oDB -> connection , " DROP TABLE $sDrop CASCADE " );
// ignore warnings/errors as they might be caused by a table having
// been deleted already by CASCADE
}
if ( ! is_null ( CONST_Osm2pgsql_Flatnode_File ))
{
if ( $aCMDResult [ 'verbose' ]) echo " deleting " . CONST_Osm2pgsql_Flatnode_File . " \n " ;
unlink ( CONST_Osm2pgsql_Flatnode_File );
}
}
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
}
2014-11-27 23:54:09 +03:00
function pgsqlRunScript ( $sScript , $bfatal = true )
2010-10-27 18:05:42 +04:00
{
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-11-27 23:54:09 +03:00
if ( $bfatal && ! $aCMDResult [ 'ignore-errors' ])
2014-05-14 01:23:31 +04:00
$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 );
2014-11-27 23:54:09 +03:00
if ( $bfatal && $iReturn > 0 )
2014-04-12 15:16:30 +04:00
{
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 );
}
2014-11-27 23:54:09 +03:00
function replace_tablespace ( $sTemplate , $sTablespace , $sSql )
{
if ( $sTablespace )
$sSql = str_replace ( $sTemplate , 'TABLESPACE "' . $sTablespace . '"' ,
$sSql );
else
$sSql = str_replace ( $sTemplate , '' , $sSql );
return $sSql ;
}
2016-04-23 12:07:12 +03:00
function create_sql_functions ( $aCMDResult )
2016-04-21 01:00:15 +03:00
{
$sTemplate = file_get_contents ( CONST_BasePath . '/sql/functions.sql' );
$sTemplate = str_replace ( '{modulepath}' , CONST_InstallPath . '/module' , $sTemplate );
if ( $aCMDResult [ 'enable-diff-updates' ])
{
$sTemplate = str_replace ( 'RETURN NEW; -- %DIFFUPDATES%' , '--' , $sTemplate );
}
if ( $aCMDResult [ 'enable-debug-statements' ])
{
$sTemplate = str_replace ( '--DEBUG:' , '' , $sTemplate );
}
if ( CONST_Limit_Reindexing )
{
$sTemplate = str_replace ( '--LIMIT INDEXING:' , '' , $sTemplate );
}
if ( ! CONST_Use_US_Tiger_Data )
{
$sTemplate = str_replace ( '-- %NOTIGERDATA% ' , '' , $sTemplate );
}
if ( ! CONST_Use_Aux_Location_data )
{
$sTemplate = str_replace ( '-- %NOAUXDATA% ' , '' , $sTemplate );
}
pgsqlRunScript ( $sTemplate );
}