refactor: move ExtractNodeIndex to BotConfig.Cluster class

This commit is contained in:
spiral 2022-01-07 14:25:08 -05:00
parent 0d42328694
commit 39c8590bce
No known key found for this signature in database
GPG Key ID: A6059F0CA0E1BD31
2 changed files with 4 additions and 5 deletions
PluralKit.Bot

View File

@ -32,5 +32,8 @@ public class BotConfig
public string NodeName { get; set; }
public int TotalShards { get; set; }
public int TotalNodes { get; set; }
// Node name eg. "pluralkit-3", want to extract the 3. blame k8s :p
public int NodeIndex => int.Parse(NodeName.Split("-").Last());
}
}

View File

@ -147,7 +147,7 @@ public class Init
// For multi-instance deployments, calculate the "span" of shards this node is responsible for
var totalNodes = config.Cluster.TotalNodes;
var totalShards = config.Cluster.TotalShards;
var nodeIndex = ExtractNodeIndex(config.Cluster.NodeName);
var nodeIndex = config.Cluster.NodeIndex;
// Should evenly distribute shards even with an uneven amount of nodes
var shardMin = (int)Math.Round(totalShards * (float)nodeIndex / totalNodes);
@ -160,8 +160,4 @@ public class Init
await cluster.Start(info);
}
}
private static int ExtractNodeIndex(string nodeName) =>
// Node name eg. "pluralkit-3", want to extract the 3. blame k8s :p
int.Parse(nodeName.Split("-").Last());
}