Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

Tabi diẹ ninu tetrisology ti a lo.
Ohun gbogbo ti titun ti wa ni daradara gbagbe atijọ.
Epigraphs.
Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

Igbekalẹ iṣoro naa

O ṣe pataki lati ṣe igbasilẹ faili akọọlẹ PostgreSQL lọwọlọwọ lati awọsanma AWS si agbegbe nigbagbogbo Linux Olùgbàlejò. Kì í ṣe ní àkókò gidi, ṣùgbọ́n, ẹ jẹ́ ká sọ, pẹ̀lú ìdádúró díẹ̀.
Akoko igbasilẹ faili log jẹ iṣẹju marun 5.
Faili log, ni AWS, jẹ yiyi ni gbogbo wakati.

Awọn irinṣẹ ti a lo

Lati gbe faili log si agbalejo naa, a lo iwe afọwọkọ bash ti o pe AWS API "aws rds download-db-log-file-ipin».

Awọn aṣayan:

  • --db-apeere-idamọ: Orukọ apẹẹrẹ ni AWS;
  • --log-file-name: orukọ faili log ti ipilẹṣẹ lọwọlọwọ
  • --max-item: Lapapọ nọmba awọn ohun kan ti o pada ni abajade ti aṣẹ naa.Awọn chunk iwọn ti awọn gbaa lati ayelujara faili.
  • --starting-token: Ibẹrẹ ami-ami

Ni ọran yii pato, iṣẹ ṣiṣe ti awọn igbasilẹ igbasilẹ dide lakoko iṣẹ lori ibojuwo iṣẹ ti awọn ibeere PostgreSQL.

Bẹẹni, ati ni irọrun - iṣẹ ti o nifẹ fun ikẹkọ ati ọpọlọpọ lakoko awọn wakati iṣẹ.
Mo ro pe a ti yanju iṣoro naa tẹlẹ nipasẹ ọna ṣiṣe. Ṣugbọn iyara Google ko daba awọn solusan, ati pe ko si ifẹ kan pato lati wa ni ijinle diẹ sii. Ni eyikeyi idiyele, o jẹ adaṣe ti o dara.

Formalization ti awọn iṣẹ-ṣiṣe

Faili log ipari jẹ ṣeto awọn ila ti ipari oniyipada. Ni ayaworan, faili log le jẹ aṣoju bi eleyi:
Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

Ṣe o ti leti ohun kan tẹlẹ bi? Kini o wa pẹlu "tetris"? Ati ki o nibi ohun ti.
Ti a ba ṣe aṣoju awọn aṣayan ti o ṣeeṣe ti o dide nigbati o ba n ṣajọpọ faili atẹle ni ayaworan (fun ayedero, ninu ọran yii, jẹ ki awọn ila ni gigun kanna), a gba. awọn isiro tetris boṣewa:

1) Faili naa ti ṣe igbasilẹ ni gbogbo rẹ ati pe o jẹ ipari. Iwọn chunk naa tobi ju iwọn faili ti o kẹhin lọ:
Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

2) Faili naa ni itesiwaju. Iwọn chunk jẹ kere ju iwọn faili ti o kẹhin lọ:
Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

3) Faili naa jẹ itesiwaju faili ti tẹlẹ ati pe o ni itesiwaju. Iwọn chunk jẹ kere ju iwọn iyokù faili ikẹhin lọ:
Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

4) Faili naa jẹ itesiwaju faili ti tẹlẹ ati pe o jẹ ipari. Iwọn chunk naa tobi ju iwọn iyokù faili ikẹhin lọ:
Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

Iṣẹ-ṣiṣe ni lati ṣajọ onigun mẹta tabi mu Tetris ṣiṣẹ lori ipele tuntun kan.
Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

Awọn iṣoro ti o dide lakoko ti o yanju iṣoro naa

1) Lẹ pọ okun ti awọn ipin 2

Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS
Ni gbogbogbo, ko si awọn iṣoro kan pato. Iṣẹ-ṣiṣe boṣewa lati iṣẹ siseto akọkọ.

Ti o dara ju sìn iwọn

Sugbon yi ni kekere kan diẹ awon.
Laanu, ko si ọna lati lo aiṣedeede lẹhin aami chunk ibere:

Bi o ti mọ tẹlẹ aṣayan --starting-token ni a lo lati pato ibiti o ti bẹrẹ paginating. Aṣayan yii gba awọn iye okun ti yoo tumọ si pe ti o ba gbiyanju lati ṣafikun iye aiṣedeede ni iwaju okun Token Next, aṣayan ko ni gba sinu ero bi aiṣedeede.

Ati nitorinaa, o ni lati ka ni awọn ipin-awọn ipin.
Ti o ba ka ni awọn ipin nla, lẹhinna nọmba awọn kika yoo jẹ iwonba, ṣugbọn iwọn didun yoo pọju.
Ti o ba ka ni awọn ipin kekere, lẹhinna ni ilodi si, nọmba awọn kika yoo jẹ o pọju, ṣugbọn iwọn didun yoo jẹ iwonba.
Nitorinaa, lati le dinku ijabọ ati fun ẹwa gbogbogbo ti ojutu, Mo ni lati wa pẹlu iru ojutu kan, eyiti, laanu, dabi diẹ bi crutch.

