首页 > Golang > Go教程

详解如何使用beegoorm在postgres中存储图片

来源：脚本之家

时间：2023-05-12 16:11:17 356浏览收藏

golang学习网今天将给大家带来《详解如何使用beegoorm在postgres中存储图片》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到BeegoORM、postgres、存储图片等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

Postgres如何存储文件

postgres提供了两种不同的方式存储二进制，要么是使用bytea类型直接存储二进制，要么就是使用postgres的LargeObject功能；决定使用哪中方式更加适合你，就需要了解这两种存储方式有哪些限制

bytea类型

bytea类型在单列中虽然可以支持1GB的容量，但是还是不建议使用bytea去储存比较大的对象，因为它会占用大量的内存

下面通过一个例子来说明，假如现在要在一个表中存储图片名和该图片的数据，创建表如下：

CREATE TABLE images (imgname text, img bytea);

在表中插入一张图片：

File file = new File("myimage.gif");
FileInputStream fis = new FileInputStream(file);
PreparedStatement ps = conn.prepareStatement("INSERT INTO images VALUES (?, ?)");
ps.setString(1, file.getName());
ps.setBinaryStream(2, fis, file.length());
ps.executeUpdate();
ps.close();
fis.close();

上面的setBinaryStream就会将图片内容设置到img字段上面，也可以使用setBytes()直接设置图片的内容

接下来，从表中取出图片，代码如下：

PreparedStatement ps = con.prepareStatement("SELECT img FROM images WHERE imgname = ?");
ps.setString(1, "myimage.gif");
ResultSet rs = ps.executeQuery();
if (rs != null) {
    while (rs.next()) {
        byte[] imgBytes = rs.getBytes(1);
        // use the data in some way here
    }
    rs.close();
}
ps.close();

Large Object

Large Object就可以存储大文件，存储的方式是在单独的一张表中存储大文件，然后通过oid在当前表中进行引用；下面通过一个例子来解释：

CREATE TABLE imageslo (imgname text, imgoid oid);

首先是创建一张表，该表中第二个字段类型为oid，之后就是通过该字段引用大文件对象；下面我们在表中插入一张图片：

// All LargeObject API calls must be within a transaction block
conn.setAutoCommit(false);
// Get the Large Object Manager to perform operations with
LargeObjectManager lobj = ((org.postgresql.PGConnection)conn).getLargeObjectAPI();
// Create a new large object
int oid = lobj.create(LargeObjectManager.READ | LargeObjectManager.WRITE);
// Open the large object for writing
LargeObject obj = lobj.open(oid, LargeObjectManager.WRITE);
// Now open the file
File file = new File("myimage.gif");
FileInputStream fis = new FileInputStream(file);
// Copy the data from the file to the large object
byte buf[] = new byte[2048];
int s, tl = 0;
while ((s = fis.read(buf, 0, 2048)) > 0) {
    obj.write(buf, 0, s);
    tl += s;
}
// Close the large object
obj.close();
// Now insert the row into imageslo
PreparedStatement ps = conn.prepareStatement("INSERT INTO imageslo VALUES (?, ?)");
ps.setString(1, file.getName());
ps.setInt(2, oid);
ps.executeUpdate();
ps.close();
fis.close();

在代码中需要使用lobp.open打开一个大文件，然后将图片的内容写入这个对象当中；下面从大文件对象中读取这个图片：

// All LargeObject API calls must be within a transaction block
conn.setAutoCommit(false);
// Get the Large Object Manager to perform operations with
LargeObjectManager lobj = ((org.postgresql.PGConnection)conn).getLargeObjectAPI();
PreparedStatement ps = con.prepareStatement("SELECT imgoid FROM imageslo WHERE imgname = ?");
ps.setString(1, "myimage.gif");
ResultSet rs = ps.executeQuery();
if (rs != null) {
    while (rs.next()) {
        // Open the large object for reading
        int oid = rs.getInt(1);
        LargeObject obj = lobj.open(oid, LargeObjectManager.READ);
        // Read the data
        byte buf[] = new byte[obj.size()];
        obj.read(buf, 0, obj.size());
        // Do something with the data read here
        // Close the object
        obj.close();
    }
    rs.close();
}
ps.close();

需要注意的是，对于Large Object的操作都需要放在一个事务（Transaction）当中；如果要删除大文件所在行，在删除这行之后，还需要再执行删除大文件的操作

注：使用Large Object会有安全问题，连接到数据库的用户，即便没有包含大对象所在列的权限，也可以操作这个大对象

Beego orm如何存储图片

看完上面的postgres对于图片的存储，再来看下如何使用beego orm存储一张图片；在beego orm中支持了go的所有基础类型，但是不支持slice；所以，不能直接将[]byte映射到bytea字段上面

好在beego orm提供了一个Fielder接口，可以自定义类型，接口定义如下：

// Fielder define field info
type Fielder interface {
    String() string
    FieldType() int
    SetRaw(interface{}) error
    RawValue() interface{}
}

所以，现在就需要定义一个字节数组的类型，然后实现这些接口就好了，代码如下：

type ByteArrayField []byte
// set value 
func (e *ByteArrayField) SetRaw(value interface{}) error {
    if value == nil {
        return nil
    }
    switch d := value.(type) {
    case []byte:
        *e = d
    case string:
        *e = []byte(d)
    default:
        return fmt.Errorf("[ByteArrayField] unsupported type")
    }
    return nil
}
func (e *ByteArrayField) RawValue() interface{} {
    return *e
}
// specified type
func (f *ByteArrayField) FieldType() int {
    return orm.TypeTextField
}
func (f *ByteArrayField) String() string {
    return string(*f)
}

然后，我们就可以在struct中进行映射了，如下：

type ImageModel struct{
    ImageName string `orm:"column(image_name)"`
    ImageData ByteArrayField `orm:"column(image_data);type(bytea)"`
}

这样就可以使用orm的接口操作imageModel，向数据库插入图片，或者从数据库读出图片的内容了

好了，本文到此结束，带大家了解了《详解如何使用beegoorm在postgres中存储图片》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！

golang BeegoORM postgres 存储图片

声明：本文转载于：脚本之家如有侵犯，请联系study_golang@163.com删除