diff --git a/SystemdCtl/Controllers/UnitController.cs b/SystemdCtl/Controllers/UnitController.cs
index da1c534..40ef46a 100644
--- a/SystemdCtl/Controllers/UnitController.cs
+++ b/SystemdCtl/Controllers/UnitController.cs
@@ -7,20 +7,20 @@ using LibSystemdCli.Models;
using Microsoft.AspNetCore.Mvc;
namespace SystemdCtl.Controllers;
+
[ApiController]
[Route("/api/")]
-public class UnitController : ControllerBase
-{
+public class UnitController : ControllerBase {
[HttpGet("listUnits")]
- public async IAsyncEnumerable<SystemdUnitListItem> GetUnits()
- {
- await foreach (var unit in SystemdExecutor.GetUnits())
- {
+ public async IAsyncEnumerable<SystemdUnitListItem> GetUnits() {
+ await foreach (var unit in SystemdExecutor.GetUnits()) {
yield return unit;
- await Response.Body.FlushAsync();
+
+ if (Response.HasStarted)
+ await Response.Body.FlushAsync();
}
}
-
+
// [HttpGet("unit/{serviceName}/logs")]
// public async IAsyncEnumerable<SystemdJournalLogItem> GetUnitLogs(string serviceName, [FromQuery] int contextLines = 100)
// {
@@ -32,32 +32,29 @@ public class UnitController : ControllerBase
// }
// }
[HttpGet("unit/{serviceName}/logs")]
- public async Task GetUnitLogs(string serviceName, [FromQuery] int contextLines = 100)
- {
+ public async Task GetUnitLogs(string serviceName, [FromQuery] int contextLines = 100) {
Response.ContentType = "application/json";
await Response.StartAsync();
await Response.Body.WriteAsync("[\n"u8.ToArray());
- await foreach (var log in SystemdExecutor.GetUnitLogs(serviceName, contextLines: contextLines))
- {
+ await foreach (var log in SystemdExecutor.GetUnitLogs(serviceName, contextLines: contextLines)) {
Console.WriteLine(log.Message);
var bytes = Encoding.UTF8.GetBytes($" {log.ToJson(indent: false)},\n");
await Response.Body.WriteAsync(bytes);
await Response.Body.FlushAsync();
}
+
await Response.Body.WriteAsync("]\n"u8.ToArray());
await Response.Body.FlushAsync();
await Response.CompleteAsync();
}
-
+
[HttpGet("unit/{serviceName}/dataStream")]
- public async Task GetUnitDataStream(string serviceName, [FromQuery] int contextLines = 100)
- {
+ public async Task GetUnitDataStream(string serviceName, [FromQuery] int contextLines = 100) {
Response.ContentType = "application/json";
await Response.StartAsync();
await Response.Body.WriteAsync("[\n"u8.ToArray());
var oldData = await SystemdExecutor.GetUnitData(serviceName);
- while(true)
- {
+ while (true) {
var data = await SystemdExecutor.GetUnitData(serviceName);
data.SetOldData(oldData);
var bytes = Encoding.UTF8.GetBytes($" {data.ToJson(indent: false)},\n");
@@ -66,29 +63,30 @@ public class UnitController : ControllerBase
oldData = data;
await Task.Delay(1000);
}
+
await Response.Body.WriteAsync("]\n"u8.ToArray());
await Response.Body.FlushAsync();
await Response.CompleteAsync();
}
-
+
[HttpGet("unit/{serviceName}/data")]
public Task<SystemdServiceData> GetUnitData(string serviceName) => SystemdExecutor.GetUnitData(serviceName);
-
+
[HttpGet("unit/{serviceName}/start")]
public Task StartUnit(string serviceName) => CommandExecutor.ExecuteCommand("systemctl", $"start {serviceName}");
-
+
[HttpGet("unit/{serviceName}/stop")]
public Task StopUnit(string serviceName) => CommandExecutor.ExecuteCommand("systemctl", $"stop {serviceName}");
-
+
[HttpGet("unit/{serviceName}/restart")]
public Task RestartUnit(string serviceName) => CommandExecutor.ExecuteCommand("systemctl", $"restart {serviceName}");
-
+
[HttpGet("unit/{serviceName}/reload")]
public Task ReloadUnit(string serviceName) => CommandExecutor.ExecuteCommand("systemctl", $"reload {serviceName}");
-
+
[HttpGet("unit/{serviceName}/enable")]
public Task EnableUnit(string serviceName) => CommandExecutor.ExecuteCommand("systemctl", $"enable {serviceName}");
-
+
[HttpGet("unit/{serviceName}/disable")]
public Task DisableUnit(string serviceName) => CommandExecutor.ExecuteCommand("systemctl", $"disable {serviceName}");
}
\ No newline at end of file
|
