Bladeren bron

Unify enqueueing code.

default 3 jaren geleden
bovenliggende
commit
e8fbc94089
1 gewijzigde bestanden met toevoegingen van 37 en 37 verwijderingen
  1. 37 37
      data.c

+ 37 - 37
data.c

@@ -892,34 +892,45 @@ d_char *history_list(snac *snac)
 }
 }
 
 
 
 
-void enqueue_input(snac *snac, char *msg, char *req, int retries)
-/* enqueues an input message */
+static int _enqueue_put(char *fn, char *msg)
+/* writes safely to the queue */
 {
 {
-    int qrt  = xs_number_get(xs_dict_get(srv_config, "queue_retry_minutes"));
-    xs *ntid = tid(retries * 60 * qrt);
-    xs *fn   = xs_fmt("%s/queue/%s.json", snac->basedir, ntid);
-    xs *tfn  = xs_fmt("%s.tmp", fn);
+    int ret = 1;
+    xs *tfn = xs_fmt("%s.tmp", fn);
     FILE *f;
     FILE *f;
 
 
     if ((f = fopen(tfn, "w")) != NULL) {
     if ((f = fopen(tfn, "w")) != NULL) {
-        xs *qmsg = xs_dict_new();
-        xs *rn   = xs_number_new(retries);
-        xs *j;
-
-        qmsg = xs_dict_append(qmsg, "type",    "input");
-        qmsg = xs_dict_append(qmsg, "object",  msg);
-        qmsg = xs_dict_append(qmsg, "req",     req);
-        qmsg = xs_dict_append(qmsg, "retries", rn);
-
-        j = xs_json_dumps_pp(qmsg, 4);
+        xs *j = xs_json_dumps_pp(msg, 4);
 
 
         fwrite(j, strlen(j), 1, f);
         fwrite(j, strlen(j), 1, f);
         fclose(f);
         fclose(f);
 
 
         rename(tfn, fn);
         rename(tfn, fn);
-
-        snac_debug(snac, 1, xs_fmt("enqueue_input %s", fn));
     }
     }
+    else
+        ret = 0;
+
+    return ret;
+}
+
+
+void enqueue_input(snac *snac, char *msg, char *req, int retries)
+/* enqueues an input message */
+{
+    int qrt  = xs_number_get(xs_dict_get(srv_config, "queue_retry_minutes"));
+    xs *ntid = tid(retries * 60 * qrt);
+    xs *fn   = xs_fmt("%s/queue/%s.json", snac->basedir, ntid);
+    xs *qmsg = xs_dict_new();
+    xs *rn   = xs_number_new(retries);
+
+    qmsg = xs_dict_append(qmsg, "type",    "input");
+    qmsg = xs_dict_append(qmsg, "object",  msg);
+    qmsg = xs_dict_append(qmsg, "req",     req);
+    qmsg = xs_dict_append(qmsg, "retries", rn);
+
+    _enqueue_put(fn, qmsg);
+
+    snac_debug(snac, 1, xs_fmt("enqueue_input %s", fn));
 }
 }
 
 
 
 
@@ -934,28 +945,17 @@ void enqueue_output(snac *snac, char *msg, char *actor, int retries)
     int qrt  = xs_number_get(xs_dict_get(srv_config, "queue_retry_minutes"));
     int qrt  = xs_number_get(xs_dict_get(srv_config, "queue_retry_minutes"));
     xs *ntid = tid(retries * 60 * qrt);
     xs *ntid = tid(retries * 60 * qrt);
     xs *fn   = xs_fmt("%s/queue/%s.json", snac->basedir, ntid);
     xs *fn   = xs_fmt("%s/queue/%s.json", snac->basedir, ntid);
-    xs *tfn  = xs_fmt("%s.tmp", fn);
-    FILE *f;
+    xs *qmsg = xs_dict_new();
+    xs *rn   = xs_number_new(retries);
 
 
-    if ((f = fopen(tfn, "w")) != NULL) {
-        xs *qmsg = xs_dict_new();
-        xs *rn   = xs_number_new(retries);
-        xs *j;
-
-        qmsg = xs_dict_append(qmsg, "type",    "output");
-        qmsg = xs_dict_append(qmsg, "actor",   actor);
-        qmsg = xs_dict_append(qmsg, "object",  msg);
-        qmsg = xs_dict_append(qmsg, "retries", rn);
-
-        j = xs_json_dumps_pp(qmsg, 4);
-
-        fwrite(j, strlen(j), 1, f);
-        fclose(f);
+    qmsg = xs_dict_append(qmsg, "type",    "output");
+    qmsg = xs_dict_append(qmsg, "actor",   actor);
+    qmsg = xs_dict_append(qmsg, "object",  msg);
+    qmsg = xs_dict_append(qmsg, "retries", rn);
 
 
-        rename(tfn, fn);
+    _enqueue_put(fn, qmsg);
 
 
-        snac_debug(snac, 1, xs_fmt("enqueue_output %s %s %d", actor, fn, retries));
-    }
+    snac_debug(snac, 1, xs_fmt("enqueue_output %s %s %d", actor, fn, retries));
 }
 }