add broken pipe maangement so we cleanly allow | head -10 etc when pipe is broken and node is still trying to write.

Author: TimelordUK

cpp/binding.cpp

@@ -1,6 +1,7 @@
 #include <napi.h>
 #include "include/utils/Logger.h"
 #include "include/js/Connection.h"
+#include "include/common/platform.h"
 
 static Napi::Value SetLogLevel(const Napi::CallbackInfo &info)
 {
@@ -56,6 +57,9 @@ static Napi::Value SetLogFile(const Napi::CallbackInfo &info)
 // Initialize the module
 Napi::Object InitModule(Napi::Env env, Napi::Object exports)
 {
+  // Initialize platform-specific signal handlers
+  Platform::InitializeSignalHandlers();
+
   // Initialize and export the Connection class
   mssql::Connection::Init(env, exports);
   exports.Set("setLogLevel", Napi::Function::New(env, SetLogLevel));

cpp/include/common/platform.h

@@ -56,6 +56,7 @@ inline void UnlockMutex(PlatformMutex* mutex) {
 #elif defined(PLATFORM_LINUX) || defined(PLATFORM_MACOS)
 #include <pthread.h>
 #include <unistd.h>
+#include <signal.h>
 typedef pthread_mutex_t PlatformMutex;
 
 inline void InitializeMutex(PlatformMutex* mutex) {
@@ -116,4 +117,41 @@ class ScopedLock {
   // Prevent copying
   ScopedLock(const ScopedLock&) = delete;
   ScopedLock& operator=(const ScopedLock&) = delete;
-};
+};
+
+// Platform-specific signal handling
+namespace Platform {
+  // Initialize platform-specific signal handlers
+  inline void InitializeSignalHandlers() {
+#if defined(PLATFORM_LINUX) || defined(PLATFORM_MACOS)
+    // Ignore SIGPIPE on Unix-like systems to handle broken pipes gracefully
+    // This prevents the process from terminating when writing to a closed pipe
+    signal(SIGPIPE, SIG_IGN);
+#endif
+    // Windows doesn't have SIGPIPE, it returns EPIPE errors instead
+  }
+  
+  // Check if a stream is still valid for writing
+  template<typename Stream>
+  inline bool IsStreamGood(Stream& stream) {
+    return stream.good();
+  }
+  
+  // Safe stream write that handles broken pipes
+  template<typename Stream>
+  inline bool SafeStreamWrite(Stream& stream, const std::string& data) {
+    if (!IsStreamGood(stream)) {
+      return false;
+    }
+    
+    stream << data;
+    
+    // Only flush if the write succeeded
+    if (IsStreamGood(stream)) {
+      stream.flush();
+      return true;
+    }
+    
+    return false;
+  }
+}

cpp/src/utils/Logger.cpp

@@ -1,4 +1,5 @@
 #include "Logger.h"
+#include "common/platform.h"
 
 #include <chrono>
 #include <codecvt>
@@ -21,8 +22,13 @@ Logger::Logger() {}
 Logger::~Logger() {
   // Flush console output before destruction
   if (logToConsole_) {
-    std::cout.flush();
-    std::cerr.flush();
+    // Only flush if streams are still good (handles EPIPE)
+    if (Platform::IsStreamGood(std::cout)) {
+      std::cout.flush();
+    }
+    if (Platform::IsStreamGood(std::cerr)) {
+      std::cerr.flush();
+    }
   }
 
   if (logFile_.is_open()) {
@@ -102,8 +108,11 @@ void Logger::Log(LogLevel level, const std::string& message) {
   logLine << "[" << timestamp.str() << "] [CPP] [" << threadId.str() << "] [" << levelStr << "] " << message;
 
   if (logToConsole_) {
-    std::cout << logLine.str() << std::endl;
-    std::cout.flush();  // Ensure immediate flush for Node.js termination
+    // Write to appropriate stream based on log level
+    std::ostream& out = (level <= LogLevel::Warning) ? std::cerr : std::cout;
+    
+    // Use platform-safe stream writing that handles broken pipes
+    Platform::SafeStreamWrite(out, logLine.str() + "\n");
   }
 
   if (logFile_.is_open()) {

lib/logger.js

@@ -3,6 +3,19 @@ const { threadId, isMainThread } = require('worker_threads')
 const fs = require('fs')
 const path = require('path')
 
+// Handle EPIPE errors gracefully when piping output
+process.stdout.on('error', (err) => {
+  if (err.code === 'EPIPE') {
+    process.exit(0)
+  }
+})
+
+process.stderr.on('error', (err) => {
+  if (err.code === 'EPIPE') {
+    process.exit(0)
+  }
+})
+
 // Mirror the C++ LogLevel enum
 const LogLevel = {
   SILENT: 0,

samples/javascript/simple-logging-config.js

@@ -1,5 +1,10 @@
 'use strict'
 
+// Handle SIGPIPE gracefully when output is piped
+process.on('SIGPIPE', () => {
+  process.exit(0)
+})
+
 const sql = require('../../lib/sql')
 
 // Example 1: Enable trace logging for debugging