package impl

import (
	"bufio"
	"context"
	"fmt"
	"io"
	"os"

	"gitee.com/go-course/go11/skill/rpc/protobuf/blog"
)

// CreateBlog 就是方法重写 --> UnimplementedBlogRpcServer
func (i *impl) CreateBlog(ctx context.Context, in *blog.CreateBlogRequest) (
	*blog.Blog, error) {
	return &blog.Blog{Spec: &blog.CreateBlogRequest{
		Content: "xxxx",
	}}, nil
}

// 定义一个流式请求
// 1. 不断的读取请求 保存到文件中
func (i *impl) UploadBlog(stream blog.BlogRpc_UploadBlogServer) error {
	for {
		data, err := stream.Recv()
		if err != nil {
			// 如果遇到io.EOF表示客户端流被关闭
			if err == io.EOF {
				return nil
			}
			return err
		}
		fmt.Println(data.Meta, string(data.Data))
	}
}

// 定义一个 Server 流式请求
func (i *impl) DownloadBlog(
	in *blog.DownloadBlogRequest,
	stream blog.BlogRpc_DownloadBlogServer) error {
	// 服务端 不但发送请求给客户端
	fmt.Println(in.FileName)

	// 获取到stream client后不断的想服务端发送数据
	f, err := os.Open("README.md")
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	r := bufio.NewReader(f)

	// conn
	for {
		line, _, err := r.ReadLine()
		if err != nil {
			return err
		}

		err = stream.Send(&blog.DownloadBlogResponse{Data: line})
		if err != nil {
			return err
		}
	}
}