mysql.pm

重新包装过的lxr

# -*- tab-width: 4 perl-indent-level: 4-*- ###############################
#
# $Id: Mysql.pm,v 1.2 2006/04/26 10:53:17 hellwolf Exp $

# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# 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 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., 675 Mass Ave, Cambridge, MA 02139, USA.

package LXR::Index::Mysql;

$CVSID = '$Id: Mysql.pm,v 1.2 2006/04/26 10:53:17 hellwolf Exp $ ';

use strict;
use DBI;
use LXR::Common;

use vars qw(%files %symcache @ISA $prefix);

@ISA = ("LXR::Index");

sub new {
	my ($self, $dbname) = @_;

	$self = bless({}, $self);
	if (defined($config->{dbpass})) {
		$self->{dbh} = DBI->connect($dbname, $config->{dbuser}, $config->{dbpass})
		  || fatal "Can't open connection to database\n";
	} else {
		$self->{dbh} = DBI->connect($dbname, "lxr", $config->{dbpass})
		  || fatal "Can't open connection to database\n";
	}

	if (defined($config->{'dbprefix'})) {
		$prefix = $config->{'dbprefix'};
	} else {
		$prefix = "lxr_";
	}

	%files    = ();
	%symcache = ();

	$self->{files_select} =
	  $self->{dbh}
	  ->prepare("select fileid from ${prefix}files where  filename = ? and  revision = ?");
	$self->{files_insert} =
	  $self->{dbh}
	  ->prepare("insert into ${prefix}files (filename, revision, fileid) values (?, ?, NULL)");

	$self->{symbols_byname} =
	  $self->{dbh}->prepare("select symid from ${prefix}symbols where  symname = ?");
	$self->{symbols_byid} =
	  $self->{dbh}->prepare("select symname from ${prefix}symbols where symid = ?");
	$self->{symbols_insert} =
	  $self->{dbh}->prepare("insert into ${prefix}symbols (symname, symid) values ( ?, NULL)");
	$self->{symbols_remove} =
	  $self->{dbh}->prepare("delete from ${prefix}symbols where symname = ?");

	$self->{indexes_select} =
	  $self->{dbh}->prepare("select f.filename, i.line, d.declaration, i.relsym "
		  . "from ${prefix}symbols s, ${prefix}indexes i, ${prefix}files f, ${prefix}releases r, ${prefix}declarations d "
		  . "where s.symid = i.symid and i.fileid = f.fileid "
		  . "and f.fileid = r.fileid "
		  . "and i.langid = d.langid and i.type = d.declid "
		  . "and  s.symname = ? and  r.release = ?");
	$self->{indexes_insert} =
	  $self->{dbh}->prepare(
		"insert into ${prefix}indexes (symid, fileid, line, langid, type, relsym) values (?, ?, ?, ?, ?, ?)"
	  );

	$self->{releases_select} =
	  $self->{dbh}->prepare("select * from ${prefix}releases where fileid = ? and  `release` = ?");
	$self->{releases_insert} =
	  $self->{dbh}->prepare("insert into ${prefix}releases (fileid, `release`) values (?, ?)");

	$self->{status_get} =
	  $self->{dbh}->prepare("select status from ${prefix}status where fileid = ?");

	$self->{status_insert} = $self->{dbh}->prepare

	  #		("insert into status select ?, 0 except select fileid, 0 from status");
	  ("insert into ${prefix}status (fileid, status) values (?, ?)");

	$self->{status_update} =
	  $self->{dbh}
	  ->prepare("update ${prefix}status set status = ? where fileid = ? and status <= ?");

	$self->{usage_insert} =
	  $self->{dbh}->prepare("insert into ${prefix}useage (fileid, line, symid) values (?, ?, ?)");
	$self->{usage_select} =
	  $self->{dbh}->prepare("select f.filename, u.line "
		  . "from ${prefix}symbols s, ${prefix}files f, ${prefix}releases r, ${prefix}useage u "
		  . "where s.symid = u.symid "
		  . "and f.fileid = u.fileid "
		  . "and u.fileid = r.fileid "
		  . "and s.symname = ? and  r.release = ? "
		  . "order by f.filename");
	$self->{decl_select} =
	  $self->{dbh}->prepare(
		"select declid from ${prefix}declarations where langid = ? and " . "declaration = ?");
	$self->{decl_insert} =
	  $self->{dbh}->prepare(
		"insert into ${prefix}declarations (declid, langid, declaration) values (NULL, ?, ?)");

	$self->{delete_indexes} =
	  $self->{dbh}->prepare("delete from ${prefix}indexes "
		  . "using ${prefix}indexes i, ${prefix}releases r "
		  . "where i.fileid = r.fileid "
		  . "and r.release = ?");
	$self->{delete_useage} =
	  $self->{dbh}->prepare("delete from ${prefix}useage "
		  . "using ${prefix}useage u, ${prefix}releases r "
		  . "where u.fileid = r.fileid "
		  . "and r.release = ?");
	$self->{delete_status} =
	  $self->{dbh}->prepare("delete from ${prefix}status "
		  . "using ${prefix}status s, ${prefix}releases r "
		  . "where s.fileid = r.fileid "
		  . "and r.release = ?");
	$self->{delete_releases} =
	  $self->{dbh}->prepare("delete from ${prefix}releases " . "where `release` = ?");
	$self->{delete_files} =
	  $self->{dbh}->prepare("delete from ${prefix}files "
		  . "using ${prefix}files f, ${prefix}releases r "
		  . "where f.fileid = r.fileid "
		  . "and r.release = ?");

	return $self;
}

