package deliver

import (
	"google.golang.org/grpc"
	"sonet/api/gen/postal"
	"sonet/pkg/utils/logger"
)

type PostalNode struct {
	addr   string
	conn   *grpc.ClientConn
	client postal.PostalClient
}

func NewPostalNode(addr string, conn *grpc.ClientConn) *PostalNode {
	return &PostalNode{
		addr:   addr,
		conn:   conn,
		client: postal.NewPostalClient(conn),
	}
}

func (p *PostalNode) Close() {
	err := p.conn.Close()
	if err != nil {
		logger.Error("postal close error: ", err)
	}
}

func (p *PostalNode) GetAddr() string {
	return p.addr
}

func (p *PostalNode) GetConn() *grpc.ClientConn {
	return p.conn
}

func (p *PostalNode) GetClient() postal.PostalClient {
	return p.client
}