[Feature] Allow forced and non-forced upstreams selection

1 files changed, 10 insertions, 0 deletions

diff --git a/src/libutil/upstream.h b/src/libutil/upstream.h
index c3cbf685a..07b9302e8 100644
--- a/src/libutil/upstream.h
+++ b/src/libutil/upstream.h
@@ -191,6 +191,16 @@ struct upstream* rspamd_upstream_get (struct upstream_list *ups,
 		const guchar *key, gsize keylen);
 
 /**
+ * Get new upstream from the list
+ * @param ups upstream list
+ * @param type type of rotation algorithm, for `RSPAMD_UPSTREAM_HASHED` it is required to specify `key` and `keylen` as arguments
+ * @return
+ */
+struct upstream* rspamd_upstream_get_forced (struct upstream_list *ups,
+		enum rspamd_upstream_rotation forced_type,
+		const guchar *key, gsize keylen);
+
+/**
  * Re-resolve addresses for all upstreams registered
  */
 void rspamd_upstream_reresolve (struct upstream_ctx *ctx);