Lati ṣapejuwe, jẹ ki a gbero ilana ti igbasilẹ faili log ni awọn ẹya 2 ti o rọrun pupọ. Nọmba awọn kika ni awọn ọran mejeeji da lori iwọn ipin.

1) Fifuye ni awọn ipin kekere:
Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

2) Fifuye ni awọn ipin nla:
Gbigbe akọọlẹ PostgreSQL kan lati inu awọsanma AWS

Gẹgẹbi igbagbogbo, ojutu ti o dara julọ wa ni aarin.
Iwọn ipin jẹ iwonba, ṣugbọn ninu ilana kika, iwọn le pọ si lati dinku nọmba awọn kika.

O yẹ ki o ṣe akiyesi pe iṣoro ti yiyan iwọn to dara julọ ti ipin kika ko tii ti yanju patapata ati pe o nilo ikẹkọ jinlẹ ati itupalẹ. Boya kekere kan nigbamii.

Gbogbogbo apejuwe ti imuse

Awọn tabili iṣẹ ti a lo

CREATE TABLE endpoint
(
id SERIAL ,
host text 
);

TABLE database
(
id SERIAL , 
…
last_aws_log_time text ,
last_aws_nexttoken text ,
aws_max_item_size integer 
);
last_aws_log_time — временная метка последнего загруженного лог-файла в формате YYYY-MM-DD-HH24.
last_aws_nexttoken — текстовая метка последней загруженной порции.
aws_max_item_size- эмпирическим путем, подобранный начальный размер порции.

Full ọrọ ti awọn akosile

download_aws_piece.sh

#!/bin/bash
#########################################################
# download_aws_piece.sh
# downloan piece of log from AWS
# version HABR
 let min_item_size=1024
 let max_item_size=1048576
 let growth_factor=3
 let growth_counter=1
 let growth_counter_max=3

 echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh:''STARTED'
 
 AWS_LOG_TIME=$1
 echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh:AWS_LOG_TIME='$AWS_LOG_TIME
  
 database_id=$2
 echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh:database_id='$database_id
 RESULT_FILE=$3 
  
 endpoint=`psql -h MONITOR_ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE_DATABASE -A -t -c "select e.host from endpoint e join database d on e.id = d.endpoint_id where d.id = $database_id "`
 echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh:endpoint='$endpoint
  
 db_instance=`echo $endpoint | awk -F"." '{print toupper($1)}'`
 
 echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh:db_instance='$db_instance

 LOG_FILE=$RESULT_FILE'.tmp_log'
 TMP_FILE=$LOG_FILE'.tmp'
 TMP_MIDDLE=$LOG_FILE'.tmp_mid'  
 TMP_MIDDLE2=$LOG_FILE'.tmp_mid2'  
  
 current_aws_log_time=`psql -h MONITOR_ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE -A -t -c "select last_aws_log_time from database where id = $database_id "`

 echo $(date +%Y%m%d%H%M)':      download_aws_piece.sh:current_aws_log_time='$current_aws_log_time
  
  if [[ $current_aws_log_time != $AWS_LOG_TIME  ]];
  then
    is_new_log='1'
	if ! psql -h MONITOR_ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE -v ON_ERROR_STOP=1 -A -t -q -c "update database set last_aws_log_time = '$AWS_LOG_TIME' where id = $database_id "
	then
	  echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: FATAL_ERROR - update database set last_aws_log_time .'
	  exit 1
	fi
  else
    is_new_log='0'
  fi
  
  echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh:is_new_log='$is_new_log
  
  let last_aws_max_item_size=`psql -h MONITOR_ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE -A -t -c "select aws_max_item_size from database where id = $database_id "`
  echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: last_aws_max_item_size='$last_aws_max_item_size
  
  let count=1
  if [[ $is_new_log == '1' ]];
  then    
	echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: START DOWNLOADING OF NEW AWS LOG'
	if ! aws rds download-db-log-file-portion 
		--max-items $last_aws_max_item_size 
		--region REGION 
		--db-instance-identifier  $db_instance 
		--log-file-name error/postgresql.log.$AWS_LOG_TIME > $LOG_FILE
	then
		echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: FATAL_ERROR - Could not get log from AWS .'
		exit 2
	fi  	
  else
    next_token=`psql -h MONITOR_ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE -v ON_ERROR_STOP=1 -A -t -c "select last_aws_nexttoken from database where id = $database_id "`
	
	if [[ $next_token == '' ]];
	then
	  next_token='0'	  
	fi
	
	echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: CONTINUE DOWNLOADING OF AWS LOG'
	if ! aws rds download-db-log-file-portion 
	    --max-items $last_aws_max_item_size 
		--starting-token $next_token 
		--region REGION 
		--db-instance-identifier  $db_instance 
		--log-file-name error/postgresql.log.$AWS_LOG_TIME > $LOG_FILE
	then
		echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: FATAL_ERROR - Could not get log from AWS .'
		exit 3
	fi       
	
	line_count=`cat  $LOG_FILE | wc -l`
	let lines=$line_count-1
	  
	tail -$lines $LOG_FILE > $TMP_MIDDLE 
	mv -f $TMP_MIDDLE $LOG_FILE
  fi
  
  next_token_str=`cat $LOG_FILE | grep NEXTTOKEN` 
  next_token=`echo $next_token_str | awk -F" " '{ print $2}' `
  
  grep -v NEXTTOKEN $LOG_FILE  > $TMP_FILE 
  
  if [[ $next_token == '' ]];
  then
	  cp $TMP_FILE $RESULT_FILE
	  
	  echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh:  NEXTTOKEN NOT FOUND - FINISH '
	  rm $LOG_FILE 
	  rm $TMP_FILE
	  rm $TMP_MIDDLE
          rm $TMP_MIDDLE2	  
	  exit 0  
  else
	psql -h MONITOR_ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE -v ON_ERROR_STOP=1 -A -t -q -c "update database set last_aws_nexttoken = '$next_token' where id = $database_id "
  fi
  
  first_str=`tail -1 $TMP_FILE`
  
  line_count=`cat  $TMP_FILE | wc -l`
  let lines=$line_count-1    
  
  head -$lines $TMP_FILE  > $RESULT_FILE

###############################################
# MAIN CIRCLE
  let count=2
  while [[ $next_token != '' ]];
  do 
    echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: count='$count
	
	echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: START DOWNLOADING OF AWS LOG'
	if ! aws rds download-db-log-file-portion 
             --max-items $last_aws_max_item_size 
             --starting-token $next_token 
             --region REGION 
             --db-instance-identifier  $db_instance 
             --log-file-name error/postgresql.log.$AWS_LOG_TIME > $LOG_FILE
	then
		echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: FATAL_ERROR - Could not get log from AWS .'
		exit 4
	fi

	next_token_str=`cat $LOG_FILE | grep NEXTTOKEN` 
	next_token=`echo $next_token_str | awk -F" " '{ print $2}' `

	TMP_FILE=$LOG_FILE'.tmp'
	grep -v NEXTTOKEN $LOG_FILE  > $TMP_FILE  
	
	last_str=`head -1 $TMP_FILE`
  
    if [[ $next_token == '' ]];
	then
	  concat_str=$first_str$last_str
	  	  
	  echo $concat_str >> $RESULT_FILE
		 
	  line_count=`cat  $TMP_FILE | wc -l`
	  let lines=$line_count-1
	  
	  tail -$lines $TMP_FILE >> $RESULT_FILE
	  
	  echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh:  NEXTTOKEN NOT FOUND - FINISH '
	  rm $LOG_FILE 
	  rm $TMP_FILE
	  rm $TMP_MIDDLE
          rm $TMP_MIDDLE2	  
	  exit 0  
	fi
	
    if [[ $next_token != '' ]];
	then
		let growth_counter=$growth_counter+1
		if [[ $growth_counter -gt $growth_counter_max ]];
		then
			let last_aws_max_item_size=$last_aws_max_item_size*$growth_factor
			let growth_counter=1
		fi
	
		if [[ $last_aws_max_item_size -gt $max_item_size ]]; 
		then
			let last_aws_max_item_size=$max_item_size
		fi 

	  psql -h MONITOR_ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE -A -t -q -c "update database set last_aws_nexttoken = '$next_token' where id = $database_id "
	  
	  concat_str=$first_str$last_str
	  	  
	  echo $concat_str >> $RESULT_FILE
		 
	  line_count=`cat  $TMP_FILE | wc -l`
	  let lines=$line_count-1
	  
	  #############################
	  #Get middle of file
	  head -$lines $TMP_FILE > $TMP_MIDDLE
	  
	  line_count=`cat  $TMP_MIDDLE | wc -l`
	  let lines=$line_count-1
	  tail -$lines $TMP_MIDDLE > $TMP_MIDDLE2
	  
	  cat $TMP_MIDDLE2 >> $RESULT_FILE	  
	  
	  first_str=`tail -1 $TMP_FILE`	  
	fi
	  
    let count=$count+1

  done
#
#################################################################

exit 0  

Awọn ajẹkù iwe afọwọkọ pẹlu awọn alaye diẹ:

Awọn paramita igbewọle iwe afọwọkọ:

  • Aami akoko ti orukọ faili log ni ọna kika YYYY-MM-DD-HH24: AWS_LOG_TIME=$1
  • ID aaye data: database_id=$2
  • Orukọ faili akọọlẹ ti a gba: RESULT_FILE=$3

Gba aami akoko ti faili akọọlẹ ti o gbejade kẹhin:

current_aws_log_time=`psql -h MONITOR_ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE -A -t -c "select last_aws_log_time from database where id = $database_id "`

Ti akoko akoko ti faili akọọlẹ ikojọpọ ti o kẹhin ko baamu paramita igbewọle, faili log titun kan ti kojọpọ:

if [[ $current_aws_log_time != $AWS_LOG_TIME  ]];
  then
    is_new_log='1'
	if ! psql -h ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE -v ON_ERROR_STOP=1 -A -t -c "update database set last_aws_log_time = '$AWS_LOG_TIME' where id = $database_id "
	then
	  echo '***download_aws_piece.sh -FATAL_ERROR - update database set last_aws_log_time .'
	  exit 1
	fi
  else
    is_new_log='0'
  fi

A gba iye ti aami atẹle lati faili ti kojọpọ:

  next_token_str=`cat $LOG_FILE | grep NEXTTOKEN` 
  next_token=`echo $next_token_str | awk -F" " '{ print $2}' `

Ami ipari ti igbasilẹ naa jẹ iye ofo ti nexttoken.

Ni lupu kan, a ka awọn ipin ti faili naa, ni ọna, awọn laini isomọ ati jijẹ iwọn ipin naa:
Yipo akọkọ

# MAIN CIRCLE
  let count=2
  while [[ $next_token != '' ]];
  do 
    echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: count='$count
	
	echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: START DOWNLOADING OF AWS LOG'
	if ! aws rds download-db-log-file-portion 
     --max-items $last_aws_max_item_size 
	 --starting-token $next_token 
     --region REGION 
     --db-instance-identifier  $db_instance 
     --log-file-name error/postgresql.log.$AWS_LOG_TIME > $LOG_FILE
	then
		echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh: FATAL_ERROR - Could not get log from AWS .'
		exit 4
	fi

	next_token_str=`cat $LOG_FILE | grep NEXTTOKEN` 
	next_token=`echo $next_token_str | awk -F" " '{ print $2}' `

	TMP_FILE=$LOG_FILE'.tmp'
	grep -v NEXTTOKEN $LOG_FILE  > $TMP_FILE  
	
	last_str=`head -1 $TMP_FILE`
  
    if [[ $next_token == '' ]];
	then
	  concat_str=$first_str$last_str
	  	  
	  echo $concat_str >> $RESULT_FILE
		 
	  line_count=`cat  $TMP_FILE | wc -l`
	  let lines=$line_count-1
	  
	  tail -$lines $TMP_FILE >> $RESULT_FILE
	  
	  echo $(date +%Y%m%d%H%M)':    download_aws_piece.sh:  NEXTTOKEN NOT FOUND - FINISH '
	  rm $LOG_FILE 
	  rm $TMP_FILE
	  rm $TMP_MIDDLE
         rm $TMP_MIDDLE2	  
	  exit 0  
	fi
	
    if [[ $next_token != '' ]];
	then
		let growth_counter=$growth_counter+1
		if [[ $growth_counter -gt $growth_counter_max ]];
		then
			let last_aws_max_item_size=$last_aws_max_item_size*$growth_factor
			let growth_counter=1
		fi
	
		if [[ $last_aws_max_item_size -gt $max_item_size ]]; 
		then
			let last_aws_max_item_size=$max_item_size
		fi 

	  psql -h MONITOR_ENDPOINT.rds.amazonaws.com -U USER -d MONITOR_DATABASE -A -t -q -c "update database set last_aws_nexttoken = '$next_token' where id = $database_id "
	  
	  concat_str=$first_str$last_str
	  	  
	  echo $concat_str >> $RESULT_FILE
		 
	  line_count=`cat  $TMP_FILE | wc -l`
	  let lines=$line_count-1
	  
	  #############################
	  #Get middle of file
	  head -$lines $TMP_FILE > $TMP_MIDDLE
	  
	  line_count=`cat  $TMP_MIDDLE | wc -l`
	  let lines=$line_count-1
	  tail -$lines $TMP_MIDDLE > $TMP_MIDDLE2
	  
	  cat $TMP_MIDDLE2 >> $RESULT_FILE	  
	  
	  first_str=`tail -1 $TMP_FILE`	  
	fi
	  
    let count=$count+1

  done

Kini atẹle?

Nitorinaa, iṣẹ agbedemeji akọkọ - “ṣe igbasilẹ faili log lati awọsanma” ti yanju. Kini lati ṣe pẹlu akọọlẹ ti a gbasile?
Ni akọkọ o nilo lati ṣe itupalẹ faili log ki o jade awọn ibeere gangan lati inu rẹ.
Iṣẹ naa ko nira pupọ. Iwe afọwọkọ bash ti o rọrun julọ ṣe daradara.
upload_log_query.sh

