go-svc-code/cmd/test-sdk/main.go

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"net"
	"net/http"
	"os"
	"os/exec"
	"strconv"
	"time"

	baseopencode "git.x2erp.com/qdy/go-base/sdk/opencode"
)

func main() {
	fmt.Println("=== OpenCode SDK 集成测试 ===")

	// 1. 获取可用端口
	port, err := getAvailablePort()
	if err != nil {
		fmt.Printf("获取可用端口失败: %v\n", err)
		os.Exit(1)
	}
	fmt.Printf("使用端口: %d\n", port)

	// 2. 启动 opencode 服务
	cmd := exec.Command("opencode", "serve",
		"--hostname", "127.0.0.1",
		"--port", strconv.Itoa(port),
		"--log-level", "INFO",
	)

	if err := cmd.Start(); err != nil {
		fmt.Printf("启动 opencode 失败: %v\n", err)
		os.Exit(1)
	}
	defer func() {
		fmt.Println("停止 opencode 服务...")
		if cmd.Process != nil {
			cmd.Process.Kill()
			cmd.Wait()
		}
	}()

	fmt.Println("启动 opencode 服务...")

	// 3. 等待服务就绪
	baseURL := fmt.Sprintf("http://127.0.0.1:%d", port)
	if err := waitForServiceReady(baseURL); err != nil {
		fmt.Printf("等待服务就绪失败: %v\n", err)
		os.Exit(1)
	}

	fmt.Println("服务已就绪，开始测试 SDK...")

	// 4. 创建 SDK 客户端
	client, err := baseopencode.NewClientWithResponses(baseURL)
	if err != nil {
		fmt.Printf("创建客户端失败: %v\n", err)
		os.Exit(1)
	}

	ctx := context.Background()

	// 5. 测试创建会话
	fmt.Println("\n1. 测试创建会话...")
	title := "SDK测试会话-" + time.Now().Format("150405")
	reqBody := baseopencode.SessionCreateJSONRequestBody{
		Title: &title,
	}

	resp, err := client.SessionCreateWithResponse(ctx, &baseopencode.SessionCreateParams{}, reqBody)
	if err != nil {
		fmt.Printf("创建会话请求失败: %v\n", err)
		os.Exit(1)
	}

	if resp.StatusCode() != http.StatusOK {
		fmt.Printf("创建会话失败，状态码: %d\n", resp.StatusCode())
		fmt.Printf("响应体: %s\n", string(resp.Body))
		os.Exit(1)
	}

	var session baseopencode.Session
	if err := json.Unmarshal(resp.Body, &session); err != nil {
		fmt.Printf("解析会话响应失败: %v\n", err)
		os.Exit(1)
	}

	fmt.Printf("创建会话成功: ID=%s, Title=%s\n", session.Id, session.Title)

	// 6. 测试发送消息（暂时跳过，因为Parts_Item结构复杂）
	fmt.Println("\n2. 测试发送消息（跳过，Parts_Item结构复杂）...")
	fmt.Println("注意：需要进一步研究如何正确构造Parts_Item")

	// 7. 测试获取会话列表
	fmt.Println("\n3. 测试获取会话列表...")
	listResp, err := client.SessionListWithResponse(ctx, &baseopencode.SessionListParams{})
	if err != nil {
		fmt.Printf("获取会话列表请求失败: %v\n", err)
		os.Exit(1)
	}

	if listResp.StatusCode() != http.StatusOK {
		fmt.Printf("获取会话列表失败，状态码: %d\n", listResp.StatusCode())
		os.Exit(1)
	}

	var sessions []baseopencode.Session
	if err := json.Unmarshal(listResp.Body, &sessions); err != nil {
		fmt.Printf("解析会话列表失败: %v\n", err)
		os.Exit(1)
	}

	fmt.Printf("获取会话列表成功: 共 %d 个会话\n", len(sessions))
	for i, s := range sessions {
		fmt.Printf("  %d. ID=%s, Title=%s\n", i+1, s.Id, s.Title)
	}

	// 8. 测试健康检查
	fmt.Println("\n4. 测试健康检查...")
	healthResp, err := http.Get(baseURL + "/global/health")
	if err != nil {
		fmt.Printf("健康检查失败: %v\n", err)
		os.Exit(1)
	}
	defer healthResp.Body.Close()

	if healthResp.StatusCode != http.StatusOK {
		fmt.Printf("健康检查失败，状态码: %d\n", healthResp.StatusCode)
		os.Exit(1)
	}

	fmt.Println("健康检查通过")

	fmt.Println("\n=== 所有测试通过 ===")
}

func getAvailablePort() (int, error) {
	listener, err := net.Listen("tcp", "127.0.0.1:0")
	if err != nil {
		return 0, err
	}
	defer listener.Close()

	addr := listener.Addr().(*net.TCPAddr)
	return addr.Port, nil
}

func waitForServiceReady(baseURL string) error {
	client := &http.Client{Timeout: 5 * time.Second}
	url := baseURL + "/global/health"

	for i := 0; i < 30; i++ {
		resp, err := client.Get(url)
		if err == nil && resp.StatusCode == http.StatusOK {
			resp.Body.Close()
			return nil
		}
		if resp != nil {
			resp.Body.Close()
		}
		time.Sleep(100 * time.Millisecond)
	}

	return fmt.Errorf("服务在3秒内未就绪")
}