使用递增函数填充结构图

时间:2018-01-06 11:29:19

标签: go

我正在尝试使用字符串键和指向struct值的指针填充地图。具体而言,在填充一个字段时出现问题,需要通过调用函数来增加另一个字段的范围。 Problem snippet is here

package main

import (
  "fmt"
  "net"
)

type bgp struct {
  tag net.IP
}

type nbrStruct struct {
  bgp
}

var mIPStr = "10.10.10.0"

func main() {
  m := make(map[string]*nbrStruct)
  s := []string{"string1", "string2"}
  mIP := net.ParseIP(mIPStr)
  mIP = mIP.To4()

  for _, v := range s {
    m[v] = &nbrStruct{
       bgp: bgp{
         tag: mIP,
       },
    }
    fmt.Println(mIP)
    mIP, _ = NextIP(mIP, 4)
  }
  for key, value := range m {
    fmt.Printf("key: %v ---> tag: %v\n", key, value.bgp.tag)
  }
}

func NextIP(ip net.IP, addIP int) (net.IP, error) {
  var err error
  if ip = ip.To4(); ip != nil {
        // Case IPv4
    ip[3] = ip[3] + uint8(addIP)
    if ip[3] > 255 {
    ip[3] = ip[3] - 255
    ip[2]++
  }
    if ip[2] > 255 {
        ip[2] = ip[2] - 255
        ip[1]++
    }
    if ip[1] > 255 {
        err = fmt.Errorf("NextIP out of space in top IPv4 Octet %v", ip)
        return nil, err
    }
    // } else if ip = ip.To16(); ip != nil {
    //  // Case IPv6
} else {
    err = fmt.Errorf("NextIP cannot understand IP: %v", ip)
    return nil, err
}
return ip, err
}

我遇到的问题是第31行

mIP, _ = NextIP(mIP, 4)

我希望这两个值是:

m["string1"].bgp.tag == 10.10.10.0
m["string2"].bgp.tag == 10.10.10.4

...函数NextIP正确返回。但是在阅读地图时,之前的m [" string1"]。bgp.tag值似乎被新的覆盖?他们都是10.10.10.8;对此非常困惑。我做错了什么?

1 个答案:

答案 0 :(得分:2)

net.IP是一个字节切片。

你基本上是经常修改mIP并存储它。但是由于slices work的方式,你实际上存储了一个指向底层数据的指针。

问题的快速解决方法是存储mIP的副本。例如:

for _, v := range s {
    tmpIP := make([]byte, len(mIP))
    copy(tmpIP, mIP)
    m[v] = &nbrStruct{
        bgp: bgp{
            tag: tmpIP,
        },
    }

    mIP, _ = NextIP(mIP, 4)
}

这会产生正确的输出:

key: string1 ---> tag: 10.10.10.0
key: string2 ---> tag: 10.10.10.4

我强烈建议您更改NextIP以返回新IP,而不是输入的修改版本。