#!/bin/bash
#########################################################
# upload_log_query.sh
# Upload table table from dowloaded aws file 
# version HABR
###########################################################  
echo 'TIMESTAMP:'$(date +%c)' Upload log_query table '
source_file=$1
echo 'source_file='$source_file
database_id=$2
echo 'database_id='$database_id

beginer=' '
first_line='1'
let "line_count=0"
sql_line=' '
sql_flag=' '    
space=' '
cat $source_file | while read line
do
  line="$space$line"

  if [[ $first_line == "1" ]]; then
    beginer=`echo $line | awk -F" " '{ print $1}' `
    first_line='0'
  fi

  current_beginer=`echo $line | awk -F" " '{ print $1}' `

  if [[ $current_beginer == $beginer ]]; then
    if [[ $sql_flag == '1' ]]; then
     sql_flag='0' 
     log_date=`echo $sql_line | awk -F" " '{ print $1}' `
     log_time=`echo $sql_line | awk -F" " '{ print $2}' `
     duration=`echo $sql_line | awk -F" " '{ print $5}' `

     #replace ' to ''
     sql_modline=`echo "$sql_line" | sed 's/'''/''''''/g'`
     sql_line=' '

	 ################
	 #PROCESSING OF THE SQL-SELECT IS HERE
     if ! psql -h ENDPOINT.rds.amazonaws.com -U USER -d DATABASE -v ON_ERROR_STOP=1 -A -t -c "select log_query('$ip_port',$database_id , '$log_date' , '$log_time' , '$duration' , '$sql_modline' )" 
     then
        echo 'FATAL_ERROR - log_query '
        exit 1
     fi
	 ################

    fi #if [[ $sql_flag == '1' ]]; then

    let "line_count=line_count+1"

    check=`echo $line | awk -F" " '{ print $8}' `
    check_sql=${check^^}    

    #echo 'check_sql='$check_sql
    
    if [[ $check_sql == 'SELECT' ]]; then
     sql_flag='1'    
     sql_line="$sql_line$line"
	 ip_port=`echo $sql_line | awk -F":" '{ print $4}' `
    fi
  else       

    if [[ $sql_flag == '1' ]]; then
      sql_line="$sql_line$line"
    fi   
    
  fi #if [[ $current_beginer == $beginer ]]; then

done

Bayi o le ṣiṣẹ pẹlu ibeere ti o jade lati faili log.

Ati pe ọpọlọpọ awọn aye to wulo wa.

Awọn ibeere itusilẹ gbọdọ wa ni ipamọ ni ibikan. Fun eyi, a lo tabili iṣẹ kan. log_query

CREATE TABLE log_query
(
   id SERIAL ,
   queryid bigint ,
   query_md5hash text not null ,
   database_id integer not null ,  
   timepoint timestamp without time zone not null,
   duration double precision not null ,
   query text not null ,
   explained_plan text[],
   plan_md5hash text  , 
   explained_plan_wo_costs text[],
   plan_hash_value text  ,
   baseline_id integer ,
   ip text ,
   port text 
);
ALTER TABLE log_query ADD PRIMARY KEY (id);
ALTER TABLE log_query ADD CONSTRAINT queryid_timepoint_unique_key UNIQUE (queryid, timepoint );
ALTER TABLE log_query ADD CONSTRAINT query_md5hash_timepoint_unique_key UNIQUE (query_md5hash, timepoint );

CREATE INDEX log_query_timepoint_idx ON log_query (timepoint);
CREATE INDEX log_query_queryid_idx ON log_query (queryid);
ALTER TABLE log_query ADD CONSTRAINT database_id_fk FOREIGN KEY (database_id) REFERENCES database (id) ON DELETE CASCADE ;

Ìbéèrè tí a ti ṣe àtúpalẹ̀ ti ni ilọsiwaju ninu plpgsql awọn iṣẹ "log_query».
log_query.sql

--log_query.sql
--verison HABR
CREATE OR REPLACE FUNCTION log_query( ip_port text ,log_database_id integer , log_date text , log_time text , duration text , sql_line text   ) RETURNS boolean AS $$
DECLARE
  result boolean ;
  log_timepoint timestamp without time zone ;
  log_duration double precision ; 
  pos integer ;
  log_query text ;
  activity_string text ;
  log_md5hash text ;
  log_explain_plan text[] ;
  
  log_planhash text ;
  log_plan_wo_costs text[] ; 
  
  database_rec record ;
  
  pg_stat_query text ; 
  test_log_query text ;
  log_query_rec record;
  found_flag boolean;
  
  pg_stat_history_rec record ;
  port_start integer ;
  port_end integer ;
  client_ip text ;
  client_port text ;
  log_queryid bigint ;
  log_query_text text ;
  pg_stat_query_text text ; 
