Facilities/IBatisNet/Castle.Facilities.IBatisNetIntegration/AutomaticSessionInterceptor.cs

   1 #region License
   2
   3 // Copyright 2004-2008 Castle Project - http://www.castleproject.org/
   4 //
   5 // Licensed under the Apache License, Version 2.0 (the "License");
   6 // you may not use this file except in compliance with the License.
   7 // You may obtain a copy of the License at
   8 //
   9 // http://www.apache.org/licenses/LICENSE-2.0
  10 //
  11 // Unless required by applicable law or agreed to in writing, software
  12 // distributed under the License is distributed on an "AS IS" BASIS,
  13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 // See the License for the specific language governing permissions and
  15 // limitations under the License.
  16 //
  17 // --
  18 //
  19 // This facility was a contribution kindly
  20 // donated by Gilles Bayon &lt;gilles.bayon@gmail.com&gt;
  21 //
  22 // --
  23
  24 #endregion
  25
  26 namespace Castle.Facilities.IBatisNetIntegration
  27 {
  28         using System;
  29         using System.Reflection;
  30         using Castle.Core.Interceptor;
  31         using Castle.MicroKernel;
  32         using Castle.Services.Transaction;
  33         using IBatisNet.Common.Logging;
  34         using IBatisNet.DataMapper;
  35         using Transaction = Services.Transaction.ITransaction;
  36
  37         public class AutomaticSessionInterceptor : IInterceptor
  38         {
  39                 private IKernel _kernel = null;
  40                 private static readonly ILog _logger = LogManager.GetLogger(MethodBase.GetCurrentMethod().DeclaringType);
  41
  42                 public AutomaticSessionInterceptor(IKernel kernel)
  43                 {
  44                         _kernel = kernel;
  45                 }
  46
  47                 public void Intercept(IInvocation invocation)
  48                 {
  49                         MethodInfo info = invocation.MethodInvocationTarget;
  50
  51                         if (info.IsDefined(typeof (NoSessionAttribute), true))
  52                         {
  53                                 invocation.Proceed();
  54                                 return;
  55                         }
  56
  57                         String key = ObtainSqlMapKeyFor(info);
  58
  59                         ISqlMapper sqlMap = ObtainSqlMapperFor(key);
  60
  61                         if (sqlMap.IsSessionStarted)
  62                         {
  63                                 invocation.Proceed();
  64                                 return;
  65                         }
  66
  67                         if (_logger.IsDebugEnabled)
  68                         {
  69                                 _logger.Debug("Automatic Open connection on method :" + invocation.Method.Name);
  70                         }
  71
  72                         sqlMap.OpenConnection();
  73
  74                         if (EnlistSessionIfHasTransactionActive(key, sqlMap))
  75                         {
  76                                 invocation.Proceed();
  77                                 return;
  78                         }
  79
  80                         try
  81                         {
  82                                 invocation.Proceed();
  83                                 return;
  84                         }
  85                         finally
  86                         {
  87                                 if (_logger.IsDebugEnabled)
  88                                 {
  89                                         _logger.Debug("Close connection on method :" + invocation.Method.Name);
  90                                 }
  91                                 sqlMap.CloseConnection();
  92                         }
  93                 }
  94
  95                 private bool EnlistSessionIfHasTransactionActive(String key, ISqlMapper sqlMap)
  96                 {
  97                         if (!_kernel.HasComponent(typeof (ITransactionManager)))
  98                         {
  99                                 return false;
 100                         }
 101
 102                         bool enlisted = false;
 103
 104                         if (key == null)
 105                         {
 106                                 key = "iBATIS.DataMapper";
 107                         }
 108
 109                         ITransactionManager manager = (ITransactionManager) _kernel[typeof (ITransactionManager)];
 110
 111                         Transaction transaction = manager.CurrentTransaction;
 112
 113                         if (transaction != null)
 114                         {
 115                                 if (!transaction.Context.Contains(key))
 116                                 {
 117                                         transaction.Context[key] = true;
 118                                         transaction.Enlist(new ResourceSqlMapAdapter(sqlMap.BeginTransaction(false)));
 119                                         transaction.RegisterSynchronization(new SqlMapKeeper(sqlMap));
 120                                         enlisted = true;
 121                                 }
 122                         }
 123
 124                         _kernel.ReleaseComponent(manager);
 125
 126                         return enlisted;
 127                 }
 128
 129                 protected String ObtainSqlMapKeyFor(MethodInfo info)
 130                 {
 131                         String sqlMapID = String.Empty;
 132
 133                         if (info.IsDefined(typeof (SessionAttribute), true))
 134                         {
 135                                 SessionAttribute[] attributs = info.GetCustomAttributes(typeof (SessionAttribute), true) as SessionAttribute[];
 136                                 sqlMapID = attributs[0].SqlMapId;
 137                         }
 138
 139                         return sqlMapID;
 140                 }
 141
 142                 protected ISqlMapper ObtainSqlMapperFor(String key)
 143                 {
 144                         // Use the key specified in the attribute - if any
 145                         if (String.Empty.Equals(key))
 146                         {
 147                                 return (ISqlMapper) _kernel[typeof (ISqlMapper)];
 148                         }
 149                         else
 150                         {
 151                                 return (ISqlMapper) _kernel[key];
 152                         }
 153                 }
 154         }
 155 }