源碼見(jiàn)：https://github.com/lovercode/GO_OJ.git，demo見(jiàn)：https://codelover.me/run.html
Compiler的實(shí)現(xiàn)
compiler的實(shí)現(xiàn)比較簡(jiǎn)單，直接調(diào)用 compile.sh編譯代碼就行畸悬，編譯期的安全涤躲，可以在腳本中限制羡疗，同時(shí)可以限制編譯的運(yùn)行時(shí)間咆疗。
OJ技術(shù)思考：評(píng)測(cè)安全
var consumer *mq.MqConsumer
var producer *mq.MqProducer
func init() {
consumer = mq.NewMqConsumer(conf.GlobalConfig.CompilerConsumer)
producer = mq.NewMqProducer(conf.GlobalConfig.RunProducer)
}
func main() {
for data := range consumer.Ch {
go compile(data)
}
ch := make(chan int)
<-ch
}
func compile(data []byte) {
pro := &problem.Problem{}
e := proto.Unmarshal(data, pro)
if e != nil {
log.Println("解析錯(cuò)誤")
return
}
topic, has := conf.GlobalConfig.RunConfig[fmt.Sprintf("%d", pro.GetType())]
if !has {
log.Println("沒(méi)有找到相關(guān)配置")
return
}
cmd := exec.Command("/bin/sh", "./compile.sh", pro.GetUuid())
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
cmd.Dir = conf.GlobalConfig.WorkDirPre + "/" + pro.GetUuid()
err := cmd.Run()
if err != nil {
bytes, err := ioutil.ReadFile(conf.GlobalConfig.WorkDirPre + "/" + pro.GetUuid() + "/compiler_err.msg")
if err != nil {
pro.ErrMsg = "獲取編譯信息失敗"
} else {
pro.ErrMsg = string(bytes)
}
pro.Status = problem.Status_Compile_Error
notifyOther(pro)
log.Println("編譯錯(cuò)誤", err)
return
}
pro.Status = problem.Status_Running
info, e := proto.Marshal(pro)
if e != nil {
panic(e)
}
log.Printf("%s:compile cost:%dms\n", pro.GetUuid(), time.Now().UnixNano()/1000000-pro.GetCreateTime())
producer.MqPublish(topic.RunTopic, string(info))
}
func notifyOther(pro *problem.Problem) {
info, e := proto.Marshal(pro)
if e != nil {
panic(e)
}
producer.MqPublish(conf.GlobalConfig.RunEndTopic, string(info))
}
編譯完就可以通知runner了
