Gecko [ www.jdcpatala.org ]

Name	Size	Permission
My	[ DIR ]	drwxr-xr-x
v1	[ DIR ]	drwxr-xr-x
mtr_cases.pm	38.71 KB	-rw-r--r--
mtr_cases_from_list.pm	3 KB	-rw-r--r--
mtr_gcov.pl	2.43 KB	-rw-r--r--
mtr_gprof.pl	1.71 KB	-rw-r--r--
mtr_io.pl	3.26 KB	-rw-r--r--
mtr_match.pm	2.4 KB	-rw-r--r--
mtr_misc.pl	7.39 KB	-rw-r--r--
mtr_process.pl	4.14 KB	-rw-r--r--
mtr_report.pm	25.85 KB	-rw-r--r--
mtr_results.pm	4.28 KB	-rw-r--r--
mtr_stress.pl	6.46 KB	-rw-r--r--
mtr_unique.pm	3.86 KB	-rw-r--r--

Code Editor : mtr_unique.pm

# -*- cperl -*-

# Copyright (c) 2004, 2022, Oracle and/or its affiliates.
# 
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License, version 2.0,
# as published by the Free Software Foundation.
#
# This program is also distributed with certain software (including
# but not limited to OpenSSL) that is licensed under separate terms,
# as designated in a particular file or component or in included license
# documentation.  The authors of MySQL hereby grant you an additional
# permission to link the program and your derivative works with the
# separately licensed software that they have included with MySQL.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License, version 2.0, for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

package mtr_unique;

use strict;
use Fcntl ':flock';

use base qw(Exporter);
our @EXPORT= qw(mtr_get_unique_id mtr_release_unique_id);

use My::Platform;

sub msg {
 # print "### unique($$) - ", join(" ", @_), "\n";
}

my $dir;

if(!IS_WINDOWS)
{
  $dir= "/tmp/mysql-unique-ids";
}
else
{
  # Try to use machine-wide directory location for unique IDs,
  # $ALLUSERSPROFILE . IF it is not available, fallback to $TEMP
  # which is typically a per-user temporary directory
  if (exists $ENV{'ALLUSERSPROFILE'} && -w $ENV{'ALLUSERSPROFILE'})
  {
    $dir= $ENV{'ALLUSERSPROFILE'}."/mysql-unique-ids";
  }
  else
  {
    $dir= $ENV{'TEMP'}."/mysql-unique-ids";
  }
}

my @mtr_unique_fh;

END
{
  mtr_release_unique_id();
}

#
# Get a unique, numerical ID in a specified range.
#
# If no unique ID within the specified parameters can be
# obtained, return undef.
#
sub mtr_get_unique_id($$$) {
  my ($min, $max, $build_threads_per_thread)= @_;

msg("get $min-$max, $$");

if (scalar @mtr_unique_fh == $build_threads_per_thread)
  {
    die "Can only get $build_threads_per_thread unique id(s) per process!";
  }

# Make sure our ID directory exists
  if (! -d $dir)
  {
    # If there is a file with the reserved
    # directory name, just delete the file.
    if (-e $dir)
    {
      unlink($dir);
    }

mkdir $dir;
    chmod 0777, $dir;

if(! -d $dir)
    {
      die "can't make directory $dir";
    }
  }

my $build_thread= 0;
  while ( $build_thread < $build_threads_per_thread )
  {
    for (my $id= $min; $id <= $max; $id++)
    {
      my $fh;
      open( $fh, ">$dir/$id");
      chmod 0666, "$dir/$id";

# Try to lock the file exclusively. If lock succeeds, we're done.
      if (flock($fh, LOCK_EX|LOCK_NB))
      {
        # Store file handle - we would need it to release the
        # ID (i.e to unlock the file)
        $mtr_unique_fh[$build_thread] = $fh;
        $build_thread= $build_thread + 1;
      }
      else
      {
        # Not able to get a lock on the file, start the search from
        # next id(i.e min+1).
        $min= $min + 1;

for (;$build_thread > 0; $build_thread--)
        {
          if (defined $mtr_unique_fh[$build_thread-1])
          {
            close $mtr_unique_fh[$build_thread-1];
          }
        }

# Close the file opened in the current iterartion.
        close $fh;
        last;
      }

if ($build_thread == $build_threads_per_thread)
      {
        return $id - $build_thread + 1;
      }
    }

return undef if ($min > $max);
  }

return undef;
}

#
# Release a unique ID.
#
sub mtr_release_unique_id()
{
  msg("release $$");

for (my $i= 0; $i <= $#mtr_unique_fh; $i++)
  {
    if (defined $mtr_unique_fh[$i])
    {
      close $mtr_unique_fh[$i];
    }
  }
  @mtr_unique_fh= ();
}

${this.title}

Code Editor : mtr_unique.pm