mirror of
https://gitee.com/johng/gf.git
synced 2024-12-02 04:07:47 +08:00
104 lines
3.2 KiB
Go
104 lines
3.2 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package gproc
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/gogf/gf/v2/container/gmap"
|
|
"github.com/gogf/gf/v2/errors/gcode"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/internal/intlog"
|
|
"github.com/gogf/gf/v2/net/gtcp"
|
|
"github.com/gogf/gf/v2/os/gfile"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
// MsgRequest is the request structure for process communication.
|
|
type MsgRequest struct {
|
|
SendPid int // Sender PID.
|
|
RecvPid int // Receiver PID.
|
|
Group string // Message group name.
|
|
Data []byte // Request data.
|
|
}
|
|
|
|
// MsgResponse is the response structure for process communication.
|
|
type MsgResponse struct {
|
|
Code int // 1: OK; Other: Error.
|
|
Message string // Response message.
|
|
Data []byte // Response data.
|
|
}
|
|
|
|
const (
|
|
defaultFolderNameForProcComm = "gf_pid_port_mapping" // Default folder name for storing pid to port mapping files.
|
|
defaultGroupNameForProcComm = "" // Default group name.
|
|
defaultTcpPortForProcComm = 10000 // Starting port number for receiver listening.
|
|
maxLengthForProcMsgQueue = 10000 // Max size for each message queue of the group.
|
|
)
|
|
|
|
var (
|
|
// commReceiveQueues is the group name to queue map for storing received data.
|
|
// The value of the map is type of *gqueue.Queue.
|
|
commReceiveQueues = gmap.NewStrAnyMap(true)
|
|
|
|
// commPidFolderPath specifies the folder path storing pid to port mapping files.
|
|
commPidFolderPath string
|
|
)
|
|
|
|
func init() {
|
|
availablePaths := []string{
|
|
"/var/tmp",
|
|
"/var/run",
|
|
}
|
|
if homePath, _ := gfile.Home(); homePath != "" {
|
|
availablePaths = append(availablePaths, gfile.Join(homePath, ".config"))
|
|
}
|
|
availablePaths = append(availablePaths, gfile.TempDir())
|
|
for _, availablePath := range availablePaths {
|
|
checkPath := gfile.Join(availablePath, defaultFolderNameForProcComm)
|
|
if !gfile.Exists(checkPath) && gfile.Mkdir(checkPath) != nil {
|
|
continue
|
|
}
|
|
if gfile.IsWritable(checkPath) {
|
|
commPidFolderPath = checkPath
|
|
break
|
|
}
|
|
}
|
|
if commPidFolderPath == "" {
|
|
intlog.Errorf(
|
|
context.TODO(),
|
|
`cannot find available folder for storing pid to port mapping files in paths: %+v, process communication feature will fail`,
|
|
availablePaths,
|
|
)
|
|
}
|
|
|
|
}
|
|
|
|
// getConnByPid creates and returns a TCP connection for specified pid.
|
|
func getConnByPid(pid int) (*gtcp.PoolConn, error) {
|
|
port := getPortByPid(pid)
|
|
if port > 0 {
|
|
if conn, err := gtcp.NewPoolConn(fmt.Sprintf("127.0.0.1:%d", port)); err == nil {
|
|
return conn, nil
|
|
} else {
|
|
return nil, err
|
|
}
|
|
}
|
|
return nil, gerror.NewCodef(gcode.CodeOperationFailed, "could not find port for pid: %d", pid)
|
|
}
|
|
|
|
// getPortByPid returns the listening port for specified pid.
|
|
// It returns 0 if no port found for the specified pid.
|
|
func getPortByPid(pid int) int {
|
|
return gconv.Int(gfile.GetContentsWithCache(getCommFilePath(pid)))
|
|
}
|
|
|
|
// getCommFilePath returns the pid to port mapping file path for given pid.
|
|
func getCommFilePath(pid int) string {
|
|
return gfile.Join(commPidFolderPath, gconv.String(pid))
|
|
}
|