Initial import from original repo

.devcontainer/Dockerfile

@@ -0,0 +1,20 @@
+FROM mcr.microsoft.com/devcontainers/go:1-1.22-bookworm
+
+# [Optional] Uncomment this section to install additional OS packages.
+RUN apt-get update && export DEBIAN_FRONTEND=noninteractive \
+    && apt-get -y install --no-install-recommends \
+    postgresql-client \
+    sqlite3 \
+    && rm -rf /var/lib/apt/lists/*
+
+# [Optional] Uncomment the next lines to use go get to install anything else you need
+USER vscode
+RUN go install google.golang.org/protobuf/cmd/protoc-gen-go@v1.33.0 \
+    && go install github.com/vektra/mockery/v2@v2.43.2 \
+    && go install github.com/golangci/golangci-lint/cmd/golangci-lint@v1.59.1 \
+    && go install github.com/magefile/mage@v1.15.0 \
+    && go clean -cache -modcache
+USER root
+
+# [Optional] Uncomment this line to install global node packages.
+# RUN su vscode -c "source /usr/local/share/nvm/nvm.sh && npm install -g <your-package-here>" 2>&1

.devcontainer/devcontainer.json

@@ -0,0 +1,66 @@
+// For format details, see https://aka.ms/devcontainer.json.
+{
+	"name": "MLflow Go",
+	"dockerComposeFile": "docker-compose.yml",
+	"service": "app",
+	"workspaceFolder": "/workspaces/${localWorkspaceFolderBasename}",
+
+	// Features to add to the dev container. More info: https://containers.dev/features.
+	"features": {
+		"ghcr.io/devcontainers/features/github-cli:1": {},
+		"ghcr.io/devcontainers/features/python:1": {
+			"version": "3.8"
+		},
+		"ghcr.io/devcontainers/features/docker-in-docker:2": {},
+		"ghcr.io/devcontainers-contrib/features/k6:1": {},
+		"ghcr.io/devcontainers-contrib/features/pre-commit:2": {},
+		"ghcr.io/devcontainers-contrib/features/protoc-asdf:1": {
+			"version": "26.0"
+		},
+		"ghcr.io/devcontainers-contrib/features/ruff:1": {}
+	},
+
+	// Configure tool-specific properties.
+	"customizations": {
+		"vscode": {
+			"settings": {
+				"terminal.integrated.defaultProfile.linux": "zsh",
+				"editor.rulers": [
+					80,
+					100
+				],
+				"editor.formatOnSave": true,
+				"git.alwaysSignOff": true,
+				"go.lintTool": "golangci-lint",
+				"gopls": {
+					"formatting.local": "github.com/mlflow/mlflow-go",
+					"formatting.gofumpt": true,
+					"build.buildFlags": ["-tags=mage"]
+				},
+				"[python]": {
+					"editor.codeActionsOnSave": {
+						"source.fixAll": "explicit",
+						"source.organizeImports": "explicit"
+					},
+					"editor.defaultFormatter": "charliermarsh.ruff"
+				}
+			},
+			"extensions": [
+				"charliermarsh.ruff",
+				"golang.Go",
+				"humao.rest-client",
+				"pbkit.vscode-pbkit",
+				"tamasfe.even-better-toml"
+			]
+		}
+	},
+
+	// Use 'forwardPorts' to make a list of ports inside the container available locally.
+	// "forwardPorts": [5432],
+
+	// Use 'postCreateCommand' to run commands after the container is created.
+	"postCreateCommand": ".devcontainer/postCreate.sh"
+
+	// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
+	// "remoteUser": "root"
+}

.devcontainer/docker-compose.yml

@@ -0,0 +1,38 @@
+volumes:
+  go-cache:
+  postgres-data:
+
+services:
+  app:
+    build: 
+      context: .
+      dockerfile: Dockerfile
+    volumes:
+      - ../..:/workspaces:cached
+      - go-cache:/var/cache/go
+    environment:
+      - GOCACHE=/var/cache/go/build
+      - GOMODCACHE=/var/cache/go/mod
+
+    # Overrides default command so things don't shut down after the process ends.
+    command: sleep infinity
+
+    # Runs app on the same network as the database container, allows "forwardPorts" in devcontainer.json function.
+    network_mode: service:db
+
+    # Use "forwardPorts" in **devcontainer.json** to forward an app port locally. 
+    # (Adding the "ports" property to this file will not forward from a Codespace.)
+
+  db:
+    image: postgres:latest
+    restart: unless-stopped
+    volumes:
+      - postgres-data:/var/lib/postgresql/data
+    environment:
+      - POSTGRES_USER=postgres
+      - POSTGRES_PASSWORD=postgres
+      - POSTGRES_DB=postgres
+      - POSTGRES_HOSTNAME=localhost=value
+
+    # Add "forwardPorts": ["5432"] to **devcontainer.json** to forward PostgreSQL locally.
+    # (Adding the "ports" property to this file will not forward from a Codespace.)

.devcontainer/postCreate.sh

@@ -0,0 +1,7 @@
+#!/bin/sh
+
+# Fix permissions for the Go cache
+sudo chown -R $(id -u):$(id -g) /var/cache/go
+
+# Install precommit (https://pre-commit.com/)
+pre-commit install -t pre-commit -t prepare-commit-msg

.github/dependabot.yml

@@ -0,0 +1,16 @@
+# To get started with Dependabot version updates, you'll need to specify which
+# package ecosystems to update and where the package manifests are located.
+# Please see the documentation for more information:
+# https://docs.github.com/github/administering-a-repository/configuration-options-for-dependency-updates
+# https://containers.dev/guide/dependabot
+
+version: 2
+updates:
+  - package-ecosystem: "devcontainers"
+    directory: "/"
+    schedule:
+      interval: weekly
+  - package-ecosystem: "github-actions"
+    directory: "/"
+    schedule:
+      interval: weekly

.github/workflows/ci.yml

@@ -0,0 +1,18 @@
+name: CI
+
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+  schedule:
+    # Run daily at 01:34 so we get notified if CI is broken before a pull request
+    # is submitted.
+    - cron: "34 1 * * *"
+
+jobs:
+  lint:
+    uses: ./.github/workflows/lint.yml
+
+  test:
+    uses: ./.github/workflows/test.yml

.github/workflows/lint.yml

@@ -0,0 +1,22 @@
+name: Lint
+
+on:
+  workflow_call:
+
+permissions:
+  contents: read
+
+jobs:
+  lint:
+    name: Lint
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - name: Setup Go
+        uses: actions/setup-go@v5
+        with:
+          go-version: "1.22"
+          check-latest: true
+          cache: false
+      - name: Run pre-commit hooks
+        run: pipx run pre-commit run --all-files

.github/workflows/test.yml

@@ -0,0 +1,59 @@
+name: Test
+
+on:
+  workflow_call:
+
+permissions:
+  contents: read
+
+jobs:
+  go:
+    name: Test Go
+    strategy:
+      matrix:
+        runner: [macos-latest, ubuntu-latest, windows-latest]
+    runs-on: ${{ matrix.runner }}
+    steps:
+      - uses: actions/checkout@v4
+      - name: Setup Go
+        uses: actions/setup-go@v5
+        with:
+          go-version: "1.22"
+          check-latest: true
+          cache: false
+      - name: Install mage
+        run: go install github.com/magefile/mage@v1.15.0
+      - name: Run unit tests
+        run: mage test:unit
+
+  python:
+    name: Test Python
+    strategy:
+      matrix:
+        runner: [macos-latest, ubuntu-latest, windows-latest]
+        python: ["3.8", "3.9", "3.10", "3.11", "3.12"]
+    runs-on: ${{ matrix.runner }}
+    steps:
+      - uses: actions/checkout@v4
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python }}
+      - name: Setup Go
+        uses: actions/setup-go@v5
+        with:
+          go-version: "1.22"
+          check-latest: true
+          cache: false
+      - name: Install mage
+        run: go install github.com/magefile/mage@v1.15.0
+      - name: Install our package in editable mode
+        run: pip install -e .
+      - name: Initialize MLflow repo
+        run: mage repo:init
+      - name: Install dependencies
+        run: pip install pytest==8.1.1 psycopg2-binary -e .mlflow.repo
+      - name: Run integration tests
+        run: mage test:python
+        # Temporary workaround for failing tests
+        continue-on-error: ${{ matrix.runner != 'ubuntu-latest' }}

.gitignore

@@ -0,0 +1,16 @@
+# Artifacts
+dist/
+*.egg-info/
+*.so
+
+# Runs
+mlruns/
+
+# Cache
+__pycache__/
+
+# MLflow repo
+.mlflow.repo/
+
+# JetBrains
+.idea

.golangci.yml

@@ -0,0 +1,55 @@
+run:
+  build-tags:
+    - mage
+  timeout: 5m
+
+linters:
+  enable:
+    - errcheck
+    - gosimple
+    - lll
+  disable:
+    - depguard
+    - gochecknoglobals # Immutable globals are fine.
+    - exhaustruct # Often the case for protobuf generated code or gorm structs.
+    - protogetter # We do want to use pointers for memory optimization.
+  presets:
+    - bugs
+    - comment
+    - complexity
+    - error
+    - format
+    - import
+    - metalinter
+    - module
+    - performance
+    - sql
+    - style
+    - test
+    - unused
+
+linters-settings:
+  gci:
+    custom-order: true
+
+    sections:
+      - standard # Standard section: captures all standard packages.
+      - default # Default section: contains all imports that could not be matched to another section type.
+      - alias # Alias section: contains all alias imports. This section is not present unless explicitly enabled.
+      - prefix(github.com/mlflow/mlflow-go) # Custom section: groups all imports with the specified Prefix.
+      - blank # Blank section: contains all blank imports. This section is not present unless explicitly enabled.
+      - dot # Dot section: contains all dot imports. This section is not present unless explicitly enabled.
+
+  gofumpt:
+    module-path: github.com/mlflow/mlflow-go
+    extra-rules: true
+
+  tagliatelle:
+    case:
+      rules:
+        json: snake
+
+issues:
+  exclude-files:
+    - ".*\\.g\\.go$"
+    - ".*\\.pb\\.go$"

.mlflow.ref

@@ -0,0 +1 @@
+https://github.com/mlflow/mlflow.git#master

.mockery.yaml

@@ -0,0 +1,8 @@
+dir: "{{ .InterfaceDir }}"
+filename: "mock_{{ .InterfaceNameSnake }}.go"
+with-expecter: true
+inpackage: true
+packages:
+  github.com/mlflow/mlflow-go/pkg/tracking/store:
+    interfaces:
+      TrackingStore:

.pre-commit-config.yaml

@@ -0,0 +1,34 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.5.0
+    hooks:
+      - id: end-of-file-fixer
+        files: \.(proto|txt|sh|rst)$
+  - repo: https://github.com/golangci/golangci-lint
+    rev: "v1.59.1"
+    hooks:
+      - id: golangci-lint-full
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.5.7
+    hooks:
+      - id: ruff
+        types_or: [python, pyi, jupyter]
+        args: [--fix]
+      - id: ruff-format
+        types_or: [python, pyi, jupyter]
+  - repo: local
+    hooks:
+      # - id: rstcheck
+      #   name: rstcheck
+      #   entry: rstcheck
+      #   language: system
+      #   files: README.rst
+      #   stages: [commit]
+      #   require_serial: true
+
+      - id: must-have-signoff
+        name: must-have-signoff
+        entry: 'grep "Signed-off-by:"'
+        language: system
+        stages: [prepare-commit-msg]
+        require_serial: true