本文整理汇总了Golang中github.com/diffeo/go-coordinate/jobserver.JobServer.GetWorkUnits方法的典型用法代码示例。如果您正苦于以下问题:Golang JobServer.GetWorkUnits方法的具体用法?Golang JobServer.GetWorkUnits怎么用?Golang JobServer.GetWorkUnits使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类github.com/diffeo/go-coordinate/jobserver.JobServer
的用法示例。
在下文中一共展示了JobServer.GetWorkUnits方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Golang代码示例。
示例1: listWorkUnits
// listWorkUnits calls GetWorkUnits (as the similarly-named Python
// function does) and validates that the response matches an expected
// set of work units.
func listWorkUnits(t *testing.T, j *jobserver.JobServer, workSpecName string, options map[string]interface{}, expected map[string]map[string]interface{}) {
gwu, msg, err := j.GetWorkUnits(workSpecName, options)
if !assert.NoError(t, err) {
return
}
assert.Empty(t, msg)
missing := make(map[string]struct{})
for name := range expected {
missing[name] = struct{}{}
}
for _, item := range gwu {
if !assert.IsType(t, cborrpc.PythonTuple{}, item) {
continue
}
tuple := item.(cborrpc.PythonTuple)
if !assert.Len(t, tuple.Items, 2) {
continue
}
if !assert.IsType(t, []byte{}, tuple.Items[0]) {
continue
}
name := string(tuple.Items[0].([]byte))
assert.IsType(t, map[string]interface{}{}, tuple.Items[1])
if assert.Contains(t, expected, name, "unexpected work unit") {
assert.Equal(t, expected[name], tuple.Items[1])
}
assert.Contains(t, missing, name, "duplicate work unit")
delete(missing, name)
}
}
示例2: getOneWorkUnit
// getOneWorkUnit calls GetWorkUnits for a single specific work unit,
// checks the results, and returns its data dictionary (or nil if absent).
func getOneWorkUnit(t *testing.T, j *jobserver.JobServer, workSpecName, workUnitKey string) map[string]interface{} {
list, msg, err := j.GetWorkUnits(workSpecName, map[string]interface{}{"work_unit_keys": []interface{}{workUnitKey}})
if !assert.NoError(t, err) {
return nil
}
assert.Empty(t, msg)
if !assert.Len(t, list, 1) {
return nil
}
if !assert.IsType(t, cborrpc.PythonTuple{}, list[0]) {
return nil
}
tuple := list[0].(cborrpc.PythonTuple)
if !assert.Len(t, tuple.Items, 2) {
return nil
}
assert.Equal(t, []byte(workUnitKey), tuple.Items[0])
if assert.IsType(t, map[string]interface{}{}, tuple.Items[1]) {
return tuple.Items[1].(map[string]interface{})
}
return nil
}