How to store ipv4 and ipv6 addresses in postgresql using GORM

在Golang项目中使用GORM与PostgreSQL数据库配合存储IPv4和IPv6地址时，一个有效的方法是使用PostgreSQL内置的 inet 或 cidr 类型。这两种类型都可以有效地存储IP地址，并且自动处理IPv4与IPv6的兼容性问题。下面我将详细介绍如何在项目中实现这一功能。

步骤 1: 定义模型

首先，您需要定义一个Golang结构体，用于映射数据库中的表。假设您有一个名为 network_devices 的表，其中包含设备的IP地址。

go
package models

import (
    "gorm.io/gorm"
)

type NetworkDevice struct {
    gorm.Model
    IPAddress string `gorm:"type:inet"` // 使用 inet 类型存储IP地址
}

在这个例子中，我们为 IPAddress 字段指定了 gorm:"type:inet"，这告诉GORM使用PostgreSQL的 inet 类型来存储该字段。这种类型自动支持IPv4和IPv6。

步骤 2: 迁移数据库

接下来，您需要迁移数据库以创建或更新表结构。可以使用GORM的迁移工具来完成这一步。

go
package main

import (
    "gorm.io/driver/postgres"
    "gorm.io/gorm"
    "log"

    "your_project/models"
)

func main() {
    // 连接到数据库
    dsn := "host=localhost user=gorm password=gorm dbname=gorm port=9920 sslmode=disable TimeZone=Asia/Shanghai"
    db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
    if err != nil {
        log.Fatalf("Failed to connect to database: %v", err)
    }

    // 迁移 schema
    err = db.AutoMigrate(&models.NetworkDevice{})
    if err != nil {
        log.Fatalf("Failed to migrate database: %v", err)
    }
}

这段代码会创建或更新 network_devices 表，使其包含适用于存储IPv4和IPv6地址的 inet 字段。

步骤 3: 插入和查询数据

最后，您可以插入和查询包含IP地址的数据。

go
package main

import (
    "fmt"
    "gorm.io/driver/postgres"
    "gorm.io/gorm"
    "log"
    "your_project/models"
)

func main() {
    // 示例：插入数据
    db, _ := gorm.Open(postgres.Open("your_connection_string"), &gorm.Config{})
    device := models.NetworkDevice{IPAddress: "192.168.1.1"} // IPv4
    result := db.Create(&device)
    if result.Error != nil {
        log.Printf("Error while inserting data: %v", result.Error)
    }

    // 示例：查询数据
    var retrievedDevice models.NetworkDevice
    db.First(&retrievedDevice, "ip_address = ?", "192.168.1.1")
    fmt.Printf("Retrieved Device: %+v\n", retrievedDevice)
}

这样，您就可以在Golang项目中使用GORM和PostgreSQL有效地处理和存储IPv4和IPv6地址了。通过使用PostgreSQL的 inet 类型，您可以确保数据的正确性和查询的效率。