From 15addf58d75f69d40fa586f859f32c94946ac117 Mon Sep 17 00:00:00 2001
From: miker <miker@9efc2488-bf62-4759-914b-345cdb29e865>
Date: Tue, 1 Mar 2005 22:58:45 +0000
Subject: [PATCH] added extended callback support to sessions

git-svn-id: svn://svn.open-ils.org/OpenSRF/trunk@142 9efc2488-bf62-4759-914b-345cdb29e865
---
 src/perlmods/OpenSRF/AppSession.pm | 25 ++++++++++++++++++++-----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/src/perlmods/OpenSRF/AppSession.pm b/src/perlmods/OpenSRF/AppSession.pm
index eb28008..cf58528 100644
--- a/src/perlmods/OpenSRF/AppSession.pm
+++ b/src/perlmods/OpenSRF/AppSession.pm
@@ -125,7 +125,8 @@ sub server_build {
 	my $self = bless { recv_queue  => [],
 			   request_queue  => [],
 			   requests  => 0,
-			   death_callbacks  => [],
+			   session_data  => {},
+			   callbacks  => { death => [], disconnect => [] },
 			   endpoint    => SERVER,
 			   state       => CONNECTING, 
 			   session_id  => $sess_id,
@@ -139,6 +140,14 @@ sub server_build {
 	return $_CACHE{$sess_id} = $self;
 }
 
+sub session_data {
+	my $self = shift;
+	my ($name, $datum) = @_;
+
+	$self->{session_data}->{$name} = $datum if (defined $datum);
+	return $self->{session_data}->{$name};
+}
+
 sub service { return shift()->{service}; }
 
 sub continue_request {
@@ -340,10 +349,11 @@ sub finish {
 	}
 }
 
-sub register_death_callback {
+sub register_callback {
 	my $self = shift;
+	my $type = shift;
 	my $cb = shift;
-	push @{ $self->{death_callbacks} }, $cb;
+	push @{ $self->{callbacks}{$type} }, $cb;
 }
 
 sub kill_me {
@@ -351,7 +361,7 @@ sub kill_me {
 	if( ! $self->session_id ) { return 0; }
 
 	# run each 'kill_me' callback;
-	for my $sub (@{$self->{death_callbacks}}) {
+	for my $sub (@{$self->{callbacks}{death}}) {
 		$sub->($self);
 	}
 
@@ -373,6 +383,11 @@ sub disconnect {
 		$self->send('DISCONNECT', "") if ($self->endpoint == CLIENT);;
 		$self->state( DISCONNECTED ); 
 	}
+	# run each 'disconnect' callback;
+	for my $sub (@{$self->{callbacks}{disconnect}}) {
+		$sub->($self);
+	}
+
 	$self->reset;
 }
 
@@ -872,7 +887,7 @@ sub respond_complete {
 sub register_death_callback {
 	my $self = shift;
 	my $cb = shift;
-	$self->session->register_death_callback( $cb );
+	$self->session->register_callback( death => $cb );
 }
 
 package OpenSRF::AppSubrequest;
-- 
2.43.2