One Level Up Top Level

src/stream/ngx_stream_upstream_round_robin.h - nginx source code

Data types defined

Functions defined

Macros defined

Source code


  2. /*
  3. * Copyright (C) Igor Sysoev
  4. * Copyright (C) Nginx, Inc.
  5. */


  8. #ifndef _NGX_STREAM_UPSTREAM_ROUND_ROBIN_H_INCLUDED_
  9. ‌#define _NGX_STREAM_UPSTREAM_ROUND_ROBIN_H_INCLUDED_


  12. #include <ngx_config.h>
  13. #include <ngx_core.h>
  14. #include <ngx_stream.h>


  17. ‌typedef struct ngx_stream_upstream_rr_peers_s  ngx_stream_upstream_rr_peers_t;
  18. ‌typedef struct ngx_stream_upstream_rr_peer_s   ngx_stream_upstream_rr_peer_t;


  21. #if (NGX_STREAM_UPSTREAM_ZONE)

  23. ‌typedef struct {
  24.     ngx_event_t                      event;         /* must be first */
  25.     ngx_uint_t                       worker;
  26.     ngx_str_t                        name;
  27.     ngx_str_t                        service;
  28.     time_t                           valid;
  29.     ngx_stream_upstream_rr_peers_t  *peers;
  30.     ngx_stream_upstream_rr_peer_t   *peer;
  31. ‌} ngx_stream_upstream_host_t;

  33. #endif


  36. ‌struct ngx_stream_upstream_rr_peer_s {
  37.     struct sockaddr                 *sockaddr;
  38.     socklen_t                        socklen;
  39.     ngx_str_t                        name;
  40.     ngx_str_t                        server;

  42.     ngx_int_t                        current_weight;
  43.     ngx_int_t                        effective_weight;
  44.     ngx_int_t                        weight;

  46.     ngx_uint_t                       conns;
  47.     ngx_uint_t                       max_conns;

  49.     ngx_uint_t                       fails;
  50.     time_t                           accessed;
  51.     time_t                           checked;

  53.     ngx_uint_t                       max_fails;
  54.     time_t                           fail_timeout;
  55.     ngx_msec_t                       slow_start;
  56.     ngx_msec_t                       start_time;

  58.     ngx_uint_t                       down;

  60.     void                            *ssl_session;
  61.     int                              ssl_session_len;

  63. #if (NGX_STREAM_UPSTREAM_ZONE)
  64.     unsigned                         zombie:1;

  66.     ngx_atomic_t                     lock;
  67.     ngx_uint_t                       refs;
  68.     ngx_stream_upstream_host_t      *host;
  69. #endif

  71.     ngx_stream_upstream_rr_peer_t   *next;

  73.     NGX_COMPAT_BEGIN(14)
  74.     NGX_COMPAT_END
  75. };


  78. ‌struct ngx_stream_upstream_rr_peers_s {
  79.     ngx_uint_t                       number;

  81. #if (NGX_STREAM_UPSTREAM_ZONE)
  82.     ngx_slab_pool_t                 *shpool;
  83.     ngx_atomic_t                     rwlock;
  84.     ngx_uint_t                      *config;
  85.     ngx_stream_upstream_rr_peer_t   *resolve;
  86.     ngx_stream_upstream_rr_peers_t  *zone_next;
  87. #endif

  89.     ngx_uint_t                       total_weight;
  90.     ngx_uint_t                       tries;

  92.     unsigned                         single:1;
  93.     unsigned                         weighted:1;

  95.     ngx_str_t                       *name;

  97.     ngx_stream_upstream_rr_peers_t  *next;

  99.     ngx_stream_upstream_rr_peer_t   *peer;
  100. };


  103. #if (NGX_STREAM_UPSTREAM_ZONE)

  105. ‌#define ngx_stream_upstream_rr_peers_rlock(peers)                             \
  106.                                                                               \
  107.     if (peers->shpool) {                                                      \
  108.         ngx_rwlock_rlock(&peers->rwlock);                                     \
  109.     }

  111. ‌#define ngx_stream_upstream_rr_peers_wlock(peers)                             \
  112.                                                                               \
  113.     if (peers->shpool) {                                                      \
  114.         ngx_rwlock_wlock(&peers->rwlock);                                     \
  115.     }

  117. ‌#define ngx_stream_upstream_rr_peers_unlock(peers)                            \
  118.                                                                               \
  119.     if (peers->shpool) {                                                      \
  120.         ngx_rwlock_unlock(&peers->rwlock);                                    \
  121.     }


  124. ‌#define ngx_stream_upstream_rr_peer_lock(peers, peer)                         \
  125.                                                                               \
  126.     if (peers->shpool) {                                                      \
  127.         ngx_rwlock_wlock(&peer->lock);                                        \
  128.     }

  130. ‌#define ngx_stream_upstream_rr_peer_unlock(peers, peer)                       \
  131.                                                                               \
  132.     if (peers->shpool) {                                                      \
  133.         ngx_rwlock_unlock(&peer->lock);                                       \
  134.     }


  137. ‌#define ngx_stream_upstream_rr_peer_ref(peers, peer)                          \
  138.     (peer)->refs++;


  141. static ngx_inline void
  142. ‌ngx_stream_upstream_rr_peer_free_locked(ngx_stream_upstream_rr_peers_t *peers,
  143.     ngx_stream_upstream_rr_peer_t *peer)
  144. {
  145.     if (peer->refs) {
  146.         peer->zombie = 1;
  147.         return;
  148.     }

  150.     ngx_slab_free_locked(peers->shpool, peer->sockaddr);
  151.     ngx_slab_free_locked(peers->shpool, peer->name.data);

  153.     if (peer->server.data) {
  154.         ngx_slab_free_locked(peers->shpool, peer->server.data);
  155.     }

  157. #if (NGX_STREAM_SSL)
  158.     if (peer->ssl_session) {
  159.         ngx_slab_free_locked(peers->shpool, peer->ssl_session);
  160.     }
  161. #endif

  163.     ngx_slab_free_locked(peers->shpool, peer);
  164. }


  167. static ngx_inline void
  168. ‌ngx_stream_upstream_rr_peer_free(ngx_stream_upstream_rr_peers_t *peers,
  169.     ngx_stream_upstream_rr_peer_t *peer)
  170. {
  171.     ngx_shmtx_lock(&peers->shpool->mutex);
  172.     ngx_stream_upstream_rr_peer_free_locked(peers, peer);
  173.     ngx_shmtx_unlock(&peers->shpool->mutex);
  174. }


  177. static ngx_inline ngx_int_t
  178. ‌ngx_stream_upstream_rr_peer_unref(ngx_stream_upstream_rr_peers_t *peers,
  179.     ngx_stream_upstream_rr_peer_t *peer)
  180. {
  181.     peer->refs--;

  183.     if (peers->shpool == NULL) {
  184.         return NGX_OK;
  185.     }

  187.     if (peer->refs == 0 && peer->zombie) {
  188.         ngx_stream_upstream_rr_peer_free(peers, peer);
  189.         return NGX_DONE;
  190.     }

  192.     return NGX_OK;
  193. }

  195. #else

  197. ‌#define ngx_stream_upstream_rr_peers_rlock(peers)
  198. ‌#define ngx_stream_upstream_rr_peers_wlock(peers)
  199. ‌#define ngx_stream_upstream_rr_peers_unlock(peers)
  200. ‌#define ngx_stream_upstream_rr_peer_lock(peers, peer)
  201. ‌#define ngx_stream_upstream_rr_peer_unlock(peers, peer)
  202. ‌#define ngx_stream_upstream_rr_peer_ref(peers, peer)
  203. ‌#define ngx_stream_upstream_rr_peer_unref(peers, peer)  NGX_OK

  205. #endif


  208. ‌typedef struct {
  209.     ngx_uint_t                       config;
  210.     ngx_stream_upstream_rr_peers_t  *peers;
  211.     ngx_stream_upstream_rr_peer_t   *current;
  212.     uintptr_t                       *tried;
  213.     uintptr_t                        data;
  214. ‌} ngx_stream_upstream_rr_peer_data_t;


  217. ngx_int_t ngx_stream_upstream_init_round_robin(ngx_conf_t *cf,
  218.     ngx_stream_upstream_srv_conf_t *us);
  219. ngx_int_t ngx_stream_upstream_init_round_robin_peer(ngx_stream_session_t *s,
  220.     ngx_stream_upstream_srv_conf_t *us);
  221. ngx_int_t ngx_stream_upstream_create_round_robin_peer(ngx_stream_session_t *s,
  222.     ngx_stream_upstream_resolved_t *ur);
  223. ngx_int_t ngx_stream_upstream_get_round_robin_peer(ngx_peer_connection_t *pc,
  224.     void *data);
  225. void ngx_stream_upstream_free_round_robin_peer(ngx_peer_connection_t *pc,
  226.     void *data, ngx_uint_t state);


  229. #endif /* _NGX_STREAM_UPSTREAM_ROUND_ROBIN_H_INCLUDED_ */

One Level Up Top Level