From 02026368acfe28fdc584b43a9a3bf3176de19498 Mon Sep 17 00:00:00 2001
From: xuxiuxi <xuxiuxi@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期五, 31 三月 2017 17:45:35 +0800
Subject: [PATCH] 

---
 RtspFace/SensetimeFaceAPIWrapper/src/FaceDBPool.cpp |  164 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 164 insertions(+), 0 deletions(-)

diff --git a/RtspFace/SensetimeFaceAPIWrapper/src/FaceDBPool.cpp b/RtspFace/SensetimeFaceAPIWrapper/src/FaceDBPool.cpp
index c53b469..680d1e0 100644
--- a/RtspFace/SensetimeFaceAPIWrapper/src/FaceDBPool.cpp
+++ b/RtspFace/SensetimeFaceAPIWrapper/src/FaceDBPool.cpp
@@ -1,2 +1,166 @@
 #include "FaceDBPool.h"
+#include "../../logger.h"
+#include "faceAPI.h"
 
+#include <pthread.h>
+#include <map>
+
+#define PLP_MUTEX_LOCK(mut,_ret) if (mut != nullptr) {\
+	int ret = pthread_mutex_lock((pthread_mutex_t*)mut); \
+	if(ret != 0) \
+	{ \
+		LOG_ERROR << "pthread_mutex_lock " << #mut <<  ": " << ret << std::endl; \
+		return _ret; \
+	} \
+}
+
+#define PLP_MUTEX_UNLOCK(mut,_ret) if (mut != nullptr) {\
+	int ret = pthread_mutex_unlock((pthread_mutex_t*)mut); \
+	if(ret != 0) \
+	{ \
+		LOG_ERROR << "pthread_mutex_unlock " << #mut <<  ": " << ret << std::endl; \
+		return _ret; \
+	} \
+}
+
+struct MutexLocker
+{
+	pthread_mutex_t* mut;
+	MutexLocker(void* _mut) : mut((pthread_mutex_t*)_mut)
+	{
+		PLP_MUTEX_LOCK(mut,);
+	}
+	~MutexLocker()
+	{
+		PLP_MUTEX_UNLOCK(mut,);
+	}
+};
+
+struct ThreadSafeFaceDB
+{
+	int dbid;
+	pthread_mutex_t db_mutex;
+	FaceDB* api;
+	
+	ThreadSafeFaceDB() : dbid(-1), db_mutex(), api(nullptr)
+	{
+		pthread_mutex_init(&db_mutex, NULL);
+	}
+	
+	~ThreadSafeFaceDB()
+	{
+		pthread_mutex_destroy(&db_mutex);
+	}
+};
+
+typedef std::map<int, ThreadSafeFaceDB*> facedb_map_t;
+
+FaceDBPool::FaceDBPool() : 
+	pool_mutex(nullptr), face_db_map(nullptr)
+{
+	pool_mutex = new pthread_mutex_t;
+	pthread_mutex_init((pthread_mutex_t*)pool_mutex, NULL);
+	
+	face_db_map = new facedb_map_t();
+}
+
+FaceDBPool::~FaceDBPool()
+{
+	facedb_map_t* _face_db_map = (facedb_map_t*)face_db_map;
+	for (facedb_map_t::iterator iter = _face_db_map->begin(); iter != _face_db_map->end(); ++iter)
+	{
+		iter->second->api->finally();
+		delete iter->second->api;
+		delete iter->second;
+	}
+	
+	_face_db_map->clear();
+	
+	pthread_mutex_destroy((pthread_mutex_t*)pool_mutex);
+	delete (pthread_mutex_t*)pool_mutex;
+	pool_mutex = nullptr;
+}
+
+void FaceDBPool::manage(int dbid, FaceDB* db)
+{
+	if (dbid < 0 || db == nullptr)
+		return;
+
+	MutexLocker _ml(pool_mutex);
+	
+	facedb_map_t* _face_db_map = (facedb_map_t*)face_db_map;
+	
+	if (_face_db_map->find(dbid) != _face_db_map->end())
+		return;
+	
+	ThreadSafeFaceDB* tsfdb = new ThreadSafeFaceDB;
+	tsfdb->dbid = dbid;
+	tsfdb->api = db;
+	
+	_face_db_map->insert(std::make_pair(dbid, tsfdb));
+}
+
+void FaceDBPool::unmanage(int dbid)
+{
+	MutexLocker _ml(pool_mutex);
+
+	facedb_map_t* _face_db_map = (facedb_map_t*)face_db_map;
+	facedb_map_t::iterator iter = _face_db_map->find(dbid);
+	if (iter == _face_db_map->end())
+		return;
+	
+	iter->second->api->finally();
+	delete iter->second->api;
+	delete iter->second;
+	
+	_face_db_map->erase(iter);
+}
+
+FaceDB* FaceDBPool::get_free(int dbid)
+{
+	ThreadSafeFaceDB* tsfdb = nullptr;
+	{
+		//avoid dead lock
+		MutexLocker _ml(pool_mutex);
+		
+		facedb_map_t* _face_db_map = (facedb_map_t*)face_db_map;
+		if (_face_db_map->empty())
+			return nullptr;
+		
+		facedb_map_t::iterator iter = _face_db_map->find(dbid);
+		if (iter == _face_db_map->end())
+			return nullptr;
+		
+		tsfdb = iter->second;
+	}
+
+	if (tsfdb != nullptr)
+	{
+		PLP_MUTEX_LOCK(&(tsfdb->db_mutex), nullptr);
+		return tsfdb->api;
+	}
+}
+
+void FaceDBPool::release(int dbid)
+{
+	ThreadSafeFaceDB* tsfdb = nullptr;
+	{
+		//avoid dead lock
+		MutexLocker _ml(pool_mutex);
+		
+		facedb_map_t* _face_db_map = (facedb_map_t*)face_db_map;
+		if (_face_db_map->empty())
+			return;
+		
+		facedb_map_t::iterator iter = _face_db_map->find(dbid);
+		if (iter == _face_db_map->end())
+			return;
+		
+		tsfdb = iter->second;
+	}
+	
+	if (tsfdb != nullptr)
+	{
+		PLP_MUTEX_UNLOCK(&(tsfdb->db_mutex),);
+	}
+}

--
Gitblit v1.8.0