How to do bytea in GORM?

Handling the bytea type in GORM primarily involves managing binary data in Go and interacting with bytea type fields in PostgreSQL databases via GORM. Here are some basic steps and examples to help you understand how to implement this:

1. Define the Model

First, define a model in Go that includes a field of type []byte, which maps to the bytea data type in PostgreSQL.

go
package main

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

type Document struct {
    gorm.Model
    Content []byte
}

In this model, the Content field stores binary data, corresponding to the bytea type in PostgreSQL.

2. Database Connection and Model Migration

Next, initialize the database connection and auto-migrate the model to ensure the corresponding table is created in the database.

go
func setupDatabase() *gorm.DB {
    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 {
        panic("Failed to connect to database!")
    }

    db.AutoMigrate(&Document{})

    return db
}

3. Insert Binary Data

You can create a Document instance and store binary data in the Content field.

go
func insertBinaryData(db *gorm.DB, data []byte) {
    document := Document{
        Content: data,
    }
    result := db.Create(&document)
    if result.Error != nil {
        panic("Failed to insert binary data!")
    }
}

4. Retrieve Binary Data

Retrieving binary data from the database is straightforward.

go
func getBinaryData(db *gorm.DB, id uint) []byte {
    var document Document
    result := db.First(&document, id)
    if result.Error != nil {
        panic("Failed to retrieve binary data!")
    }
    return document.Content
}

Example Application

Integrating these functionalities into a simple application:

go
func main() {
    db := setupDatabase()

    // Assume some binary data
    binaryData := []byte{0x00, 0x01, 0x02, 0x03, 0x04}

    // Insert data
    insertBinaryData(db, binaryData)

    // Retrieve and print data
    retrievedData := getBinaryData(db, 1) // Assume ID is 1
    fmt.Println("Retrieved Binary Data:", retrievedData)
}

In this example, we use the []byte type to handle binary data and interact with the bytea type in PostgreSQL via GORM. This approach is simple and efficient, making it ideal for scenarios involving image storage, file handling, or other binary data operations.