问题
最近使用golang写一个分布式任务系统,需要将过程中的变量存储到数据库,以用于任务执行过程中,进程挂了,调度到另一个实例上去, 过程中的变量需要被存储。该变量的获取存在两种场景:直接从内存中获取,任务被重新调度,需要从数据库里获取,在内存中使用interface获取发现获取不到的现象,具体如下面示例代码的GetB,解决办法是内存通过type实现,或者reflect实现,如果存储到数据库的使用json marshal unmarshal即可
示例代码
func main() {
data := new(BT)
GetB(data)
fmt.Println("GetB ", data)
GetB1(data)
fmt.Println("GetB1 ", data)
data = new(BT)
GetB2(data)
fmt.Println("GetB2 ", data)
}
var b = &BT{
A: 1,
}
type BT struct {
A int
}
// 直接传入指针发现,返回的是空
func GetB(r interface{}) {
r = b
}
// 通过marshal和unmarshal获取发现可以
func GetB1(r interface{}) {
tmp, err := json.Marshal(b)
if err != nil {
fmt.Println("marshal failed, ", err.Error())
return
}
if json.Unmarshal(tmp, r) != nil {
fmt.Println("unmarshal failed, ", err.Error())
return
}
}
// 通过reflect的set value发现可以
func GetB2(r interface{}) {
rv := reflect.ValueOf(r)
bv := reflect.ValueOf(b)
if rv.Kind() == bv.Kind() {
rv.Elem().Set(bv.Elem())
}
}