Unity Catalog 함수 (Functions)

CREATE [OR REPLACE] FUNCTION [IF NOT EXISTS]
  <catalog>.<schema>.<function_name>(<param_name> <data_type>, ...)
  RETURNS <return_type>
  [LANGUAGE SQL]
  [COMMENT '<설명>']
  RETURN <expression>;

-- 1. 이메일 도메인 추출 함수
CREATE OR REPLACE FUNCTION production.utils.get_email_domain(email STRING)
  RETURNS STRING
  COMMENT '이메일 주소에서 도메인 부분을 추출합니다'
  RETURN SPLIT(email, '@')[1];

-- 사용
SELECT production.utils.get_email_domain('alice@company.com');
-- 결과: 'company.com'

-- 2. 한국 전화번호 포맷팅 함수
CREATE OR REPLACE FUNCTION production.utils.format_phone_kr(phone STRING)
  RETURNS STRING
  COMMENT '전화번호를 010-XXXX-XXXX 형식으로 변환합니다'
  RETURN CONCAT(
    SUBSTR(REGEXP_REPLACE(phone, '[^0-9]', ''), 1, 3), '-',
    SUBSTR(REGEXP_REPLACE(phone, '[^0-9]', ''), 4, 4), '-',
    SUBSTR(REGEXP_REPLACE(phone, '[^0-9]', ''), 8, 4)
  );

-- 3. 나이 그룹 분류 함수
CREATE OR REPLACE FUNCTION production.utils.age_group(age INT)
  RETURNS STRING
  COMMENT '나이를 기반으로 연령대 그룹을 반환합니다'
  RETURN CASE
    WHEN age < 20 THEN '10대'
    WHEN age < 30 THEN '20대'
    WHEN age < 40 THEN '30대'
    WHEN age < 50 THEN '40대'
    WHEN age < 60 THEN '50대'
    ELSE '60대 이상'
  END;

-- 테이블 쿼리에서 사용
SELECT
  name,
  age,
  production.utils.age_group(age) AS age_group
FROM production.ecommerce.customers;

CREATE [OR REPLACE] FUNCTION <catalog>.<schema>.<function_name>(<param_name> <data_type>, ...)
  RETURNS <return_type>
  LANGUAGE PYTHON
  [COMMENT '<설명>']
  AS $$
    # Python 코드
    return result
  $$;

-- 1. 주민번호 유효성 검사 함수
CREATE OR REPLACE FUNCTION production.utils.validate_ssn(ssn STRING)
  RETURNS BOOLEAN
  LANGUAGE PYTHON
  COMMENT '한국 주민등록번호의 유효성을 검사합니다'
  AS $$
    import re
    if not re.match(r'^\d{6}-?\d{7}$', ssn):
        return False
    digits = re.sub(r'-', '', ssn)
    weights = [2, 3, 4, 5, 6, 7, 8, 9, 2, 3, 4, 5]
    total = sum(int(d) * w for d, w in zip(digits[:12], weights))
    check = (11 - total % 11) % 10
    return check == int(digits[12])
  $$;

-- 2. JSON 특정 필드 추출 함수
CREATE OR REPLACE FUNCTION production.utils.extract_json_field(
  json_str STRING,
  field_name STRING
)
  RETURNS STRING
  LANGUAGE PYTHON
  COMMENT 'JSON 문자열에서 지정된 필드 값을 추출합니다'
  AS $$
    import json
    try:
        data = json.loads(json_str)
        return str(data.get(field_name, ''))
    except (json.JSONDecodeError, TypeError):
        return None
  $$;

-- 날짜 범위를 생성하는 테이블 반환 함수
CREATE OR REPLACE FUNCTION production.utils.date_range(
  start_date DATE,
  end_date DATE
)
  RETURNS TABLE (dt DATE)
  RETURN SELECT EXPLODE(SEQUENCE(start_date, end_date, INTERVAL 1 DAY)) AS dt;