BEGIN
  result = TRUE ;

  RAISE NOTICE '***log_query';
  
  port_start = position('(' in ip_port);
  port_end = position(')' in ip_port);
  client_ip = substring( ip_port from 1 for port_start-1 );
  client_port = substring( ip_port from port_start+1 for port_end-port_start-1 );

  SELECT e.host , d.name , d.owner_pwd 
  INTO database_rec
  FROM database d JOIN endpoint e ON e.id = d.endpoint_id
  WHERE d.id = log_database_id ;
  
  log_timepoint = to_timestamp(log_date||' '||log_time,'YYYY-MM-DD HH24-MI-SS');
  log_duration = duration:: double precision; 

  
  pos = position ('SELECT' in UPPER(sql_line) );
  log_query = substring( sql_line from pos for LENGTH(sql_line));
  log_query = regexp_replace(log_query,' +',' ','g');
  log_query = regexp_replace(log_query,';+','','g');
  log_query = trim(trailing ' ' from log_query);
 

  log_md5hash = md5( log_query::text );
  
  --Explain execution plan--
  EXECUTE 'SELECT dblink_connect(''LINK1'',''host='||database_rec.host||' dbname='||database_rec.name||' user=DATABASE password='||database_rec.owner_pwd||' '')'; 
  
  log_explain_plan = ARRAY ( SELECT * FROM dblink('LINK1', 'EXPLAIN '||log_query ) AS t (plan text) );
  log_plan_wo_costs = ARRAY ( SELECT * FROM dblink('LINK1', 'EXPLAIN ( COSTS FALSE ) '||log_query ) AS t (plan text) );
    
  PERFORM dblink_disconnect('LINK1');
  --------------------------
  BEGIN
	INSERT INTO log_query
	(
		query_md5hash ,
		database_id , 
		timepoint ,
		duration ,
		query ,
		explained_plan ,
		plan_md5hash , 
		explained_plan_wo_costs , 
		plan_hash_value , 
		ip , 
		port
	) 
	VALUES 
	(
		log_md5hash ,
		log_database_id , 
		log_timepoint , 
		log_duration , 
		log_query ,
		log_explain_plan , 
		md5(log_explain_plan::text) ,
		log_plan_wo_costs , 
		md5(log_plan_wo_costs::text),
		client_ip , 
		client_port		
	);
	activity_string = 	'New query has logged '||
						' database_id = '|| log_database_id ||
						' query_md5hash='||log_md5hash||
						' , timepoint = '||to_char(log_timepoint,'YYYYMMDD HH24:MI:SS');
					
	RAISE NOTICE '%',activity_string;					
					 
	PERFORM pg_log( log_database_id , 'log_query' , activity_string);  

	EXCEPTION
	  WHEN unique_violation THEN
		RAISE NOTICE '*** unique_violation *** query already has logged';
	END;

	SELECT 	queryid
	INTO   	log_queryid
	FROM 	log_query 
	WHERE 	query_md5hash = log_md5hash AND
			timepoint = log_timepoint;

	IF log_queryid IS NOT NULL 
	THEN 
	  RAISE NOTICE 'log_query with query_md5hash = % and timepoint = % has already has a QUERYID = %',log_md5hash,log_timepoint , log_queryid ;
	  RETURN result;
	END IF;
	
	------------------------------------------------
	RAISE NOTICE 'Update queryid';	
	
	SELECT * 
	INTO log_query_rec
	FROM log_query
	WHERE query_md5hash = log_md5hash AND timepoint = log_timepoint ; 
	
	log_query_rec.query=regexp_replace(log_query_rec.query,';+','','g');
	
	FOR pg_stat_history_rec IN
	 SELECT 
         queryid ,
	  query 
	 FROM 
         pg_stat_db_queries 
     WHERE  
      database_id = log_database_id AND
       queryid is not null 
	LOOP
	  pg_stat_query = pg_stat_history_rec.query ; 
	  pg_stat_query=regexp_replace(pg_stat_query,'n+',' ','g');
	  pg_stat_query=regexp_replace(pg_stat_query,'t+',' ','g');
	  pg_stat_query=regexp_replace(pg_stat_query,' +',' ','g');
	  pg_stat_query=regexp_replace(pg_stat_query,'$.','%','g');
	
	  log_query_text = trim(trailing ' ' from log_query_rec.query);
	  pg_stat_query_text = pg_stat_query; 
	
	  
	  --SELECT log_query_rec.query like pg_stat_query INTO found_flag ; 
	  IF (log_query_text LIKE pg_stat_query_text) THEN
		found_flag = TRUE ;
	  ELSE
		found_flag = FALSE ;
	  END IF;	  
	  
	  
	  IF found_flag THEN
	    
		UPDATE log_query SET queryid = pg_stat_history_rec.queryid WHERE query_md5hash = log_md5hash AND timepoint = log_timepoint ;
		activity_string = 	' updated queryid = '||pg_stat_history_rec.queryid||
		                    ' for log_query with id = '||log_query_rec.id               
		   				    ;						
	    RAISE NOTICE '%',activity_string;	
		EXIT ;
	  END IF ;
	  
	END LOOP ;
	
  RETURN result ;
