使用 Go 构建 Node.js 插件的可能性探讨

[Node.js]  <-- 调用 -->  [C 语言插件 (.node)]  <-- IPC -->  [Go 程序]

package main

import (
    "fmt"
    "net"
    "os"
)

func main() {
    ln, err := net.Listen("tcp", ":8081")
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
    defer ln.Close()

    conn, err := ln.Accept()
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
    defer conn.Close()

    buf := make([]byte, 1024)
    n, err := conn.Read(buf)
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }

    message := string(buf[:n])
    fmt.Printf("Received: %s\n", message)

    response := "Hello from Go: " + message
    conn.Write([]byte(response))
}

#include <node_api.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>

napi_value CallGo(napi_env env, napi_callback_info info) {
  napi_status status;

  size_t argc = 1;
  napi_value args[1];
  status = napi_get_cb_info(env, info, &argc, args, NULL, NULL);
  if (status != napi_ok) {
    napi_throw_type_error(env, NULL, "Wrong number of arguments");
    return NULL;
  }

  napi_valuetype argtype;
  status = napi_typeof(env, args[0], &argtype);
  if (status != napi_ok || argtype != napi_string) {
    napi_throw_type_error(env, NULL, "Wrong argument type. String expected.");
    return NULL;
  }

  size_t str_len;
  status = napi_get_value_string_utf8(env, args[0], NULL, 0, &str_len);
  if (status != napi_ok) {
    napi_throw_error(env, NULL, "Failed to get string length");
    return NULL;
  }

  char *input_str = (char*)malloc(str_len + 1);
  if (input_str == NULL) {
    napi_throw_error(env, NULL, "Memory allocation failed");
    return NULL;
  }

  status = napi_get_value_string_utf8(env, args[0], input_str, str_len + 1, &str_len);
  if (status != napi_ok) {
    free(input_str);
    napi_throw_error(env, NULL, "Failed to get string value");
    return NULL;
  }


  int sock = 0, valread;
  struct sockaddr_in serv_addr;
  char buffer[1024] = {0};

  if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
    printf("\n Socket creation error \n");
    free(input_str);
    return NULL;
  }

  serv_addr.sin_family = AF_INET;
  serv_addr.sin_port = htons(8081);

  // Convert IPv4 and IPv6 addresses from text to binary form
  if(inet_pton(AF_INET, "127.0.0.1", &serv_addr.sin_addr)<=0) {
    printf("\nInvalid address/ Address not supported \n");
    free(input_str);
    return NULL;
  }

  if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) {
    printf("\nConnection Failed \n");
    close(sock);
    free(input_str);
    return NULL;
  }
  send(sock , input_str , strlen(input_str) , 0 );
  valread = read( sock , buffer, 1024);

  close(sock);
  free(input_str);

  napi_value result;
  status = napi_create_string_utf8(env, buffer, NAPI_AUTO_LENGTH, &result);
  if (status != napi_ok) {
    napi_throw_error(env, NULL, "Failed to create string");
    return NULL;
  }
  return result;
}

napi_value Init(napi_env env, napi_value exports) {
  napi_status status;
  napi_value fn;

  status = napi_create_function(env, NULL, 0, CallGo, NULL, &fn);
  if (status != napi_ok) {
    napi_throw_error(env, NULL, "Unable to create function");
    return NULL;
  }

  status = napi_set_named_property(env, exports, "callGo", fn);
  if (status != napi_ok) {
    napi_throw_error(env, NULL, "Unable to populate exports");
    return NULL;
  }

  return exports;
}

NAPI_MODULE(NODE_GYP_MODULE_NAME, Init)

const addon = require('./build/Release/c_addon');

// Start the Go program (you need to compile and run it separately)
// e.g., go run go_program.go

const result = addon.callGo("Hello from Node.js");
console.log('Result from Go:', result);