X-Agents/server/internal/service/document_parser_client.go

package service

import (
	"context"
	"fmt"
	"time"

	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"

	docparser "x-agents/server"
)

// AICoreClient AI-Core 文档解析服务客户端
type AICoreClient struct {
	conn    *grpc.ClientConn
	address string
}

// ParseResult 解析结果
type ParseResult struct {
	Success       bool
	Content       string
	Message       string
	ContentLength int32
	FileType      string
	ParserEngine  string
}

// NewAICoreClient 创建 AI-Core 客户端
func NewAICoreClient(address string) (*AICoreClient, error) {
	return &AICoreClient{address: address}, nil
}

// Connect 连接到 gRPC 服务
func (c *AICoreClient) Connect() error {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	conn, err := grpc.DialContext(ctx, c.address,
		grpc.WithTransportCredentials(insecure.NewCredentials()),
		grpc.WithBlock(),
	)
	if err != nil {
		return fmt.Errorf("failed to connect to AI-Core service: %w", err)
	}
	c.conn = conn
	return nil
}

// Close 关闭连接
func (c *AICoreClient) Close() {
	if c.conn != nil {
		c.conn.Close()
	}
}

// ParseDocument 解析文档 - 使用生成的 protobuf 代码
func (c *AICoreClient) ParseDocument(fileURL, fileName, fileType string) (*ParseResult, error) {
	if c.conn == nil {
		if err := c.Connect(); err != nil {
			return nil, err
		}
	}

	// 使用生成的 protobuf 客户端
	client := docparser.NewDocumentParserClient(c.conn)

	req := &docparser.ParseRequest{
		FileUrl:  fileURL,
		FileName: fileName,
		FileType: fileType,
	}

	ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
	defer cancel()

	resp, err := client.ParseDocument(ctx, req)
	if err != nil {
		return nil, fmt.Errorf("failed to parse document: %w", err)
	}

	return &ParseResult{
		Success:       resp.GetSuccess(),
		Content:       resp.GetContent(),
		Message:       resp.GetMessage(),
		ContentLength: resp.GetContentLength(),
		FileType:      resp.GetFileType(),
		ParserEngine:  resp.GetParserEngine(),
	}, nil
}