zmDevelop/MultiThread.h at master · bale-john/zmDevelop · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#ifndef MULTITHREAD_H
#define MULTITHREAD_H
#include <pthread.h>
#include <iostream>
#include <cstdio>
#include <thread>
#include <vector>
#include <map>
#include <string>
#include <fstream>
#include <list>
#include <pthread.h>
#include "Util.h"
#include "Config.h"
#include "ConstDef.h"
#include "Log.h"
#include "Process.h"
#include "Zk.h"
#include "LoadBalance.h"
#include "ServiceListener.h"
using namespace std;
class MultiThread {
private:
	MultiThread();
	static bool threadError;
	static MultiThread* mlInstance;
	Config* conf;
	ServiceListener* sl;
    LoadBalance* lb;
	unordered_map<string, int> updateServiceInfo;
	list<string> priority;
	//key is the thread id of check thread. value is the index this thread in thread pool
	map<pthread_t, size_t> threadPos;
    //spinlock_t threadPosLock;
	pthread_mutex_t threadPosLock;
	Zk* zk;
	int serviceFatherNum;
	//copy of myServiceFather in loadBalance
	vector<string> serviceFathers;
    //spinlock_t serviceFathersLock;
	pthread_mutex_t serviceFathersLock;
	//marked weather there is a thread checking this service father
	vector<bool> hasThread;
	//spinlock_t hasThreadLock;
	pthread_mutex_t hasThreadLock;
	//the next service father waiting for check
	int waitingIndex;
	//spinlock_t waitingIndexLock;
	pthread_mutex_t waitingIndexLock;

public:
	~MultiThread();
    static MultiThread* getInstance();

	int runMainThread();
	static void* staticUpdateService(void* args);
	static void* staticCheckService(void* args);
	void checkService();
	void updateService();
	int tryConnect(string curServiceFather);
	int isServiceExist(struct in_addr *addr, char* host, int port, int timeout, int curStatus);

	int updateConf(string node, int val);
	int updateZk(string node, int val);
	bool isOnlyOneUp(string node, int val);
	bool isOnlyOneUp(string node);

	static bool isThreadError();
	static void setThreadError();
	static void clearThreadError();

	void setWaitingIndex(int val);
	int getAndAddWaitingIndex();

	void insertUpdateServiceInfo(const string& service, const int val);

    void clearHasThread(int sz);
    void setHasThread(int index, bool val);
    bool getHasThread(int index);
};
#endif