GithubStar/workflow
1
0
Fork 0
mirror of https://github.com/sogou/workflow.git synced 2026-10-24 00:16:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

add some includes

Browse Source
This commit is contained in:
XieHan
2021-02-02 23:49:58 +08:00
parent 4591cf42f3
commit c357fe44b2
4 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CMakeLists_Headers.txt
+1 -1
View File
@@ -66,6 +66,7 @@ set(INCLUDE_HEADERS
src/manager/WFFuture.h
src/manager/WFFacilities.h
src/manager/WFFacilities.inl
src/manager/UpstreamPolicies.h
src/util/EncodeStream.h
src/util/LRUCache.h
src/util/StringUtil.h
@@ -84,7 +85,6 @@ set(INCLUDE_HEADERS
src/factory/WFOperator.h
src/factory/WFNameService.h
src/factory/WFDNSResolver.h
src/manager/UpstreamPolicies.h
)
if(KAFKA STREQUAL "y")
src/manager/UpstreamManager.cc
+6 -7
View File
@@ -16,6 +16,7 @@
Authors: Wu Jiaxu (wujiaxu@sogou-inc.com)
*/
#include <pthread.h>
#include "UpstreamManager.h"
#include "WFNameService.h"
#include "UpstreamPolicies.h"
@@ -31,15 +32,15 @@ public:
void add_policy_name(const std::string& name)
{
pthread_mutex_lock(&this->lock);
pthread_mutex_lock(&this->mutex);
this->upstream_names.push_back(name);
pthread_mutex_unlock(&this->lock);
pthread_mutex_unlock(&this->mutex);
}
private:
__UpstreamManager()
__UpstreamManager() :
mutex(PTHREAD_MUTEX_INITIALIZER)
{
pthread_mutex_init(&this->lock, NULL);
}
~__UpstreamManager()
@@ -52,11 +53,9 @@ private:
policy = ns->del_policy(name.c_str());
delete policy;
}
pthread_mutex_destroy(&this->lock);
}
pthread_mutex_t lock;
pthread_mutex_t mutex;
std::vector<std::string> upstream_names;
};
src/manager/UpstreamPolicies.cc
+2
View File
@@ -16,7 +16,9 @@
Authors: Wu Jiaxu (wujiaxu@sogou-inc.com)
*/
#include <pthread.h>
#include <algorithm>
#include "URIParser.h"
#include "StringUtil.h"
#include "UpstreamPolicies.h"
#include "WFDNSResolver.h"
src/manager/UpstreamPolicies.h
+6 -4
View File
@@ -19,6 +19,11 @@
#ifndef _UPSTREAM_POLICIES_H_
#define _UPSTREAM_POLICIES_H_
#include <pthread.h>
#include <unordered_map>
#include <vector>
#include <atomic>
#include "URIParser.h"
#include "EndpointParams.h"
#include "WFNameService.h"
#include "WFDNSResolver.h"
@@ -26,9 +31,6 @@
#include "WFTaskError.h"
#include "UpstreamManager.h"
#include <unordered_map>
#include <vector>
#define MTTR_SECOND 30
#define VIRTUAL_GROUP_SIZE 16
@@ -49,7 +51,7 @@ public:
short port_value;
struct list_head list;
std::atomic<unsigned int> fail_count;
int64_t broken_timeout;
long long broken_timeout;
unsigned int consistent_hash[VIRTUAL_GROUP_SIZE];
public: