From a0f21a6d02dd0d1cf71ab7bcc290ac4a64b03d24 Mon Sep 17 00:00:00 2001 From: Brooke Morrison Date: Fri, 6 Jan 2023 18:27:59 -0800 Subject: [PATCH] Add ability to delete server by server name --- src/models/robotServer.js | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/models/robotServer.js b/src/models/robotServer.js index 9f7b24a..9ece981 100644 --- a/src/models/robotServer.js +++ b/src/models/robotServer.js @@ -296,7 +296,12 @@ module.exports.deleteRobotServer = async (server_id) => { console.log(server_id); const db = require("../services/db"); try { - const query = "DELETE FROM robot_servers WHERE server_id =$1"; + let query; + if (query.includes("serv-")) { // assume server ID + query = "DELETE FROM robot_servers WHERE server_id =$1"; + } else { // assume server name + query = "DELETE FROM robot_servers WHERE server_name =$1"; + } const result = await db.query(query, [server_id]); console.log("Deleted row count: ", result.rowCount);