jobid generate

src/lib/idgen.c

@@ -0,0 +1,70 @@
+#include <stdint.h>
+#include <time.h>
+
+#include "idgen.h"
+#include "mylock.h"
+#include "utils.h"
+#include "xce/errnox.h"
+
+int idgen_ctx_init(idgen_ctx_t* ctx)
+{
+    if (ctx == NULL) {
+        return -EINVAL;
+    }
+
+    mylock_init(&(ctx->lock));
+    ctx->timestamp = time(NULL);
+    ctx->sequence  = 0;
+
+    return 0;
+}
+
+int idgen(idgen_ctx_t* ctx, size_t clusterId, size_t apiserverId,
+          size_t arraySize, uint64_t* pidx)
+{
+    time_t   now;
+    uint64_t idx;
+    uint64_t ts, tag, seq;
+
+    if ((ctx == NULL) || (clusterId > MAX_CLUSTER_ID)
+        || (apiserverId > MAX_APISERVER_ID) || (arraySize < 1)
+        || (arraySize > MAX_ARRAY_IDX + 1) || (pidx == NULL)) {
+        return -EINVAL;
+    }
+
+    mylock_lock(&(ctx->lock));
+    now = time(NULL);
+
+    if (now != ctx->timestamp) {
+        ctx->timestamp = now;
+        ctx->sequence  = 0;
+    }
+
+    ts = ctx->timestamp;
+
+    if (arraySize == 1) {
+        /* non array job */
+        tag = 0x0;
+        seq = ctx->sequence;
+        ctx->sequence++;
+    } else {
+        tag           = TAG_JOB_ARRAY;
+        seq           = ROUNDUP(ctx->sequence, N_ARRAY_IDX_BITS);
+        ctx->sequence = ROUNDUP(seq + 1, N_ARRAY_IDX_BITS);
+    }
+    mylock_unlock(&(ctx->lock));
+
+    idx = MAKE_JOBID(clusterId, apiserverId, ts, tag, seq);
+
+    *pidx = idx;
+    return arraySize;
+}
+
+void idgen_ctx_fini(idgen_ctx_t* ctx)
+{
+    if (ctx == NULL) {
+        return;
+    }
+
+    mylock_fini(&(ctx->lock));
+}