feat: add metrics for dokploy server

Dokploy · Jan 5, 2025 · 04f217c · 04f217c
1 parent 20aba3d
commit 04f217c
Show file tree

Hide file tree

Showing 11 changed files with 4,761 additions and 18 deletions.
diff --git a/apps/dokploy/components/dashboard/settings/servers/setup-monitoring.tsx b/apps/dokploy/components/dashboard/settings/servers/setup-monitoring.tsx
@@ -15,8 +15,6 @@ import { useForm } from "react-hook-form";
 import { toast } from "sonner";
 import { zodResolver } from "@hookform/resolvers/zod";
 import { Input, NumberInput } from "@/components/ui/input";
-import { CardTitle } from "@/components/ui/card";
-import { AlertTriangle, BarChart, BarcodeIcon } from "lucide-react";
 import { RefreshCw } from "lucide-react";
 import { useEffect, useState } from "react";
 import {
@@ -39,7 +37,7 @@ import { useUrl } from "@/utils/hooks/use-url";
 import { AlertBlock } from "@/components/shared/alert-block";
 
 interface Props {
-	serverId: string;
+	serverId?: string;
 }
 
 const Schema = z.object({
@@ -76,9 +74,16 @@ const Schema = z.object({
 type Schema = z.infer<typeof Schema>;
 
 export const SetupMonitoring = ({ serverId }: Props) => {
-	const { data, isLoading } = api.server.one.useQuery({
-		serverId,
-	});
+	const { data, isLoading } = serverId
+		? api.server.one.useQuery(
+				{
+					serverId: serverId || "",
+				},
+				{
+					enabled: !!serverId,
+				},
+			)
+		: api.admin.one.useQuery();
 
 	const url = useUrl();
 
@@ -89,9 +94,11 @@ export const SetupMonitoring = ({ serverId }: Props) => {
 
 		const allServices = projects.flatMap((project) => {
 			const services = extractServices(project);
-			return services
-				.filter((service) => service.serverId === serverId)
-				.map((service) => service.appName);
+			return serverId
+				? services
+						.filter((service) => service.serverId === serverId)
+						.map((service) => service.appName)
+				: services.map((service) => service.appName);
 		});
 
 		return [...new Set(allServices)];
@@ -156,7 +163,7 @@ export const SetupMonitoring = ({ serverId }: Props) => {
 	const { mutateAsync } = api.server.setupMonitoring.useMutation();
 
 	const generateToken = () => {
-		const array = new Uint8Array(32);
+		const array = new Uint8Array(64);
 		crypto.getRandomValues(array);
 		return Array.from(array, (byte) => byte.toString(16).padStart(2, "0")).join(
 			"",
@@ -165,7 +172,7 @@ export const SetupMonitoring = ({ serverId }: Props) => {
 
 	const onSubmit = async (values: Schema) => {
 		await mutateAsync({
-			serverId,
+			serverId: serverId || "",
 			...values,
 			containerRefreshRateMetrics: values.containerRefreshRateMetrics,
 			serverRefreshRateMetrics: values.serverRefreshRateMetrics,
@@ -518,9 +525,11 @@ export const SetupMonitoring = ({ serverId }: Props) => {
 						)}
 					/>
 				</div>
-				<Button type="submit" isLoading={form.formState.isSubmitting}>
-					Save changes
-				</Button>
+				<div className="flex items-center justify-end gap-2">
+					<Button type="submit" isLoading={form.formState.isSubmitting}>
+						Save changes
+					</Button>
+				</div>
 			</form>
 		</Form>
 	);

diff --git a/apps/dokploy/drizzle/0060_goofy_triton.sql b/apps/dokploy/drizzle/0060_goofy_triton.sql
@@ -0,0 +1,8 @@
+ALTER TABLE "admin" ADD COLUMN "serverRefreshRateMetrics" integer DEFAULT 5 NOT NULL;--> statement-breakpoint
+ALTER TABLE "admin" ADD COLUMN "containerRefreshRateMetrics" integer DEFAULT 5 NOT NULL;--> statement-breakpoint
+ALTER TABLE "admin" ADD COLUMN "containersMetricsDefinition" json;--> statement-breakpoint
+ALTER TABLE "admin" ADD COLUMN "defaultPortMetrics" integer DEFAULT 4500 NOT NULL;--> statement-breakpoint
+ALTER TABLE "admin" ADD COLUMN "metricsToken" text DEFAULT '' NOT NULL;--> statement-breakpoint
+ALTER TABLE "admin" ADD COLUMN "metricsUrlCallback" text DEFAULT '' NOT NULL;--> statement-breakpoint
+ALTER TABLE "admin" ADD COLUMN "thresholdCpu" integer DEFAULT 0 NOT NULL;--> statement-breakpoint
+ALTER TABLE "admin" ADD COLUMN "thresholdMemory" integer DEFAULT 0 NOT NULL;