-- 사용: FROM 절에서 함수 호출
SELECT dt
FROM production.utils.date_range('2025-01-01', '2025-01-07');

# AI 에이전트에서 UC 함수를 도구로 사용하는 예시
from databricks.sdk import WorkspaceClient

w = WorkspaceClient()

# UC 함수를 에이전트 도구로 등록
tools = [
    {
        "type": "uc_function",
        "function": {
            "name": "production.utils.get_customer_info",
            "description": "고객 ID로 고객 정보를 조회합니다"
        }
    }
]

-- 분석가 그룹에게 유틸리티 함수 실행 권한 부여
GRANT EXECUTE ON FUNCTION production.utils.get_email_domain TO `analysts`;

-- 스키마 내 모든 함수 실행 권한 부여
GRANT EXECUTE ON SCHEMA production.utils TO `analysts`;

-- 데이터 엔지니어에게 함수 생성 권한 부여
GRANT CREATE FUNCTION ON SCHEMA production.utils TO `data_engineers`;

-- 권한 확인
SHOW GRANTS ON FUNCTION production.utils.get_email_domain;

-- 함수 목록 조회
SHOW FUNCTIONS IN production.utils;

-- 함수 상세 정보
DESCRIBE FUNCTION production.utils.get_email_domain;
DESCRIBE FUNCTION EXTENDED production.utils.get_email_domain;

-- 함수 삭제
DROP FUNCTION IF EXISTS production.utils.get_email_domain;

-- 1. 고객 등급 분류 (모든 팀이 동일 기준 사용해야 함)
CREATE OR REPLACE FUNCTION production.rules.customer_tier(
  total_spend DECIMAL(12,2),
  order_count INT,
  registration_days INT
)
RETURNS STRING
COMMENT '고객 등급을 산정합니다. 마케팅, CS, 분석 팀 모두 이 함수를 사용하세요.'
RETURN CASE
  WHEN total_spend >= 1000000 AND order_count >= 50 THEN 'VIP'
  WHEN total_spend >= 300000 AND order_count >= 20 THEN 'Gold'
  WHEN total_spend >= 100000 AND order_count >= 5 THEN 'Silver'
  ELSE 'Standard'
END;

-- 2. 매출 계산 (할인, 세금, 포인트 차감 등 복잡한 규칙)
CREATE OR REPLACE FUNCTION production.rules.net_revenue(
  gross_amount DECIMAL(12,2),
  discount_pct DECIMAL(5,2),
  tax_rate DECIMAL(5,2),
  point_used DECIMAL(12,2)
)
RETURNS DECIMAL(12,2)
COMMENT '순매출을 계산합니다. 할인, 세금, 포인트를 모두 반영합니다.'
RETURN ROUND(
  (gross_amount * (1 - discount_pct/100) - point_used) * (1 + tax_rate/100),
  2
);

-- 에이전트 도구용 함수: LLM 친화적 설계
CREATE OR REPLACE FUNCTION production.agent_tools.lookup_product(
  search_term STRING COMMENT '제품명 또는 제품코드. 예: "갤럭시 S24", "PRD-001"'
)
RETURNS TABLE(
  product_name STRING,
  price DECIMAL(10,2),
  stock_status STRING,
  description STRING
)
COMMENT '제품 정보를 검색합니다. 고객이 특정 제품에 대해 문의할 때 사용하세요.
검색어는 제품명의 일부 또는 제품코드를 입력할 수 있습니다.
정확한 제품코드를 아는 경우 "PRD-001" 형태로 입력하면 정확한 결과를 얻을 수 있습니다.
결과가 없으면 빈 테이블이 반환됩니다.'
RETURN
  SELECT
    name AS product_name,
    price,
    CASE
      WHEN stock > 0 THEN CONCAT('재고 있음 (', CAST(stock AS STRING), '개)')
      ELSE '품절'
    END AS stock_status,
    description
  FROM production.ecommerce.products
  WHERE name LIKE CONCAT('%', search_term, '%')
     OR product_code = search_term
  LIMIT 5;