unkinben
49d514c050
Forgebot is a K8s operator + API service for dispatching AI agent jobs from git forge commands. Includes: - CRDs: AgentPool, AgentTask, ProviderQueue, RepositoryBinding - API server with webhook handler, task queue, and comment proxy - Operator controllers for task scheduling and job management - Gitea provider with webhook parsing and signature verification - PostgreSQL database with auto-migration - Woodpecker CI pipelines and multi-stage Dockerfiles
58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/log"
|
|
|
|
forgebotv1alpha1 "git.unkin.net/unkin/forgebot/api/v1alpha1"
|
|
)
|
|
|
|
type AgentPoolReconciler struct {
|
|
client.Client
|
|
Scheme *runtime.Scheme
|
|
}
|
|
|
|
// +kubebuilder:rbac:groups=forgebot.unkin.net,resources=agentpools,verbs=get;list;watch;create;update;patch;delete
|
|
// +kubebuilder:rbac:groups=forgebot.unkin.net,resources=agentpools/status,verbs=get;update;patch
|
|
|
|
func (r *AgentPoolReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
|
|
logger := log.FromContext(ctx)
|
|
|
|
var pool forgebotv1alpha1.AgentPool
|
|
if err := r.Get(ctx, req.NamespacedName, &pool); err != nil {
|
|
return ctrl.Result{}, client.IgnoreNotFound(err)
|
|
}
|
|
|
|
var taskList forgebotv1alpha1.AgentTaskList
|
|
if err := r.List(ctx, &taskList, client.InNamespace(req.Namespace)); err != nil {
|
|
return ctrl.Result{}, err
|
|
}
|
|
|
|
active := 0
|
|
for _, task := range taskList.Items {
|
|
if task.Spec.PoolRef == pool.Name && task.Status.Phase == forgebotv1alpha1.TaskRunning {
|
|
active++
|
|
}
|
|
}
|
|
|
|
if pool.Status.ActiveJobs != active {
|
|
pool.Status.ActiveJobs = active
|
|
if err := r.Status().Update(ctx, &pool); err != nil {
|
|
return ctrl.Result{}, err
|
|
}
|
|
logger.Info("updated pool status", "pool", pool.Name, "active", active)
|
|
}
|
|
|
|
return ctrl.Result{}, nil
|
|
}
|
|
|
|
func (r *AgentPoolReconciler) SetupWithManager(mgr ctrl.Manager) error {
|
|
return ctrl.NewControllerManagedBy(mgr).
|
|
For(&forgebotv1alpha1.AgentPool{}).
|
|
Complete(r)
|
|
}
|