INNER CODE UNIT · Go

buildAISessionRegisterHook

yaklang/yaklang · cmd/legion-smoke-node/main.go:362

func buildAISessionRegisterHook() func(node.SessionState) {
	sessionmgrURL := strings.TrimSpace(os.Getenv("LEGION_SESSIONMGR_URL"))
	sessionID := strings.TrimSpace(os.Getenv("LEGION_AI_SESSION_ID"))
	if sessionmgrURL == "" || sessionID == "" {
		log.Printf("ai_session register hook disabled: LEGION_SESSIONMGR_URL or LEGION_AI_SESSION_ID not set")
		return nil
	}
	client := &http.Client{Timeout: 10 * time.Second}
	return func(session node.SessionState) {
		finish := startStartupStage(true, "sessionmgr_register")
		body, err := json.Marshal(map[string]string{
			"node_id":         session.NodeID,
			"node_session_id": session.SessionID,
		})
		if err != nil {
			finish(err)
			log.Printf("ai_session register marshal: %v", err)
			return

View source record →