package com.superbiz.agent.service;

import io.milvus.client.MilvusServiceClient;
import io.milvus.param.ConnectParam;
import io.milvus.param.R;
import io.milvus.param.collection.HasCollectionParam;
import org.junit.jupiter.api.Test;

/**
 * 简单的 Milvus 连接测试
 */
public class SimpleMilvusTest {

    @Test
    public void testConnection() {
        String host = "in03-4a578da0f27ce9d.serverless.aws-eu-central-1.cloud.zilliz.com";
        int port = 443;
        String token = "d246a77f43a109685596e3c68ecfd359e1cd8b29d35c41d708160f02b97ae623d2ab392738df740d0c77b58ca1bbadfa7c412140";

        System.out.println("尝试连接 Milvus...");
        System.out.println("Host: " + host);
        System.out.println("Port: " + port);

        try {
            MilvusServiceClient client = new MilvusServiceClient(
                ConnectParam.newBuilder()
                    .withHost(host)
                    .withPort(port)
                    .withToken(token)
                    .withSecure(true)
                    .withConnectTimeout(10L, java.util.concurrent.TimeUnit.SECONDS)
                    .build()
            );

            System.out.println("✓ 客户端创建成功");

            // 测试连接：查询是否存在某个 collection
            R<Boolean> response = client.hasCollection(
                HasCollectionParam.newBuilder()
                    .withCollectionName("test_collection")
                    .build()
            );

            System.out.println("✓ 连接成功！");
            System.out.println("Status Code: " + response.getStatus());

            if (response.getStatus() == 0 || response.getStatus() == io.milvus.param.R.Status.Success.getCode()) {
                System.out.println("✓ Milvus 集群状态：正常运行");
            } else {
                System.out.println("✗ 响应状态异常: " + response.getStatus());
            }

            client.close();
            System.out.println("✓ 连接已关闭");

        } catch (Exception e) {
            System.err.println("✗ 连接失败：" + e.getMessage());
            e.printStackTrace();
            throw new RuntimeException("Milvus 连接失败", e);
        }
    }
}