You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
690 B
40 lines
690 B
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 |
|
}
|
|
|