postgresml · sax · May 30, 2024
diff --git a/.gitignore b/.gitignore
@@ -27,6 +27,17 @@ share/python-wheels/
 *.egg
 MANIFEST
 
+# Erlang
+*.beam
+*.plt
+.erlang.cookie
+.eunit
+.rebar
+.rebar3
+_build
+priv/crates
+rebar3.crashdump
+
 # PyInstaller
 #  Usually these files are written by a python script from a template
 #  before PyInstaller builds the exe, so as to inject date/other infos into it.

diff --git a/pgml-sdks/pgml/erlang/Makefile b/pgml-sdks/pgml/erlang/Makefile
@@ -0,0 +1,44 @@
+REBAR3 ?= $(shell test -e `which rebar3` 2>/dev/null && which rebar3 || echo "./rebar3")
+
+.PHONY: deps test build
+
+all: build
+
+build: $(REBAR3)
+	@$(REBAR3) compile
+
+deps:
+	@$(REBAR3) deps
+
+fmt:
+	@$(REBAR3) fmt --write
+
+fmt-check:
+	@$(REBAR3) fmt --check
+
+shell:
+	@$(REBAR3) cargo clean
+	@$(REBAR3) shell
+
+clean:
+	@$(REBAR3) clean
+
+clean-all:
+	rm -rf $(CURDIR)/priv/crates
+	rm -rf $(CURDIR)/_build
+
+distclean: clean
+	@$(REBAR3) clean --all
+
+docs:
+	@$(REBAR3) edoc
+
+test:
+	@$(REBAR3) cargo clean
+	@$(REBAR3) eunit
+
+test-all:
+	@$(REBAR3) do eunit, ct, cover
+
+release: test
+	@$(REBAR3) as prod release
diff --git a/pgml-sdks/pgml/erlang/README.md b/pgml-sdks/pgml/erlang/README.md
@@ -0,0 +1,17 @@
+# pgml
+
+An Erlang NIF library written in Rust
+
+## Build
+
+``` shell
+make
+make fmt
+```
+
+## Test
+
+``` shell
+make test
+```
+
diff --git a/pgml-sdks/pgml/erlang/native/pgml/.cargo/config.toml b/pgml-sdks/pgml/erlang/native/pgml/.cargo/config.toml
@@ -0,0 +1,5 @@
+[target.'cfg(target_os = "macos")']
+rustflags = [
+    "-C", "link-arg=-undefined",
+    "-C", "link-arg=dynamic_lookup",
+]
diff --git a/pgml-sdks/pgml/erlang/native/pgml/Cargo.lock b/pgml-sdks/pgml/erlang/native/pgml/Cargo.lock
diff --git a/pgml-sdks/pgml/erlang/native/pgml/Cargo.toml b/pgml-sdks/pgml/erlang/native/pgml/Cargo.toml
@@ -0,0 +1,14 @@
+[package]
+name = "pgml"
+version = "0.1.0"
+edition = "2021"
+
+
+[lib]
+name = "pgml"
+path = "src/lib.rs"
+crate-type = ["dylib"]
+
+
+[dependencies]
+rustler = "0.33.0"
diff --git a/pgml-sdks/pgml/erlang/native/pgml/README.md b/pgml-sdks/pgml/erlang/native/pgml/README.md
@@ -0,0 +1,35 @@
+# NIF for pgml
+
+## To compile the NIF in Erlang:
+
+Add the following lines to your `rebar.config` file:
+
+``` erlang
+{plugins, [rebar3_rustler]}.
+
+{cargo_opts, [
+    {src_dir, "native/pgml"}
+]}.
+
+{provider_hooks, [
+    {pre, [
+        {compile, {cargo, build}}
+    ]},
+    {post, [
+        {clean, {cargo, clean}},
+        {eunit, {cargo, test}}
+    ]}
+]}.
+```
+
+## Build
+
+``` shell
+cargo build
+```
+
+## Test
+
+``` shell
+cargo test
+```
diff --git a/pgml-sdks/pgml/erlang/native/pgml/rust-toolchain.toml b/pgml-sdks/pgml/erlang/native/pgml/rust-toolchain.toml
@@ -0,0 +1,4 @@
+[toolchain]
+channel = "nightly"
+components = [ "rustfmt", "clippy" ]
+profile = "minimal"
diff --git a/pgml-sdks/pgml/erlang/native/pgml/src/lib.rs b/pgml-sdks/pgml/erlang/native/pgml/src/lib.rs
@@ -0,0 +1,4 @@
+rustler::init!("pgml", []);
+
+#[cfg(test)]
+mod tests {}
diff --git a/pgml-sdks/pgml/erlang/rebar.config b/pgml-sdks/pgml/erlang/rebar.config
@@ -0,0 +1,27 @@
+{project_plugins, [{erlfmt, "~> 1.3"}]}.
+
+{erl_opts, [debug_info]}.
+
+{deps, []}.
+
+{plugins, [rebar3_cargo]}.
+
+{cargo_opts, [
+    {src_dir, "native/pgml"}
+]}.
+
+{provider_hooks, [
+    {pre, [
+        {compile, {cargo, build}}
+    ]},
+    {post, [
+        {clean, {cargo, clean}},
+        {eunit, {cargo, test}}
+    ]}
+]}.
+
+{erlfmt, [
+    check,
+    verbose,
+    {files, ["src/*", "rebar.config"]}
+]}.
diff --git a/pgml-sdks/pgml/erlang/rebar.lock b/pgml-sdks/pgml/erlang/rebar.lock
@@ -0,0 +1 @@
+[].
diff --git a/pgml-sdks/pgml/erlang/src/cargo.hrl b/pgml-sdks/pgml/erlang/src/cargo.hrl
@@ -0,0 +1,14 @@
+-cargo_header_version(1).
+-ifndef(CARGO_LOAD_APP).
+-define(CARGO_LOAD_APP, pgml).
+-endif.
+-ifndef(CARGO_HRL).
+-define(CARGO_HRL, 1).
+-define(load_nif_from_crate(__CRATE, __INIT),
+    (fun() ->
+        __APP = ?CARGO_LOAD_APP,
+        __PATH = filename:join([code:priv_dir(__APP), "crates", __CRATE, __CRATE]),
+        erlang:load_nif(__PATH, __INIT)
+    end)()
+).
+-endif.
diff --git a/pgml-sdks/pgml/erlang/src/pgml.app.src b/pgml-sdks/pgml/erlang/src/pgml.app.src
@@ -0,0 +1,14 @@
+{application, pgml, [
+    {description, "An OTP library"},
+    {vsn, "0.1.0"},
+    {registered, []},
+    {applications, [
+        kernel,
+        stdlib
+    ]},
+    {env, []},
+    {modules, []},
+
+    {licenses, ["Apache 2.0"]},
+    {links, []}
+]}.
diff --git a/pgml-sdks/pgml/erlang/src/pgml.erl b/pgml-sdks/pgml/erlang/src/pgml.erl
@@ -0,0 +1,33 @@
+-module(pgml).
+
+-export([]).
+
+-include("cargo.hrl").
+
+-on_load init/0.
+
+-define(NOT_LOADED, not_loaded(?LINE)).
+
+%%%===================================================================
+%%% API
+%%%===================================================================
+
+%%%===================================================================
+%%% NIF
+%%%===================================================================
+
+init() ->
+  ?load_nif_from_crate(pgml, 0).
+
+not_loaded(Line) ->
+  erlang:nif_error({not_loaded, [{module, ?MODULE}, {line, Line}]}).
+
+%%%===================================================================
+%%% Tests
+%%%===================================================================
+
+-ifdef(TEST).
+
+-include_lib("eunit/include/eunit.hrl").
+
+-endif.