END
$$ LANGUAGE plpgsql;

Nigbati o ba n ṣiṣẹ, tabili iṣẹ ni a lo pg_stat_db_queriesA ti o ni aworan ti awọn ibeere lọwọlọwọ ninu tabili pg_stat_itan (Lilo tabili ti wa ni apejuwe nibi - Abojuto iṣẹ ṣiṣe ti awọn ibeere PostgreSQL. Apá 1 - iroyin)

TABLE pg_stat_db_queries
(
   database_id integer,  
   queryid bigint ,  
   query text , 
   max_time double precision 
);

TABLE pg_stat_history 
(
…
database_id integer ,
…
queryid bigint ,
…
max_time double precision	 , 	
…
);

Iṣẹ naa gba ọ laaye lati ṣe nọmba awọn ẹya ti o wulo fun awọn ibeere sisẹ lati faili log kan. Eyun:

Anfani #1 - Ìbéèrè Itan ipaniyan

O wulo pupọ fun ibẹrẹ iṣẹlẹ iṣẹ kan. Ni akọkọ, faramọ itan-akọọlẹ - ati nigbawo ni idinku bẹrẹ?
Lẹhinna, ni ibamu si awọn alailẹgbẹ, wa awọn idi ita. O le jẹ pe fifuye data data ti pọ si pupọ ati pe ibeere kan pato ko ni nkankan lati ṣe pẹlu rẹ.
Ṣafikun titẹsi tuntun si tabili log_query

  port_start = position('(' in ip_port);
  port_end = position(')' in ip_port);
  client_ip = substring( ip_port from 1 for port_start-1 );
  client_port = substring( ip_port from port_start+1 for port_end-port_start-1 );

  SELECT e.host , d.name , d.owner_pwd 
  INTO database_rec
  FROM database d JOIN endpoint e ON e.id = d.endpoint_id
  WHERE d.id = log_database_id ;
  
  log_timepoint = to_timestamp(log_date||' '||log_time,'YYYY-MM-DD HH24-MI-SS');
  log_duration = to_number(duration,'99999999999999999999D9999999999'); 

  
  pos = position ('SELECT' in UPPER(sql_line) );
  log_query = substring( sql_line from pos for LENGTH(sql_line));
  log_query = regexp_replace(log_query,' +',' ','g');
  log_query = regexp_replace(log_query,';+','','g');
  log_query = trim(trailing ' ' from log_query);
 
  RAISE NOTICE 'log_query=%',log_query ;   

  log_md5hash = md5( log_query::text );
  
  --Explain execution plan--
  EXECUTE 'SELECT dblink_connect(''LINK1'',''host='||database_rec.host||' dbname='||database_rec.name||' user=DATABASE password='||database_rec.owner_pwd||' '')'; 
  
  log_explain_plan = ARRAY ( SELECT * FROM dblink('LINK1', 'EXPLAIN '||log_query ) AS t (plan text) );
  log_plan_wo_costs = ARRAY ( SELECT * FROM dblink('LINK1', 'EXPLAIN ( COSTS FALSE ) '||log_query ) AS t (plan text) );
    
  PERFORM dblink_disconnect('LINK1');
  --------------------------
  BEGIN
	INSERT INTO log_query
	(
		query_md5hash ,
		database_id , 
		timepoint ,
		duration ,
		query ,
		explained_plan ,
		plan_md5hash , 
		explained_plan_wo_costs , 
		plan_hash_value , 
		ip , 
		port
	) 
	VALUES 
	(
		log_md5hash ,
		log_database_id , 
		log_timepoint , 
		log_duration , 
		log_query ,
		log_explain_plan , 
		md5(log_explain_plan::text) ,
		log_plan_wo_costs , 
		md5(log_plan_wo_costs::text),
		client_ip , 
		client_port		
	);

Ẹya #2 - Fipamọ Awọn ero Ipaniyan ibeere

Ni aaye yii, asọye-asọye-ọrọ le dide: “Ṣugbọn autoexplain ti wa tẹlẹ". Bẹẹni, o jẹ, ṣugbọn kini aaye ti ero ipaniyan ba wa ni ipamọ sinu faili log kanna ati lati le fipamọ fun itupalẹ siwaju, o ni lati ṣagbejade faili log naa?

Sibẹsibẹ, Mo nilo:
akọkọ: tọju eto ipaniyan ni tabili iṣẹ ti ibi ipamọ data ibojuwo;
keji: lati ni anfani lati ṣe afiwe awọn eto ipaniyan pẹlu ara wọn lati le rii lẹsẹkẹsẹ pe eto ipaniyan ibeere ti yipada.