sub index {
	my ($self, $symname, $fileid, $line, $langid, $type, $relsym) = @_;

	$self->{indexes_insert}->execute($self->symid($symname),
		$fileid, $line, $langid, $type, $relsym ? $self->symid($relsym) : undef);
}

sub reference {
	my ($self, $symname, $fileid, $line) = @_;

	$self->{usage_insert}->execute($fileid, $line, $self->symid($symname));

}

sub getindex {
	my ($self, $symname, $release) = @_;
	my ($rows, @ret);

	$rows = $self->{indexes_select}->execute("$symname", "$release");

	while ($rows-- > 0) {
		push(@ret, [ $self->{indexes_select}->fetchrow_array ]);
	}

	$self->{indexes_select}->finish();

	map { $$_[3] &&= $self->symname($$_[3]) } @ret;

	return @ret;
}

sub getreference {
	my ($self, $symname, $release) = @_;
	my ($rows, @ret);

	$rows = $self->{usage_select}->execute("$symname", "$release");

	while ($rows-- > 0) {
		push(@ret, [ $self->{usage_select}->fetchrow_array ]);
	}

	$self->{usage_select}->finish();

	return @ret;
}

sub fileid {
	my ($self, $filename, $revision) = @_;
	my ($fileid);

	# CAUTION: $revision is not $release!
	unless (defined($fileid = $files{"$filename\t$revision"})) {
		$self->{files_select}->execute($filename, $revision);
		($fileid) = $self->{files_select}->fetchrow_array();
		unless ($fileid) {
			$self->{files_insert}->execute($filename, $revision);
			$self->{files_select}->execute($filename, $revision);
			($fileid) = $self->{files_select}->fetchrow_array();
		}
		$files{"$filename\t$revision"} = $fileid;
		$self->{files_select}->finish();
	}
	return $fileid;
}

# Indicate that this filerevision is part of this release
sub release {
	my ($self, $fileid, $release) = @_;

	my $rows = $self->{releases_select}->execute($fileid + 0, $release);
	$self->{releases_select}->finish();

	unless ($rows > 0) {
		$self->{releases_insert}->execute($fileid, $release);
		$self->{releases_insert}->finish();
	}
}

