parallel.go

// Copyright (c) 2026 John Dewey

// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:

// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.

// Package main demonstrates parallel task execution. Steps at the same
// DAG level run concurrently.
//
// DAG:
//
//	health-check
//	    ├── get-hostname
//	    ├── get-disk
//	    ├── get-memory
//	    ├── get-load
//	    └── get-uptime
//
// Run with: OSAPI_TOKEN="<jwt>" go run parallel.go
package main

import (
	"context"
	"fmt"
	"log"
	"os"

	osapi "github.com/retr0h/osapi/pkg/sdk/client"

	"github.com/osapi-io/osapi-orchestrator/pkg/orchestrator"
)

func main() {
	token := os.Getenv("OSAPI_TOKEN")
	if token == "" {
		log.Fatal("OSAPI_TOKEN is required")
	}

	url := os.Getenv("OSAPI_URL")
	if url == "" {
		url = "http://localhost:8080"
	}

	o := orchestrator.New(url, token)

	health := o.HealthCheck()

	// Five queries share the same dependency so the orchestrator
	// schedules them concurrently.
	o.NodeHostnameGet("_any").After(health)
	o.NodeDiskGet("_any").After(health)
	o.NodeMemoryGet("_any").After(health)
	o.NodeLoadGet("_any").After(health)
	o.NodeUptimeGet("_any").After(health)

	report, err := o.Run(context.Background())
	if err != nil {
		log.Fatal(err)
	}

	var h osapi.HostnameResult
	if err := report.Decode("get-hostname", &h); err == nil {
		fmt.Printf("Hostname: %s\n", h.Hostname)
	}

	var m osapi.MemoryResult
	if err := report.Decode("get-memory", &m); err == nil && m.Memory != nil {
		fmt.Printf("Memory: %.1f GB total\n", float64(m.Memory.Total)/(1024*1024*1024))
	}
}