Adding conditional check for column addition.

This commit is contained in:
Colin Kuskie 2009-06-26 17:26:18 +00:00
parent 8383fd9935
commit e5a0af1260

View file

@ -49,8 +49,11 @@ finish($session); # this line required
# Describe what our function does
sub addSessionTokenId {
my $session = shift;
print "\tAdding CSRF token to session... " unless $quiet;
$session->db->write(q|ALTER TABLE userSession ADD COLUMN tokenId CHAR(22)|);
print "\tAdding CSRF token to userSession, if needed... " unless $quiet;
my $sth = $session->db->read('describe userSession tokenId');
if (! defined $sth->hashRef) {
$session->db->write(q|ALTER TABLE userSession ADD COLUMN tokenId CHAR(22)|);
}
# and here's our code
print "DONE!\n" unless $quiet;
}