乐趣区

实现Golang和Erlang的连接(Port)

title: 实现 Golang 和 Erlang 的连接(Port)
categories: Golang
在 Erlang 中,有很多种方式去实现与其他语言的交互,常见的几种方式有

使用 TCP 协议交互
使用 Port
使用 Erl_Interface 方式去实现
CNode
NIF

后面几种难度都是有的,也使用了比较复杂的 C /C++,而且比较容易出现问题。TCP 的方式是通过网络协议,个人也不是很喜欢,那就剩下 Port 方式去连接 Erlang 服务器。
Erlang 的官方文档中对 Port 的介绍
在 Erlang 中使用 Port 非常简单,实际上就是通过标准输入输出流与外部程序就行交互。
现在我还是沿用官方文档中代码,仅做部分修改,实现 Erlang 与 Golang 的交互。
%% complex1.erl

-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 数组。
// port.go

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
}

使用
步骤 1 编译 golang
go build port.go
步骤 2 运行 Erlang 虚拟机
unix> erl
Erlang (BEAM) emulator version 4.9.1.2

Eshell V4.9.1.2 (abort with ^G)
1> c(complex1).
{ok,complex1}
步骤 3 运行
2> complex1:start(“extprg”).
<0.34.0>
3> complex1:foo(3).
4
4> complex1:bar(5).
10
5> complex1:stop().
stop

退出移动版