mirror of
https://gitee.com/johng/gf.git
synced 2024-11-30 03:07:45 +08:00
58 lines
1.2 KiB
Go
58 lines
1.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 (
|
|
"github.com/gogf/gf/errors/gerror"
|
|
"github.com/gogf/gf/internal/json"
|
|
"github.com/gogf/gf/net/gtcp"
|
|
"io"
|
|
)
|
|
|
|
// Send sends data to specified process of given pid.
|
|
func Send(pid int, data []byte, group ...string) error {
|
|
msg := MsgRequest{
|
|
SendPid: Pid(),
|
|
RecvPid: pid,
|
|
Group: defaultGroupNameForProcComm,
|
|
Data: data,
|
|
}
|
|
if len(group) > 0 {
|
|
msg.Group = group[0]
|
|
}
|
|
msgBytes, err := json.Marshal(msg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var conn *gtcp.PoolConn
|
|
conn, err = getConnByPid(pid)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close()
|
|
// Do the sending.
|
|
var result []byte
|
|
result, err = conn.SendRecvPkg(msgBytes, gtcp.PkgOption{
|
|
Retry: gtcp.Retry{
|
|
Count: 3,
|
|
},
|
|
})
|
|
if len(result) > 0 {
|
|
response := new(MsgResponse)
|
|
if err = json.UnmarshalUseNumber(result, response); err == nil {
|
|
if response.Code != 1 {
|
|
err = gerror.New(response.Message)
|
|
}
|
|
}
|
|
}
|
|
// EOF is not really an error.
|
|
if err == io.EOF {
|
|
err = nil
|
|
}
|
|
return err
|
|
}
|