JezK

Edit File: class-db-upgrade.php

<?php defined( 'ABSPATH' ) or die( "you do not have access to this page!" );
if ( ! class_exists( "burst_db_upgrade" ) ) {
	class burst_db_upgrade {
		private static $_this;

public function __construct() {
			if ( isset( self::$_this ) ) {
				wp_die( burst_sprintf( '%s is a singleton class and you cannot create a second instance.',
					get_class( $this ) ) );
			}

self::$_this = $this;
			// actions and filters
			add_action( 'admin_init', array( $this, 'init' ) );
		}

/**
		 * Init the upgrades
		 * - upgrade only if admin is logged in
		 * - only one upgrade at a time
		 *
		 * return void
		 */

public function init() {
			if ( ! burst_admin_logged_in() ) {
				return;
			}
			// check if we need to upgrade
			$db_upgrades = $this->get_db_upgrades();
			// check if all upgrades are done
			$do_upgrade = false;
			foreach ( $db_upgrades as $upgrade ) {
				if ( get_option( "burst_db_upgrade_$upgrade" ) == true ) { // if any upgrade is not done
					$do_upgrade = $upgrade;
					// if we need to upgrade break the loop
					break;
				}
			}
			// only one upgrade at a time
			if ( $do_upgrade === 'bounces' ) {
				$this->upgrade_bounces();
			}
			if ( $do_upgrade === 'goals_remove_columns' ) {
				$this->upgrade_goals_remove_columns();
			}
		}

/**
		 * Get the upgrades
		 *
		 * @return array
		 */

private function get_db_upgrades() {
			return apply_filters( 'burst_db_upgrades', [
				'bounces',
				'goals_remove_columns',
			] );
		}

private function upgrade_bounces() {
			if ( ! burst_admin_logged_in() ) {
				return;
			}
			if ( ! get_option( 'burst_db_upgrade_bounces' ) ) {
				return;
			}

global $wpdb;
			$table_name = $wpdb->prefix . 'burst_statistics';

$sql    = "UPDATE $table_name
					SET bounce = 0
					WHERE
					  (session_id IN (
					    SELECT session_id
					    FROM (
					      SELECT session_id
					      FROM $table_name
					      GROUP BY session_id
					      HAVING COUNT(*) >= 2
					    ) as t
					  ))";

$result = $wpdb->query( $sql );
			if ( $result === false ) {
				if ( WP_DEBUG ) {
					error_log( 'Burst Statistics: db upgrade bounces multiple sessions failed' );
				}
				return;
			}

$sql    = "UPDATE $table_name
					SET bounce = 0
					WHERE bounce = 1 AND time_on_page > 5000";
			$result = $wpdb->query( $sql );

// if query is successful
			if ( $result !== false ) {
				update_option( 'burst_db_upgrade_bounces', false);
			} else {
				if ( WP_DEBUG ) {
					error_log( 'Burst Statistics: db upgrade bounces failed' );
				}
			}
		}

private function upgrade_goals_remove_columns(){
			if ( ! burst_admin_logged_in() ) {
				return;
			}
			if ( ! get_option( 'burst_db_upgrade_goals_remove_columns' ) ) {
				return;
			}

global $wpdb;
			$table_name = $wpdb->prefix . 'burst_goals';

// check if columns exist first
			$columns = $wpdb->get_col( "DESC $table_name", 0 );
			if ( ! in_array( 'event', $columns ) || ! in_array( 'action', $columns ) ) {
				update_option( 'burst_db_upgrade_goals_remove_columns', false);
				return;
			}

// run an sql query to remove the columns `event` and `action`
			$sql    = "ALTER TABLE $table_name
					DROP COLUMN `event`,
					DROP COLUMN `action`";

$remove = $wpdb->query( $sql );

if ( $remove !== false ) {
				update_option( 'burst_db_upgrade_goals_remove_columns', false);
			}
			
		}
	}
}