88 lines
1.8 KiB
Go
88 lines
1.8 KiB
Go
package mmlclient
|
||
|
||
import (
|
||
"bufio"
|
||
"fmt"
|
||
"io"
|
||
"net"
|
||
"time"
|
||
|
||
"be.ems/lib/core/conf"
|
||
)
|
||
|
||
// 定义MMLClient结构体
|
||
type MMLClient struct {
|
||
awaitTime time.Duration // 等待时间
|
||
conn net.Conn
|
||
reader *bufio.Reader
|
||
size int // 包含字符
|
||
}
|
||
|
||
// 封装NewMMLClient函数，用于创建MMLClient实例
|
||
// 网元UDM的IP地址 "198.51.100.1"
|
||
func NewMMLClient(ip string) (*MMLClient, error) {
|
||
// 创建TCP连接
|
||
portMML := conf.Get("mml.port").(int)
|
||
hostMML := fmt.Sprintf("%s:%d", ip, portMML)
|
||
conn, err := net.Dial("tcp", hostMML)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 进行登录
|
||
usernameMML := conf.Get("mml.user").(string)
|
||
passwordMML := conf.Get("mml.password").(string)
|
||
fmt.Fprintln(conn, usernameMML)
|
||
fmt.Fprintln(conn, passwordMML)
|
||
|
||
// 发送后等待
|
||
awaitTime := time.Duration(300) * time.Millisecond
|
||
time.Sleep(awaitTime)
|
||
|
||
// 读取内容
|
||
buf := make([]byte, 1024*1024*1)
|
||
n, err := conn.Read(buf)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 创建MMLClient实例
|
||
client := &MMLClient{
|
||
conn: conn,
|
||
reader: bufio.NewReader(conn),
|
||
awaitTime: awaitTime,
|
||
size: n,
|
||
}
|
||
|
||
return client, nil
|
||
}
|
||
|
||
// 封装Send函数，用于向TCP连接发送数据
|
||
func (c *MMLClient) Send(msg string) error {
|
||
_, err := fmt.Fprintln(c.conn, msg)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
time.Sleep(c.awaitTime)
|
||
return nil
|
||
}
|
||
|
||
// 封装Receive函数，用于从TCP连接中接收数据
|
||
func (c *MMLClient) Receive() (string, error) {
|
||
buf := make([]byte, 1024*1024*1)
|
||
n, err := c.reader.Read(buf)
|
||
if err != nil {
|
||
if err == io.EOF {
|
||
return "", fmt.Errorf("server closed the connection")
|
||
}
|
||
return "", err
|
||
}
|
||
c.size += n
|
||
return string(buf[0:n]), nil
|
||
}
|
||
|
||
// 封装Close函数，用于关闭TCP连接
|
||
func (c *MMLClient) Close() error {
|
||
return c.conn.Close()
|
||
}
|