详解如何使用beego orm在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)"`

									}