fix 使用线程安全的map

zhuxiujia · Apr 24, 2019 · c89dfe3 · c89dfe3
1 parent 96b0f69
commit c89dfe3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/GoroutineSessionMap.go b/GoroutineSessionMap.go
@@ -24,3 +24,7 @@ func (it *GoroutineSessionMap) Get(k int64) Session {
 		return nil
 	}
 }
+
+func (it *GoroutineSessionMap) Delete(k int64) {
+	it.m.Delete(k)
+}
diff --git a/TransactionAspectSupport.go b/TransactionAspectSupport.go
@@ -31,7 +31,7 @@ func AopProxyService(service reflect.Value, engine *GoMybatisEngine) {
 				session, err = engine.NewSession(beanName)
 				defer func() {
 					session.Close()
-					engine.GoroutineSessionMap().Put(goroutineID, nil)
+					engine.GoroutineSessionMap().Delete(goroutineID)
 				}()
 				if err != nil {
 					panic(err)