Ibeere kan pẹlu awọn aye ipaniyan pato wa. Gbigba ati fifipamọ ero ipaniyan rẹ nipa lilo EXPLAIN jẹ iṣẹ-ṣiṣe alakọbẹrẹ.
Pẹlupẹlu, ni lilo alaye EXPLAIN (COSTS FALSE), o le gba ilana ti ero naa, eyiti yoo lo lati gba iye hash ti ero naa, eyiti yoo ṣe iranlọwọ ninu itupalẹ atẹle ti itan-akọọlẹ iyipada ti ero ipaniyan.
Gba awoṣe eto ipaniyan

  --Explain execution plan--
  EXECUTE 'SELECT dblink_connect(''LINK1'',''host='||database_rec.host||' dbname='||database_rec.name||' user=DATABASE password='||database_rec.owner_pwd||' '')'; 
  
  log_explain_plan = ARRAY ( SELECT * FROM dblink('LINK1', 'EXPLAIN '||log_query ) AS t (plan text) );
  log_plan_wo_costs = ARRAY ( SELECT * FROM dblink('LINK1', 'EXPLAIN ( COSTS FALSE ) '||log_query ) AS t (plan text) );
    
  PERFORM dblink_disconnect('LINK1');

Anfani #3 - Lilo Wọle Ibeere fun Abojuto

Niwọn bi a ti tunto awọn metiriki iṣẹ kii ṣe fun ọrọ ibeere, ṣugbọn fun ID rẹ, o nilo lati ṣepọ awọn ibeere lati faili log pẹlu awọn ibeere fun eyiti awọn metiriki iṣẹ ṣe tunto.
O dara, o kere ju lati le ni akoko gangan ti iṣẹlẹ ti iṣẹlẹ iṣẹ kan.

Nitorinaa, nigbati iṣẹlẹ iṣẹ kan ba waye fun ID ibeere, ọna asopọ yoo wa si ibeere kan pato pẹlu awọn iye paramita kan pato ati akoko ipaniyan gangan ati iye akoko ibeere naa. Gba alaye ti a fun ni lilo wiwo nikan pg_stat_gbólóhùn - o jẹ ewọ.
Wa ibeere ibeere naa ki o ṣe imudojuiwọn titẹ sii ninu tabili log_query

SELECT * 
	INTO log_query_rec
	FROM log_query
	WHERE query_md5hash = log_md5hash AND timepoint = log_timepoint ; 
	
	log_query_rec.query=regexp_replace(log_query_rec.query,';+','','g');
	
	FOR pg_stat_history_rec IN
	 SELECT 
      queryid ,
	  query 
	 FROM 
       pg_stat_db_queries 
     WHERE  
	   database_id = log_database_id AND
       queryid is not null 
	LOOP
	  pg_stat_query = pg_stat_history_rec.query ; 
	  pg_stat_query=regexp_replace(pg_stat_query,'n+',' ','g');
	  pg_stat_query=regexp_replace(pg_stat_query,'t+',' ','g');
	  pg_stat_query=regexp_replace(pg_stat_query,' +',' ','g');
	  pg_stat_query=regexp_replace(pg_stat_query,'$.','%','g');
	
	  log_query_text = trim(trailing ' ' from log_query_rec.query);
	  pg_stat_query_text = pg_stat_query; 
	  
	  --SELECT log_query_rec.query like pg_stat_query INTO found_flag ; 
	  IF (log_query_text LIKE pg_stat_query_text) THEN
		found_flag = TRUE ;
	  ELSE
		found_flag = FALSE ;
	  END IF;	  
	  
	  
	  IF found_flag THEN
	    
		UPDATE log_query SET queryid = pg_stat_history_rec.queryid WHERE query_md5hash = log_md5hash AND timepoint = log_timepoint ;
		activity_string = 	' updated queryid = '||pg_stat_history_rec.queryid||
		                    ' for log_query with id = '||log_query_rec.id		                    
		   				    ;						
					
	    RAISE NOTICE '%',activity_string;	
		EXIT ;
	  END IF ;
	  
	END LOOP ;

Lẹhin Ọrọ

Bi abajade, ọna ti a ṣalaye ti rii ohun elo rẹ ninu eto idagbasoke fun mimojuto iṣẹ ti awọn ibeere PostgreSQL, gbigba ọ laaye lati ni alaye diẹ sii fun itupalẹ nigbati o ba yanju awọn iṣẹlẹ iṣẹ ṣiṣe ibeere ti n yọ jade.

Botilẹjẹpe, nitorinaa, ninu ero ti ara ẹni, yoo tun jẹ pataki lati ṣiṣẹ lori algorithm fun yiyan ati yiyipada iwọn ti apakan ti a gbasile. Iṣoro naa ko tii yanju ni ọran gbogbogbo. O yoo jasi jẹ awon.

Ṣugbọn iyẹn jẹ itan ti o yatọ patapata…

orisun: www.habr.com

Ra alejo gbigba igbẹkẹle fun awọn aaye pẹlu aabo DDoS, awọn olupin VPS VDS 🔥 Ra gbigbalejo oju opo wẹẹbu ti o gbẹkẹle pẹlu aabo DDoS, awọn olupin VPS VDS | ProHoster