privatevoid.net/depot
1
1
Fork 0
Code Issues 23 Pull requests 2 Projects 3 Releases Packages Wiki Activity Actions

modules/consul-service-registry: fix bugs, add external registration mode

Browse source
This commit is contained in:
Max Headroom 2023-03-05 23:50:26 +01:00
parent 674fde3617
commit 7aab09157e
1 changed files with 38 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
modules/consul-service-registry/default.nix
View file

@ -1,4 +1,4 @@
{ config, lib, ... }:
{ config, lib, pkgs, ... }:
with lib;
@ -7,28 +7,51 @@ let
consul = "${config.services.consul.package}/bin/consul";
consulServiceDefinition = submodule ({ name, ... }: {
unit = mkOption {
description = "Which systemd service to attach to.";
default = name;
type = types.str;
};
definition = mkOption {
description = "Consul service definition.";
type = types.attrs;
consulServiceDefinition = types.submodule ({ name, ... }: {
options = {
unit = mkOption {
description = "Which systemd service to attach to.";
default = name;
type = types.str;
};
mode = mkOption {
description = "How to attach command executions to the service.";
type = types.enum [ "direct" "external" ];
default = "direct";
};
definition = mkOption {
description = "Consul service definition.";
type = types.attrs;
};
};
});
attachToService = name: conf: let
serviceJson = pkgs.writeText "consul-service-${name}.json" (builtins.toJSON conf.definition);
in {
name = conf.unit;
name = {
direct = conf.unit;
external = "register-consul-svc-${conf.unit}";
}.${conf.mode};
value = {
serviceConfig = {
ExecStartPost = "${consul} services register ${serviceJson}";
ExecStopPre = "${consul} services deregister ${serviceJson}";
direct = {
serviceConfig = {
ExecStartPost = "${consul} services register ${serviceJson}";
ExecStopPre = "${consul} services deregister ${serviceJson}";
};
};
};
external = {
after = [ "${conf.unit}.service" ];
wantedBy = [ "${conf.unit}.service" ];
unitConfig.BindsTo = "${conf.unit}.service";
serviceConfig = {
Type = "oneshot";
RemainAfterExit = true;
ExecStart = "${consul} services register ${serviceJson}";
ExecStop = "${consul} services deregister ${serviceJson}";
};
};
}.${conf.mode};
};
in