Function runTestCaseGen

client/packmgr/generator.go:21–68

运行测试数据生成

(session *executor.JudgeSession, tCase *structs.TestCase, withAnswer bool)

// 运行测试数据生成
func runTestCaseGen(session *executor.JudgeSession, tCase *structs.TestCase, withAnswer bool) error {
	// 如果是generator脚本
	if tCase.UseGenerator {
		ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
		defer cancel()
		inbytes, err := utils.CallGenerator(ctx, tCase, session.ConfigDir)
		if err != nil {
			return err
		}
		// 写入到文件
		err = ioutil.WriteFile(path.Join(session.ConfigDir, tCase.Input), inbytes, 0664)
		if err != nil {
			return err
		}
		log.Printf("[generator] generate input done!")
	}
	if withAnswer {
		fin, err := os.Open(path.Join(session.ConfigDir, tCase.Input))
		if err != nil {
			return err
		}
		fout, err := os.Create(path.Join(session.ConfigDir, tCase.Output))
		if err != nil {
			return err
		}
		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
		defer cancel()
		rel, err := utils.RunUnixShell(&structs.ShellOptions{
			Context: ctx,
			Name:    session.Commands[0],
			Args:    session.Commands[1:],
			StdWriter: &structs.ShellWriters{
				Input:  fin,
				Output: fout,
				Error:  nil,
			},
		})
		if err != nil {
			return err
		}
		if !rel.Success {
			log.Printf("[generator] run answer code error: %s", rel.Stderr)
			return errors.Errorf("[generator] run answer code error: %s", rel.Stderr)
		}
		log.Printf("[generator] generate answer done!")
	}
	return nil
}

// 运行test cases的数据生成
// caseIndex < 0 表示校验全部