Fix #202
This commit is contained in:
parent
fef4fffce0
commit
18d5d1c09b
|
@ -22,12 +22,23 @@ import (
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/b3log/wide/conf"
|
"github.com/b3log/wide/conf"
|
||||||
"github.com/b3log/wide/session"
|
"github.com/b3log/wide/session"
|
||||||
"github.com/b3log/wide/util"
|
"github.com/b3log/wide/util"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
outputBufMax = 128 // 128 string(rune)
|
||||||
|
outputTimeout = 100 // 100ms
|
||||||
|
)
|
||||||
|
|
||||||
|
type outputBuf struct {
|
||||||
|
content string
|
||||||
|
millisecond int64
|
||||||
|
}
|
||||||
|
|
||||||
// RunHandler handles request of executing a binary file.
|
// RunHandler handles request of executing a binary file.
|
||||||
func RunHandler(w http.ResponseWriter, r *http.Request) {
|
func RunHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
data := map[string]interface{}{"succ": true}
|
data := map[string]interface{}{"succ": true}
|
||||||
|
@ -122,18 +133,21 @@ func RunHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
}
|
}
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
buf := outputBuf{}
|
||||||
r, _, err := outReader.ReadRune()
|
|
||||||
|
|
||||||
if nil == session.OutputWS[sid] {
|
for {
|
||||||
|
wsChannel := session.OutputWS[sid]
|
||||||
|
if nil == wsChannel {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
wsChannel := session.OutputWS[sid]
|
r, _, err := outReader.ReadRune()
|
||||||
|
|
||||||
buf := string(r)
|
oneRuneStr := string(r)
|
||||||
buf = strings.Replace(buf, "<", "<", -1)
|
oneRuneStr = strings.Replace(oneRuneStr, "<", "<", -1)
|
||||||
buf = strings.Replace(buf, ">", ">", -1)
|
oneRuneStr = strings.Replace(oneRuneStr, ">", ">", -1)
|
||||||
|
|
||||||
|
buf.content += oneRuneStr
|
||||||
|
|
||||||
if nil != err {
|
if nil != err {
|
||||||
// remove the exited process from user process set
|
// remove the exited process from user process set
|
||||||
|
@ -141,31 +155,38 @@ func RunHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
|
|
||||||
logger.Tracef("User [%s, %s] 's running [id=%d, file=%s] has done [stdout %v], ", wSession.Username, sid, runningId, filePath, err)
|
logger.Tracef("User [%s, %s] 's running [id=%d, file=%s] has done [stdout %v], ", wSession.Username, sid, runningId, filePath, err)
|
||||||
|
|
||||||
if nil != wsChannel {
|
channelRet["cmd"] = "run-done"
|
||||||
channelRet["cmd"] = "run-done"
|
channelRet["output"] = buf.content
|
||||||
channelRet["output"] = buf
|
err := wsChannel.WriteJSON(&channelRet)
|
||||||
err := wsChannel.WriteJSON(&channelRet)
|
if nil != err {
|
||||||
if nil != err {
|
logger.Error(err)
|
||||||
logger.Error(err)
|
break
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
wsChannel.Refresh()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
wsChannel.Refresh()
|
||||||
|
|
||||||
break
|
break
|
||||||
} else {
|
}
|
||||||
if nil != wsChannel {
|
|
||||||
channelRet["cmd"] = "run"
|
|
||||||
channelRet["output"] = buf
|
|
||||||
err := wsChannel.WriteJSON(&channelRet)
|
|
||||||
if nil != err {
|
|
||||||
logger.Error(err)
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
wsChannel.Refresh()
|
now := time.Now().UnixNano() / int64(time.Millisecond)
|
||||||
|
|
||||||
|
if 0 == buf.millisecond {
|
||||||
|
buf.millisecond = now
|
||||||
|
}
|
||||||
|
|
||||||
|
if now-outputTimeout >= buf.millisecond || len(buf.content) > outputBufMax || oneRuneStr == "\n" {
|
||||||
|
channelRet["cmd"] = "run"
|
||||||
|
channelRet["output"] = buf.content
|
||||||
|
|
||||||
|
buf = outputBuf{} // a new buffer
|
||||||
|
|
||||||
|
err = wsChannel.WriteJSON(&channelRet)
|
||||||
|
if nil != err {
|
||||||
|
logger.Error(err)
|
||||||
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
|
wsChannel.Refresh()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
@ -173,12 +194,11 @@ func RunHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
for {
|
for {
|
||||||
r, _, err := errReader.ReadRune()
|
r, _, err := errReader.ReadRune()
|
||||||
|
|
||||||
if nil != err || nil == session.OutputWS[sid] {
|
wsChannel := session.OutputWS[sid]
|
||||||
|
if nil != err || nil == wsChannel {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
wsChannel := session.OutputWS[sid]
|
|
||||||
|
|
||||||
buf := string(r)
|
buf := string(r)
|
||||||
buf = strings.Replace(buf, "<", "<", -1)
|
buf = strings.Replace(buf, "<", "<", -1)
|
||||||
buf = strings.Replace(buf, ">", ">", -1)
|
buf = strings.Replace(buf, ">", ">", -1)
|
||||||
|
|
|
@ -418,6 +418,8 @@ var wide = {
|
||||||
|
|
||||||
break;
|
break;
|
||||||
case 'run-done':
|
case 'run-done':
|
||||||
|
bottomGroup.fillOutput($('.bottom-window-group .output > div').html().replace(/<\/pre>$/g, data.output + '</pre>'));
|
||||||
|
|
||||||
wide.curProcessId = undefined;
|
wide.curProcessId = undefined;
|
||||||
$("#buildRun").removeClass("ico-stop")
|
$("#buildRun").removeClass("ico-stop")
|
||||||
.addClass("ico-buildrun").attr("title", config.label.build_n_run);
|
.addClass("ico-buildrun").attr("title", config.label.build_n_run);
|
||||||
|
|
Loading…
Reference in New Issue