121 lines
2.8 KiB
Go
121 lines
2.8 KiB
Go
package file
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"net"
|
||
"net/http"
|
||
"time"
|
||
|
||
"ems.agt/lib/dborm"
|
||
"ems.agt/lib/global"
|
||
"ems.agt/lib/log"
|
||
"ems.agt/lib/services"
|
||
"ems.agt/restagent/config"
|
||
"github.com/gorilla/mux"
|
||
)
|
||
|
||
const (
|
||
//经过测试，linux下，延时需要大于100ms
|
||
TIME_DELAY_AFTER_WRITE = 200
|
||
)
|
||
|
||
type Response struct {
|
||
Data []string `json:"data"`
|
||
}
|
||
|
||
type MMLRequest struct {
|
||
MML []string `json:"mml"`
|
||
}
|
||
|
||
func GetFile(w http.ResponseWriter, r *http.Request) {
|
||
log.Debug("PostMMLToNF processing... ")
|
||
|
||
vars := mux.Vars(r)
|
||
neType := vars["elementTypeValue"]
|
||
params := r.URL.Query()
|
||
neId := params["ne_id"]
|
||
log.Debug("neType:", neType, "neId", neId)
|
||
|
||
neInfo := new(dborm.NeInfo)
|
||
var err error
|
||
if len(neId) == 0 {
|
||
log.Error("ne_id NOT FOUND")
|
||
services.ResponseBadRequest400WrongParamValue(w)
|
||
return
|
||
}
|
||
neInfo, err = dborm.XormGetNeInfo(neType, neId[0])
|
||
if err != nil {
|
||
log.Error("dborm.XormGetNeInfo is failed:", err)
|
||
services.ResponseInternalServerError500DatabaseOperationFailed(w)
|
||
return
|
||
}
|
||
|
||
var buf [8192]byte
|
||
var n int
|
||
var mmlResult []string
|
||
|
||
if neInfo != nil {
|
||
hostMML := fmt.Sprintf("%s:%d", neInfo.Ip, config.GetYamlConfig().MML.Port)
|
||
conn, err := net.Dial("tcp", hostMML)
|
||
if err != nil {
|
||
errMsg := fmt.Sprintf("Failed to dial %s: %v", hostMML, err)
|
||
log.Error(errMsg)
|
||
mmlResult = append(mmlResult, errMsg)
|
||
response := Response{mmlResult}
|
||
services.ResponseWithJson(w, http.StatusOK, response)
|
||
return
|
||
}
|
||
|
||
loginStr := fmt.Sprintf("%s\n%s\n", config.GetYamlConfig().MML.User, config.GetYamlConfig().MML.Password)
|
||
n, err = conn.Write([]byte(loginStr))
|
||
if err != nil {
|
||
log.Errorf("Error: %s", err.Error())
|
||
return
|
||
}
|
||
|
||
time.Sleep(time.Millisecond * TIME_DELAY_AFTER_WRITE)
|
||
|
||
n, err = conn.Read(buf[0:])
|
||
if err != nil {
|
||
log.Errorf("Error: %s", err.Error())
|
||
return
|
||
}
|
||
log.Debug(string(buf[0:n]))
|
||
|
||
body, err := io.ReadAll(io.LimitReader(r.Body, global.RequestBodyMaxLen))
|
||
if err != nil {
|
||
log.Error("io.ReadAll is failed:", err)
|
||
services.ResponseNotFound404UriNotExist(w, r)
|
||
return
|
||
}
|
||
log.Debug("Body:", string(body))
|
||
|
||
mmlRequest := new(MMLRequest)
|
||
_ = json.Unmarshal(body, mmlRequest)
|
||
|
||
for _, mml := range mmlRequest.MML {
|
||
mmlCommand := fmt.Sprintf("%s\n", mml)
|
||
log.Debug("mml command:", mmlCommand)
|
||
n, err = conn.Write([]byte(mmlCommand))
|
||
if err != nil {
|
||
log.Errorf("Error: %s", err.Error())
|
||
return
|
||
}
|
||
time.Sleep(time.Millisecond * TIME_DELAY_AFTER_WRITE)
|
||
|
||
n, err = conn.Read(buf[0:])
|
||
if err != nil {
|
||
log.Errorf("Error: %s", err.Error())
|
||
return
|
||
}
|
||
log.Debug(string(buf[0 : n-len(neType)-2]))
|
||
mmlResult = append(mmlResult, string(buf[0:n-len(neType)-2]))
|
||
}
|
||
}
|
||
|
||
response := Response{mmlResult}
|
||
services.ResponseWithJson(w, http.StatusOK, response)
|
||
}
|