nats-exec: move out

This commit is contained in:
2026-02-21 15:59:53 +01:00
parent ef521f6a56
commit 24da672a58
4 changed files with 4 additions and 158 deletions

View File

@@ -38,5 +38,6 @@ inputs:
required: false
default: 'false'
runs:
# noinspection YAMLSchemaValidation
using: 'go'
main: 'cmd/nats-upload/main.go'
main: 'main.go'

View File

@@ -1,155 +0,0 @@
package main
import (
"context"
"fmt"
"io"
"log"
"os"
"path/filepath"
"runtime"
"strings"
"syscall"
"github.com/nats-io/nats.go"
"github.com/nats-io/nats.go/jetstream"
"golang.org/x/mod/semver"
)
func main() {
if len(os.Args) < 2 {
_, _ = fmt.Fprintf(os.Stderr, "Usage: %s <binary-name> [args...]\n", os.Args[0])
os.Exit(1)
}
binaryName := os.Args[1]
remainingArgs := os.Args[2:]
natsURL := getEnv("NATS_URL", "nats://localhost:4222")
bucketName := getEnv("NATS_BUCKET", "binaries")
cacheDir := getEnv("NATS_CACHE_DIR", filepath.Join(os.TempDir(), "nats-exec-cache"))
natsDomain := getEnv("NATS_JS_DOMAIN", "")
ctx := context.Background()
nc, err := nats.Connect(natsURL)
if err != nil {
log.Fatalf("Failed to connect to NATS: %v", err)
}
defer nc.Close()
js, err := createJetStream(nc, natsDomain)
if err != nil {
log.Fatalf("Failed to create JetStream context: %v", err)
}
store, err := js.ObjectStore(ctx, bucketName)
if err != nil {
log.Fatalf("Failed to get object store %s: %v", bucketName, err)
}
// Format: binary/arch/version
arch := runtime.GOARCH
objects, err := store.List(ctx)
if err != nil {
log.Fatalf("Failed to list objects: %v", err)
}
var latestVersion string
var latestKey string
for _, info := range objects {
// Key structure: <binaryName>/<arch>/<version>
parts := strings.Split(info.Name, "/")
if len(parts) != 3 {
continue
}
if parts[0] != binaryName {
continue
}
if parts[1] != arch {
continue
}
version := parts[2]
if !strings.HasPrefix(version, "v") {
version = "v" + version
}
if semver.IsValid(version) {
if latestVersion == "" || semver.Compare(version, latestVersion) > 0 {
latestVersion = version
latestKey = info.Name
}
}
}
if latestKey == "" {
log.Fatalf("No version of %s found for arch %s", binaryName, arch)
}
localPath := filepath.Join(cacheDir, latestKey)
if _, err := os.Stat(localPath); os.IsNotExist(err) {
log.Printf("Downloading %s version %s...", binaryName, latestVersion)
if err := downloadBinary(ctx, store, latestKey, localPath); err != nil {
log.Fatalf("Failed to download binary: %v", err)
}
} else {
log.Printf("Using cached %s version %s", binaryName, latestVersion)
}
if err := os.Chmod(localPath, 0755); err != nil {
log.Fatalf("Failed to make binary executable: %v", err)
}
fullPath, err := filepath.Abs(localPath)
if err != nil {
log.Fatalf("Failed to get absolute path: %v", err)
}
env := os.Environ()
args := append([]string{fullPath}, remainingArgs...)
err = syscall.Exec(fullPath, args, env)
if err != nil {
log.Fatalf("Failed to exec %s: %v", fullPath, err)
}
}
func downloadBinary(ctx context.Context, store jetstream.ObjectStore, key, localPath string) error {
if err := os.MkdirAll(filepath.Dir(localPath), 0755); err != nil {
return err
}
obj, err := store.Get(ctx, key)
if err != nil {
return err
}
f, err := os.OpenFile(localPath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0755)
if err != nil {
return err
}
defer f.Close()
_, err = io.Copy(f, obj)
return err
}
func getEnv(key, defaultValue string) string {
if value, ok := os.LookupEnv(key); ok {
return value
}
return defaultValue
}
func createJetStream(nc *nats.Conn, domain string) (jetstream.JetStream, error) {
if domain != "" {
return jetstream.NewWithDomain(nc, domain)
}
return jetstream.New(nc)
}

View File

@@ -59,13 +59,13 @@ build GOOS GOARCH:
-ldflags {{LD_FLAGS}} {{GO_OPTS}} \
-mod mod \
-o bin/{{OUTBIN}}{{SUFFIX}}_{{GOOS}}_{{GOARCH}} \
./cmd/{{OUTBIN}}/main.go
./main.go
build-race:
GOOS=linux GOARCH=amd64 CGO_ENABLED=1 go build -race \
-ldflags {{LD_FLAGS}} {{GO_OPTS}} \
-mod mod \
-o bin/{{OUTBIN}}{{SUFFIX}}_linux_amd64 \
./cmd/{{OUTBIN}}/main.go
./main.go
# Install binaries
install: