詳解如何使用beego orm在postgres中儲存圖片

2023-09-07 18:00:05

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，向資料庫插入圖片，或者從資料庫讀出圖片的內容了

以上就是詳解如何使用beego orm在postgres中儲存圖片的詳細內容，更多關於beego orm postgres儲存圖片的資料請關注it145.com其它相關文章！

詳解如何使用beego orm在postgres中儲存圖片

目錄

Postgres如何儲存檔案

bytea型別

Large Object

Beego orm如何儲存圖片

熱門文章