File size: 1,320 Bytes
8da2481 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | defmodule Plausible.MigrationUtils do
@moduledoc """
Base module for to use in Clickhouse migrations
"""
use Plausible
alias Plausible.IngestRepo
def on_cluster_statement(table) do
if(IngestRepo.clustered_table?(table), do: "ON CLUSTER '{cluster}'", else: "")
end
# See https://clickhouse.com/docs/en/sql-reference/dictionaries#clickhouse for context
def dictionary_connection_params() do
IngestRepo.config()
|> Enum.map(fn
{:database, database} -> "DB '#{database}'"
{:username, username} -> "USER '#{username}'"
{:password, password} -> "PASSWORD '#{password}'"
_ -> nil
end)
|> Enum.reject(&is_nil/1)
|> Enum.join(" ")
end
def table_settings() do
IngestRepo.config()
|> Keyword.get(:table_settings)
|> Enum.reject(fn {_, v} -> is_nil(v) end)
end
def table_settings_expr(type \\ :prefix) do
expr = Enum.map_join(table_settings(), ", ", fn {k, v} -> "#{k} = #{encode(v)}" end)
case {table_settings(), type} do
{[], _} -> ""
{_, :prefix} -> "SETTINGS #{expr}"
{_, :suffix} -> ", #{expr}"
end
end
def enterprise_edition?(), do: ee?()
def community_edition?(), do: ce?()
defp encode(value) when is_number(value), do: value
defp encode(value) when is_binary(value), do: "'#{value}'"
end
|