sub symid {
	my ($self, $symname) = @_;
	my ($symid);

	$symid = $symcache{$symname};
	unless (defined($symid)) {
		$self->{symbols_byname}->execute($symname);
		($symid) = $self->{symbols_byname}->fetchrow_array();
		$self->{symbols_byname}->finish();
		unless ($symid) {
			$self->{symbols_insert}->execute($symname);

			# Get the id of the new symbol
			$self->{symbols_byname}->execute($symname);
			($symid) = $self->{symbols_byname}->fetchrow_array();
			$self->{symbols_byname}->finish();
		}
		$symcache{$symname} = $symid;
	}

	return $symid;
}

sub symname {
	my ($self, $symid) = @_;
	my ($symname);

	$self->{symbols_byid}->execute($symid + 0);
	($symname) = $self->{symbols_byid}->fetchrow_array();
	$self->{symbols_byid}->finish();

	return $symname;
}

sub issymbol {
	my ($self, $symname) = @_;
	my ($symid);

	$symid = $symcache{$symname};
	unless (defined($symid)) {
		$self->{symbols_byname}->execute($symname);
		($symid) = $self->{symbols_byname}->fetchrow_array();
		$self->{symbols_byname}->finish();
		$symcache{$symname} = $symid;
	}

	return $symid;
}

# If this file has not been indexed earlier return true.  Return false
# if already indexed.
sub toindex {
	my ($self, $fileid) = @_;
	my ($status);

	$self->{status_get}->execute($fileid);
	$status = $self->{status_get}->fetchrow_array();
	$self->{status_get}->finish();

	if (!defined($status)) {
		$self->{status_insert}->execute($fileid + 0, 0);
	}

	return $status == 0;
}

sub setindexed {
	my ($self, $fileid) = @_;
	$self->{status_update}->execute(1, $fileid, 0);
}

sub toreference {
	my ($self, $fileid) = @_;
	my ($status);

	$self->{status_get}->execute($fileid);
	$status = $self->{status_get}->fetchrow_array();
	$self->{status_get}->finish();

	return $status < 2;
}

sub setreferenced {
	my ($self, $fileid) = @_;
	$self->{status_update}->execute(2, $fileid, 1);
}

# This function should be called before parsing each new file,
# if this is not done the too much memory will be used and
# tings will become very slow.
sub empty_cache {
	%symcache = ();
}

sub getdecid {
	my ($self, $lang, $string) = @_;

	my $rows = $self->{decl_select}->execute($lang, $string);
	$self->{decl_select}->finish();

	unless ($rows > 0) {
		$self->{decl_insert}->execute($lang, $string);
	}

	$self->{decl_select}->execute($lang, $string);
	my $id = $self->{decl_select}->fetchrow_array();
	$self->{decl_select}->finish();

	return $id;
}

sub purge {
	my ($self, $version) = @_;

	# we don't delete symbols, because they might be used by other versions
	# so we can end up with unused symbols, but that doesn't cause any problems
	$self->{delete_indexes}->execute($version);
	$self->{delete_useage}->execute($version);
	$self->{delete_status}->execute($version);
	$self->{delete_releases}->execute($version);
	$self->{delete_files}->execute($version);
}

sub DESTROY {
	my ($self) = @_;
	$self->{files_select}    = undef;
	$self->{files_insert}    = undef;
	$self->{symbols_byname}  = undef;
	$self->{symbols_byid}    = undef;
	$self->{symbols_insert}  = undef;
	$self->{indexes_insert}  = undef;
	$self->{releases_insert} = undef;
	$self->{status_insert}   = undef;
	$self->{status_update}   = undef;
	$self->{usage_insert}    = undef;
	$self->{usage_select}    = undef;
	$self->{decl_select}     = undef;
	$self->{decl_insert}     = undef;
	$self->{delete_indexes}  = undef;
	$self->{delete_useage}   = undef;
	$self->{delete_status}   = undef;
	$self->{delete_releases} = undef;
	$self->{delete_files}    = undef;

	if ($self->{dbh}) {
		$self->{dbh}->disconnect();
		$self->{dbh} = undef;
	}
}

1;