feat(packages): Add random-alerts package

Author: MartinNikov

packages/default.nix

@@ -57,6 +57,7 @@
         {
           lido-withdrawals-automation = pkgs.callPackage ./lido-withdrawals-automation { };
           pyroscope = pkgs.callPackage ./pyroscope { };
+          random-alerts = pkgs.callPackage ./random-alerts { };
         }
         // optionalAttrs (system == "x86_64-linux" || system == "aarch64-darwin") {
           grafana-agent = import ./grafana-agent { inherit inputs'; };

packages/random-alerts/.gitignore

@@ -0,0 +1 @@
+build

packages/random-alerts/default.nix

@@ -0,0 +1,14 @@
+{ buildDubPackage, ... }:
+buildDubPackage rec {
+  pname = "random-alerts";
+  version = "1.0.0";
+  src = ./.;
+  dubLock = {
+    dependencies = { };
+  };
+  installPhase = ''
+    mkdir -p $out/bin
+    install -m755 ./build/${pname} $out/bin/${pname}
+  '';
+  meta.mainProgram = pname;
+}

packages/random-alerts/dub.sdl

@@ -0,0 +1,3 @@
+name "random-alerts"
+
+targetPath "build"

packages/random-alerts/dub.selections.json

@@ -0,0 +1,5 @@
+{
+	"fileVersion": 1,
+	"versions": {
+	}
+}

packages/random-alerts/src/main.d

@@ -0,0 +1,164 @@
+import core.thread : Thread;
+import std.datetime : Duration, Clock, seconds, TimeOfDay;
+import std.format : format;
+import std.getopt : getopt, getOptConfig = config;
+import std.json : JSONValue, parseJSON, JSONOptions;
+import std.logger : infof, errorf, tracef, LogLevel;
+import std.random : uniform;
+import std.exception : enforce;
+
+import utils.json : toJSON;
+
+struct Params
+{
+    Duration minWaitTime;
+    Duration maxWaitTime;
+    Duration alertDuration;
+    string url;
+    TimeOfDay startTime;
+    TimeOfDay endTime;
+}
+
+struct Alert
+{
+    string[string] labels;
+    Annotation annotations;
+    string startsAt;
+    string endsAt;
+    string generatorURL;
+
+    struct Annotation
+    {
+        string alert_type;
+        string title;
+        string summary;
+    }
+}
+
+int main(string[] args)
+{
+    LogLevel logLevel = LogLevel.info;
+    string url;
+    string startTime = "00:00:00";
+    string endTime = "23:59:59";
+    uint minWaitTimeInSeconds = 3600;  // 1 hour
+    uint maxWaitTimeInSeconds = 14400; // 4 hours
+    uint alertDurationInSeconds = 3600;  // 1 hour
+
+    try
+    {
+        args.getopt(
+            getOptConfig.required, "url", &url,
+            "start-time", &startTime,
+            "end-time", &endTime,
+            "min-wait-time", &minWaitTimeInSeconds,
+            "max-wait-time", &maxWaitTimeInSeconds,
+            "alert-duration", &alertDurationInSeconds,
+            "log-level", &logLevel,
+        );
+
+        enforce(minWaitTimeInSeconds <= maxWaitTimeInSeconds, "Make sure that `max-wait-time` is greater than `min-wait-time`.");
+
+        setLogLevel(logLevel);
+
+        executeAtRandomIntervals(
+            Params(
+                url: url,
+                startTime: TimeOfDay.fromISOExtString(startTime),
+                endTime: TimeOfDay.fromISOExtString(endTime),
+                minWaitTime: minWaitTimeInSeconds.seconds(),
+                maxWaitTime: maxWaitTimeInSeconds.seconds(),
+                alertDuration: alertDurationInSeconds.seconds(),
+            )
+        );
+    }
+    catch (Exception e)
+    {
+        errorf("Exception: %s", e.message);
+        return 1;
+    }
+    return 0;
+}
+
+auto getRandomDuration(Duration min, Duration max) =>
+    uniform(min.total!"seconds", max.total!"seconds").seconds;
+
+void executeAtRandomIntervals(Params params)
+{
+    with(params) while (true)
+    {
+        auto currentTime = Clock.currTime();
+        auto currentTimeTOD = cast(TimeOfDay)currentTime;
+        Duration randomDuration = getRandomDuration(minWaitTime, maxWaitTime);
+        auto randomTime = currentTime + randomDuration;
+
+        tracef("The operating time is: [%s .. %s]", startTime, endTime);
+        tracef("The next alarm will be activated in %s", randomTime);
+
+        Thread.sleep(randomDuration); // sleep till the request is ready to be posted.
+
+        if (currentTimeTOD >= startTime && currentTimeTOD <= endTime)
+        {
+            infof("Posting alert... ");
+            postAlert(url, alertDuration);
+            infof("Alert posted successfully.");
+        }
+        else
+        {
+            infof("This service is outside working hours. The operating time is: [%s .. %s]", startTime, endTime);
+        }
+
+        Duration remainingTime = maxWaitTime - randomDuration;
+
+        tracef("The current interval's end will be at %s", (currentTime + remainingTime));
+        Thread.sleep(remainingTime); // sleep till the cycle is over.
+    }
+}
+
+void postAlert(string alertManagerEndpoint, Duration alertDuration)
+{
+    string url = alertManagerEndpoint ~ "/api/v2/alerts";
+
+    postJson(url, [
+        Alert(
+            startsAt: Clock.currTime.toUTC.toISOExtString(0),
+            endsAt: (Clock.currTime + alertDuration).toUTC.toISOExtString(0),
+            generatorURL: "http://localhost:9090",
+            labels: [
+                "alertname": "Random alert",
+                "severity": "critical",
+                "environment": "staging",
+                "job": "test-monitoring",
+            ],
+            annotations: Alert.Annotation(
+                alert_type: "critical",
+                title: "Write report",
+                summary: "The alert was triggered at '%s'".format(Clock.currTime.toUTC)
+            )
+        )
+    ]);
+}
+
+JSONValue postJson(T)(string url, in T value)
+{
+    import std.net.curl : HTTP, post, HTTPStatusException;
+
+    auto jsonRequest = value
+        .toJSON
+        .toPrettyString(JSONOptions.doNotEscapeSlashes);
+
+    tracef("Sending request to '%s':\n%s", url, jsonRequest);
+
+    auto http = HTTP();
+    http.addRequestHeader("Content-Type", "application/json");
+
+    auto response = post(url, jsonRequest, http);
+    return response.parseJSON;
+}
+
+void setLogLevel(LogLevel l)
+{
+    import std.logger : globalLogLevel, sharedLog;
+    globalLogLevel = l;
+    (cast()sharedLog()).logLevel = l;
+}

packages/random-alerts/src/utils/json.d

@@ -0,0 +1,66 @@
+module utils.json;
+
+import std.traits: isNumeric, isArray, isSomeChar, ForeachType, isBoolean;
+import std.json: JSONValue;
+import std.conv: to;
+import std.string: strip;
+import std.range: front;
+import std.algorithm: map;
+import std.array: array;
+import std.datetime: SysTime;
+import core.stdc.string: strlen;
+
+JSONValue toJSON(T)(in T value, bool simplify = false)
+{
+    static if (is(T == enum))
+    {
+        return JSONValue(value.enumToString);
+    }
+    else static if (is(T == bool) || is(T == string) || isSomeChar!T || isNumeric!T)
+        return JSONValue(value);
+    else static if ((isArray!T && isSomeChar!(ForeachType!T)) ) {
+        return JSONValue(value.idup[0..(strlen(value.ptr)-1)]);
+    }
+    else static if (isArray!T)
+    {
+        if (simplify && value.length == 1)
+            return value.front.toJSON(simplify);
+        else if (simplify  && isBoolean!(ForeachType!T) ) {
+            static if (isBoolean!(ForeachType!T)) {
+                return JSONValue((value.map!(a => a ? '1' : '0').array).to!string);
+            }
+            else {assert(0);}
+        }
+        else {
+            JSONValue[] result;
+            foreach (elem; value)
+                result ~= elem.toJSON(simplify);
+            return JSONValue(result);
+        }
+    }
+    else static if (is(T == SysTime)) {
+        return JSONValue(value.toISOExtString());
+    }
+    else static if (is(T == struct))
+    {
+        JSONValue[string] result;
+        auto name = "";
+        static foreach (idx, field; T.tupleof)
+        {
+            name = __traits(identifier, field).strip("_");
+            result[name] = value.tupleof[idx].toJSON(simplify);
+        }
+        return JSONValue(result);
+    }
+    else static if (is(T == K[V], K, V))
+    {
+        JSONValue[string] result;
+        foreach (key, field; value)
+        {
+            result[key] = field.toJSON(simplify);
+        }
+        return JSONValue(result);
+    }
+    else
+        static assert(false, "Unsupported type: `" ~ __traits(identifier, T) ~ "`");
+}