在Erlang中,有很多种方式去实现与其他语言的交互,常见的几种方式有
- 使用TCP协议交互
- 使用Port
- 使用Erl_Interface方式去实现
- CNode
- NIF
后面几种难度都是有的,也使用了比较复杂的C/C++,而且比较容易出现问题。TCP的方式是通过网络协议,个人也不是很喜欢,那就剩下Port方式去连接Erlang服务器。
在Erlang中使用Port非常简单,实际上就是通过标准输入输出流与外部程序就行交互。
现在我还是沿用官方文档中代码,仅做部分修改,实现Erlang与Golang的交互。
-module(complex1).
-export([start/1, stop/0, init/1]).
-export([foo/1, bar/1]).
start(ExtPrg) ->
spawn(?MODULE, init, [ExtPrg]).
stop() ->
complex ! stop.
foo(X) ->
call_port({foo, X}).
bar(Y) ->
call_port({bar, Y}).
call_port(Msg) ->
complex ! {call, self(), Msg},
receive
{complex, Result} ->
Result
end.
init(ExtPrg) ->
register(complex, self()),
process_flag(trap_exit, true),
%% 注意{packet, 2}代表的是用2个字节表示传输的
%% Port = open_port({spawn, ExtPrg}, [{packet, 2}]),
%% 在这里不去处理这个数据头,不需要这个参数
Port = open_port({spawn, ExtPrg}, []),
loop(Port).
loop(Port) ->
receive
{call, Caller, Msg} ->
Port ! {self(), {command, encode(Msg)}},
receive
{Port, {data, Data}} ->
Caller ! {complex, decode(Data)}
end,
loop(Port);
stop ->
Port ! {self(), close},
receive
{Port, closed} ->
exit(normal)
end;
{'EXIT', Port, Reason} ->
exit(port_terminated)
end.
%% 进行编码,将数据转成2进制
encode({foo, X}) -> <<1:8, X:8>>;
encode({bar, Y}) -> <<2:8, Y:8>>].
%% 这里有点特别,
decode(Data) -> erlang:list_to_binary(Data).
对于Golang,需要读取标准输入流的数据,这里我用bufio处理下(其实也没有太多必要,哈哈哈)。因为这里没有使用数据流的头部来记录数据的长度,所以默认将数据长度设置为2的byte数组。
package main
import (
"bufio"
"os"
)
func main() {
reader := bufio.NewReader(os.Stdin)
writer := bufio.NewWriter(os.Stdout)
for {
buff := make([]byte, 2)
_, err := reader.Read(buff)
if err != nil {
panic(err.Error())
}
switch int(buff[0]) {
case 1:
buff[1] = byte(foo(int(buff[1])))
case 2:
buff[1] = byte(bar(int(buff[1])))
}
writer.Write(buff[1:2])
// 这里需要注意,要用Flush进行处理,否则erlang端收不到信息
writer.Flush()
}
}
func foo(num int) int {
return num + 1
}
func bar(num int) int {
return 2 * num
}