Switched to local functions

Author: ankane

src/pgvector.lua

@@ -34,28 +34,14 @@ end
 
 -- halfvec
 
-local halfvec_mt = {
-  pgmoon_serialize = function(v)
-    return 0, halfvec_serialize(v)
-  end
-}
-
-function pgvector.halfvec(v)
-  local vec = {}
-  for _, x in ipairs(v) do
-    table.insert(vec, x)
-  end
-  return setmetatable(vec, halfvec_mt)
-end
-
-function halfvec_serialize(v)
+local function halfvec_serialize(v)
   for _, v in ipairs(v) do
     assert(type(v) == "number")
   end
   return "[" .. table.concat(v, ",") .. "]"
 end
 
-function halfvec_deserialize(v)
+local function halfvec_deserialize(v)
   local vec = {}
   for x in string.gmatch(string.sub(v, 2, -2), "[^,]+") do
     table.insert(vec, tonumber(x))
@@ -64,37 +50,31 @@ function halfvec_deserialize(v)
   return setmetatable(vec, halfvec_mt)
 end
 
--- sparsevec
-
-local sparsevec_mt = {
+local halfvec_mt = {
   pgmoon_serialize = function(v)
-    return 0, sparsevec_serialize(v)
+    return 0, halfvec_serialize(v)
   end
 }
 
-function pgvector.sparsevec(elements, dim)
-  for k, v in pairs(elements) do
-    assert(type(k) == "number")
-    assert(type(v) == "number")
+function pgvector.halfvec(v)
+  local vec = {}
+  for _, x in ipairs(v) do
+    table.insert(vec, x)
   end
-  assert(type(dim) == "number")
-
-  local vec = {
-    elements = elements,
-    dim = dim
-  }
-  return setmetatable(vec, sparsevec_mt)
+  return setmetatable(vec, halfvec_mt)
 end
 
-function sparsevec_serialize(vec)
+-- sparsevec
+
+local function sparsevec_serialize(vec)
   local elements = {}
   for i, v in pairs(vec["elements"]) do
     table.insert(elements, tonumber(i) .. ":" .. tonumber(v))
   end
   return "{" .. table.concat(elements, ",") .. "}/" .. tonumber(vec["dim"])
 end
 
-function sparsevec_deserialize(v)
+local function sparsevec_deserialize(v)
   local m = string.gmatch(v, "[^/]+")
   local elements = {}
   for e in string.gmatch(string.sub(m(), 2, -2), "[^,]+") do
@@ -110,6 +90,26 @@ function sparsevec_deserialize(v)
   return setmetatable(vec, sparsevec_mt)
 end
 
+local sparsevec_mt = {
+  pgmoon_serialize = function(v)
+    return 0, sparsevec_serialize(v)
+  end
+}
+
+function pgvector.sparsevec(elements, dim)
+  for k, v in pairs(elements) do
+    assert(type(k) == "number")
+    assert(type(v) == "number")
+  end
+  assert(type(dim) == "number")
+
+  local vec = {
+    elements = elements,
+    dim = dim
+  }
+  return setmetatable(vec, sparsevec_mt)
+end
+
 -- register
 
 function pgvector.setup_vector(pg)