title: 实现Golang和Erlang的链接(Port)html

categories: Golang

在Erlang中,有不少种方式去实现与其余语言的交互,常见的几种方式有golang

  1. 使用TCP协议交互
  2. 使用Port
  3. 使用Erl_Interface方式去实现
  4. CNode
  5. NIF

后面几种难度都是有的,也使用了比较复杂的C/C++,并且比较容易出现问题。TCP的方式是经过网络协议,我的也不是很喜欢,那就剩下Port方式去链接Erlang服务器。shell

在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数组。oop

// 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