--- loncom/interface/lonmysql.pm 2004/07/21 21:01:04 1.21
+++ loncom/interface/lonmysql.pm 2007/03/12 17:10:02 1.36
@@ -1,7 +1,7 @@
# The LearningOnline Network with CAPA
# MySQL utility functions
#
-# $Id: lonmysql.pm,v 1.21 2004/07/21 21:01:04 matthew Exp $
+# $Id: lonmysql.pm,v 1.36 2007/03/12 17:10:02 albertel Exp $
#
# Copyright Michigan State University Board of Trustees
#
@@ -32,23 +32,38 @@ package Apache::lonmysql;
use strict;
use DBI;
use POSIX qw(strftime mktime);
+use Apache::lonnet;
my $mysqluser;
my $mysqlpassword;
+my $mysqldatabase;
+my %db_config;
sub set_mysql_user_and_password {
# If we are running under Apache and LONCAPA, use the LON-CAPA
# user and password. Otherwise...? ? ? ?
- ($mysqluser,$mysqlpassword) = @_;
+ my ($input_mysqluser,$input_mysqlpassword,$input_mysqldatabase) = @_;
+ if (! defined($mysqldatabase)) {
+ $mysqldatabase = 'loncapa';
+ }
+ if (defined($input_mysqldatabase)) {
+ $mysqldatabase = $input_mysqldatabase;
+ }
if (! defined($mysqluser) || ! defined($mysqlpassword)) {
if (! eval 'require Apache::lonnet();') {
$mysqluser = 'www';
$mysqlpassword = $Apache::lonnet::perlvar{'lonSqlAccess'};
} else {
- $mysqluser = 'fuck';
+ $mysqluser = '';
$mysqlpassword = '';
}
}
+ if (defined($input_mysqluser)) {
+ $mysqluser = $input_mysqluser;
+ }
+ if (defined($input_mysqlpassword)) {
+ $mysqlpassword = $input_mysqlpassword;
+ }
}
######################################################################
@@ -72,14 +87,15 @@ and provide a common interface. The goa
a complete reimplementation of the DBI interface. Instead we try to
make using mysql as painless as possible.
-Each table has a numeric ID that is a parameter to most lonmysql functions.
-The table id is returned by &create_table.
-If you lose the table id, it is lost forever.
-The table names in MySQL correspond to
-$ENV{'user.name'}.'_'.$ENV{'user.domain'}.'_'.$table_id. If the table id
-is non-numeric, it is assumed to be the full name of a table. If you pass
-the table id in a form, you MUST ensure that what you send to lonmysql is
-numeric, otherwise you are opening up all the tables in the MySQL database.
+Each table has a numeric ID that is a parameter to most lonmysql
+functions. The table id is returned by &create_table. If you lose
+the table id, it is lost forever. The table names in MySQL correspond
+to $env{'user.name'}.'_'.$env{'user.domain'}.'_'.$table_id. (With all
+non-word characters removed form user.name and user.domain) If the
+table id is non-numeric, it is assumed to be the full name of a table.
+If you pass the table id in a form, you MUST ensure that what you send
+to lonmysql is numeric, otherwise you are opening up all the tables in
+the MySQL database.
=over 4
@@ -349,7 +365,7 @@ sub connect_to_db {
if (! defined($mysqluser) || ! defined($mysqlpassword)) {
&set_mysql_user_and_password();
}
- if (! ($dbh = DBI->connect("DBI:mysql:loncapa",$mysqluser,$mysqlpassword,
+ if (! ($dbh = DBI->connect("DBI:mysql:$mysqldatabase",$mysqluser,$mysqlpassword,
{ RaiseError=>0,PrintError=>0}))) {
$debugstring = "Unable to connect to loncapa database.";
if (! defined($dbh)) {
@@ -361,6 +377,22 @@ sub connect_to_db {
return undef;
}
$debugstring = "Successfully connected to loncapa database.";
+ # Determine DB configuration
+ undef(%db_config);
+ my $sth = $dbh->prepare("SHOW VARIABLES");
+ $sth->execute();
+ if ($sth->err()) {
+ $debugstring = "Unable to retrieve db config variables";
+ return undef;
+ }
+ foreach my $row (@{$sth->fetchall_arrayref}) {
+ $db_config{$row->[0]} = $row->[1];
+ }
+ #&Apache::lonnet::logthis("MySQL configuration variables");
+ #while (my ($k,$v) = each(%db_config)) {
+ # &Apache::lonnet::logthis(" '$k' => '$v'");
+ #}
+ #
return 1;
}
@@ -540,10 +572,6 @@ sub update_table_info {
#
# Get MySQLs table status information.
#
- my @tabledesc = qw/
- Name Type Row_format Rows Avg_row_length Data_length
- Max_data_length Index_length Data_free Auto_increment
- Create_time Update_time Check_time Create_options Comment /;
my $db_command = "SHOW TABLE STATUS FROM loncapa LIKE '$tablename'";
my $sth = $dbh->prepare($db_command);
$sth->execute();
@@ -553,10 +581,16 @@ sub update_table_info {
&disconnect_from_db();
return undef;
}
+ my @column_name = @{$sth->{NAME}};
#
my @info=$sth->fetchrow_array;
for (my $i=0;$i<= $#info ; $i++) {
- $Tables{$tablename}->{$tabledesc[$i]}= $info[$i];
+ if ($column_name[$i] =~ /^(Create_|Update_|Check_)time$/) {
+ $Tables{$tablename}->{$column_name[$i]}=
+ &unsqltime($info[$i]);
+ } else {
+ $Tables{$tablename}->{$column_name[$i]}= $info[$i];
+ }
}
#
# Determine the column order
@@ -580,6 +614,29 @@ sub update_table_info {
$debugstring = "Retrieved table info for $tablename";
return 1;
}
+
+###############################
+
+=pod
+
+=item &table_information()
+
+Inputs: table id
+
+Returns: hash with the table status
+
+=cut
+
+###############################
+sub table_information {
+ my $table_id=shift;
+ if (&update_table_info($table_id)) {
+ return %{$Tables{$table_id}};
+ } else {
+ return ();
+ }
+}
+
###############################
=pod
@@ -592,7 +649,7 @@ Returns: array with column order
=cut
-
+###############################
sub col_order {
my $table_id=shift;
if (&update_table_info($table_id)) {
@@ -619,7 +676,7 @@ Returns:
sub create_table {
return undef if (!defined(&connect_to_db($dbh)));
my ($table_des)=@_;
- my $request = &build_table_creation_request($table_des);
+ my ($request,$table_id) = &build_table_creation_request($table_des);
#
# Execute the request to create the table
#############################################
@@ -629,15 +686,6 @@ sub create_table {
$dbh->errstr();
return undef;
}
- #
- # Set up the internal bookkeeping
- #############################################
- my $table_id;
- if (exists($table_des->{'id'})) {
- $table_id = $table_des->{'id'};
- } else {
- $table_id = &get_new_table_id();
- }
my $tablename = &translate_id($table_id);
delete($Tables{$tablename}) if (exists($Tables{$tablename}));
return undef if (! defined(&update_table_info($table_id)));
@@ -703,6 +751,7 @@ sub build_table_creation_request {
$table_id = &get_new_table_id();
}
my $tablename = &translate_id($table_id);
+ &Apache::lonnet::logthis(" hmm $tablename, $table_id, ". $table_des->{'id'});
my $request = "CREATE TABLE IF NOT EXISTS ".$tablename." ";
foreach my $coldata (@{$table_des->{'columns'}}) {
my $column = $coldata->{'name'};
@@ -770,7 +819,26 @@ sub build_table_creation_request {
$request.="COMMENT = 'temporary' ";
}
$request .= "TYPE=MYISAM";
- return $request;
+ return $request,$table_id;
+}
+
+###############################
+
+=pod
+
+=item &get_table_prefix()
+
+returns the cleaned version of user.name and user.domain for us in table names
+
+=cut
+
+###############################
+sub get_table_prefix {
+ my $clean_name = $env{'user.name'};
+ my $clean_domain = $env{'user.domain'};
+ $clean_name =~ s/\W//g;
+ $clean_domain =~ s/\W//g;
+ return $clean_name.'_'.$clean_domain.'_';
}
###############################
@@ -787,8 +855,9 @@ Used internally to prevent table name co
sub get_new_table_id {
my $newid = 0;
my @tables = &tables_in_db();
+ my $prefix = &get_table_prefix();
foreach (@tables) {
- if (/^$ENV{'user.name'}_$ENV{'user.domain'}_(\d+)$/) {
+ if (/^\Q$prefix\E(\d+)$/) {
$newid = $1 if ($1 > $newid);
}
}
@@ -901,6 +970,59 @@ sub store_row {
return 1;
}
+
+###############################
+
+=pod
+
+=item &bulk_store_rows()
+
+Inputs: table id, [columns],[[row data1].[row data2],...]
+
+returns undef on error, 1 on success.
+
+=cut
+
+###############################
+sub bulk_store_rows {
+ my ($table_id,$columns,$rows) = @_;
+ #
+ return undef if (! defined(&connect_to_db()));
+ my $dbh = &get_dbh();
+ return undef if (! defined($dbh));
+ my $table_status = &check_table($table_id);
+ return undef if (! defined($table_status));
+ if (! $table_status) {
+ $errorstring = "table $table_id does not exist.";
+ return undef;
+ }
+ #
+ my $tablename = &translate_id($table_id);
+ #
+ my $request = 'INSERT IGNORE INTO '.$tablename.' ';
+ if (defined($columns) && ref($columns) eq 'ARRAY') {
+ $request .= join(',',@$columns).' ';
+ }
+ if (! defined($rows) || ref($rows) ne 'ARRAY') {
+ $errorstring = "no input rows given.";
+ return undef;
+ }
+ $request .= 'VALUES ';
+ foreach my $row (@$rows) {
+ # avoid doing row stuff here...
+ $request .= '('.join(',',@$row).'),';
+ }
+ $request =~ s/,$//;
+ # $debugstring = "Executed ".$/.$request; # commented out - this is big
+ $dbh->do($request);
+ if ($dbh->err) {
+ $errorstring = 'Attempted '.$/.$request.$/.'Got error '.$dbh->errstr();
+ return undef;
+ }
+ return 1;
+}
+
+
###############################
=pod
@@ -1014,7 +1136,7 @@ sub translate_id {
# id should be a digit. If it is not a digit we assume the given id
# is complete and does not need to be translated.
return $id if ($id =~ /\D/);
- return $ENV{'user.name'}.'_'.$ENV{'user.domain'}.'_'.$id;
+ return &get_table_prefix().$id;
}
###########################################
@@ -1111,7 +1233,22 @@ sub drop_table {
return 1; # if we got here there was no error, so return a 'true' value
}
+##########################################
+
+=pod
+=item fix_table_name
+
+Fixes a table name so that it will work with MySQL.
+
+=cut
+
+##########################################
+sub fix_table_name {
+ my ($name) = @_;
+ $name =~ s/^(\d+[eE]\d+)/_$1/;
+ return $name;
+}
# ---------------------------- convert 'time' format into a datetime sql format