-
Notifications
You must be signed in to change notification settings - Fork 994
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move ProxySQL_Poll() to its own source file
- Loading branch information
1 parent
5181812
commit bd7846f
Showing
6 changed files
with
159 additions
and
146 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,50 @@ | ||
| #ifndef __CLASS_PROXYSQL_POLL | ||
| #define __CLASS_PROXYSQL_POLL | ||
|
|
||
| //#include "MySQL_Data_Stream.h" | ||
|
|
||
| class iface_info { | ||
| public: | ||
| char *iface; | ||
| char *address; | ||
| int port; | ||
| int fd; | ||
| iface_info(char *_i, char *_a, int p, int f) { | ||
| iface=strdup(_i); | ||
| address=strdup(_a); | ||
| port=p; | ||
| fd=f; | ||
| } | ||
| ~iface_info() { | ||
| free(iface); | ||
| free(address); | ||
| close(fd); | ||
| } | ||
| }; | ||
|
|
||
| class ProxySQL_Poll { | ||
| private: | ||
| void shrink(); | ||
| void expand(unsigned int more); | ||
|
|
||
| public: | ||
| unsigned int len; | ||
| unsigned int size; | ||
| struct pollfd *fds; | ||
| MySQL_Data_Stream **myds; | ||
| unsigned long long *last_recv; | ||
| unsigned long long *last_sent; | ||
| std::atomic<bool> bootstrapping_listeners; | ||
| volatile int pending_listener_add; | ||
| volatile int pending_listener_del; | ||
| unsigned int poll_timeout; | ||
| unsigned long loops; | ||
| StatCounters *loop_counters; | ||
|
|
||
| ProxySQL_Poll(); | ||
| ~ProxySQL_Poll(); | ||
| void add(uint32_t _events, int _fd, MySQL_Data_Stream *_myds, unsigned long long sent_time); | ||
| void remove_index_fast(unsigned int i); | ||
| int find_index(int fd); | ||
| }; | ||
| #endif // __CLASS_PROXYSQL_POLL |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,107 @@ | ||
| #include <stdlib.h> | ||
| #include <unistd.h> | ||
| #include <string.h> | ||
| #include "StatCounters.h" | ||
| #include "MySQL_Data_Stream.h" | ||
| #include "ProxySQL_Poll.h" | ||
| #include "proxysql_structs.h" | ||
| #include <poll.h> | ||
| #include "cpp.h" | ||
|
|
||
| void ProxySQL_Poll::shrink() { | ||
| unsigned int new_size=l_near_pow_2(len+1); | ||
| fds=(struct pollfd *)realloc(fds,new_size*sizeof(struct pollfd)); | ||
| myds=(MySQL_Data_Stream **)realloc(myds,new_size*sizeof(MySQL_Data_Stream *)); | ||
| last_recv=(unsigned long long *)realloc(last_recv,new_size*sizeof(unsigned long long)); | ||
| last_sent=(unsigned long long *)realloc(last_sent,new_size*sizeof(unsigned long long)); | ||
| size=new_size; | ||
| } | ||
|
|
||
| void ProxySQL_Poll::expand(unsigned int more) { | ||
| if ( (len+more) > size ) { | ||
| unsigned int new_size=l_near_pow_2(len+more); | ||
| fds=(struct pollfd *)realloc(fds,new_size*sizeof(struct pollfd)); | ||
| myds=(MySQL_Data_Stream **)realloc(myds,new_size*sizeof(MySQL_Data_Stream *)); | ||
| last_recv=(unsigned long long *)realloc(last_recv,new_size*sizeof(unsigned long long)); | ||
| last_sent=(unsigned long long *)realloc(last_sent,new_size*sizeof(unsigned long long)); | ||
| size=new_size; | ||
| } | ||
| } | ||
|
|
||
| ProxySQL_Poll::ProxySQL_Poll() { | ||
| loop_counters=new StatCounters(15,10); | ||
| poll_timeout=0; | ||
| loops=0; | ||
| len=0; | ||
| pending_listener_add=0; | ||
| pending_listener_del=0; | ||
| bootstrapping_listeners = true; | ||
| size=MIN_POLL_LEN; | ||
| fds=(struct pollfd *)malloc(size*sizeof(struct pollfd)); | ||
| myds=(MySQL_Data_Stream **)malloc(size*sizeof(MySQL_Data_Stream *)); | ||
| last_recv=(unsigned long long *)malloc(size*sizeof(unsigned long long)); | ||
| last_sent=(unsigned long long *)malloc(size*sizeof(unsigned long long)); | ||
| } | ||
|
|
||
|
|
||
| ProxySQL_Poll::~ProxySQL_Poll() { | ||
| unsigned int i; | ||
| for (i=0;i<len;i++) { | ||
| if ( | ||
| myds[i] && // fix bug #278 . This should be caused by not initialized datastreams used to ping the backend | ||
| myds[i]->myds_type==MYDS_LISTENER) { | ||
| delete myds[i]; | ||
| } | ||
| } | ||
| free(myds); | ||
| free(fds); | ||
| free(last_recv); | ||
| free(last_sent); | ||
| delete loop_counters; | ||
| } | ||
|
|
||
|
|
||
| void ProxySQL_Poll::add(uint32_t _events, int _fd, MySQL_Data_Stream *_myds, unsigned long long sent_time) { | ||
| if (len==size) { | ||
| expand(1); | ||
| } | ||
| myds[len]=_myds; | ||
| fds[len].fd=_fd; | ||
| fds[len].events=_events; | ||
| fds[len].revents=0; | ||
| if (_myds) { | ||
| _myds->mypolls=this; | ||
| _myds->poll_fds_idx=len; // fix a serious bug | ||
| } | ||
| last_recv[len]=monotonic_time(); | ||
| last_sent[len]=sent_time; | ||
| len++; | ||
| } | ||
|
|
||
| void ProxySQL_Poll::remove_index_fast(unsigned int i) { | ||
| if ((int)i==-1) return; | ||
| myds[i]->poll_fds_idx=-1; // this prevents further delete | ||
| if (i != (len-1)) { | ||
| myds[i]=myds[len-1]; | ||
| fds[i].fd=fds[len-1].fd; | ||
| fds[i].events=fds[len-1].events; | ||
| fds[i].revents=fds[len-1].revents; | ||
| myds[i]->poll_fds_idx=i; // fix a serious bug | ||
| last_recv[i]=last_recv[len-1]; | ||
| last_sent[i]=last_sent[len-1]; | ||
| } | ||
| len--; | ||
| if ( ( len>MIN_POLL_LEN ) && ( size > len*MIN_POLL_DELETE_RATIO ) ) { | ||
| shrink(); | ||
| } | ||
| } | ||
|
|
||
| int ProxySQL_Poll::find_index(int fd) { | ||
| unsigned int i; | ||
| for (i=0; i<len; i++) { | ||
| if (fds[i].fd==fd) { | ||
| return i; | ||
| } | ||
| } | ||
| return -1; | ||
| } |