memos/store/resource.go

205 lines
4.0 KiB
Go
Raw Normal View History

package store
2022-02-03 10:32:03 +03:00
import (
2022-05-21 19:59:22 +03:00
"database/sql"
2022-02-03 10:32:03 +03:00
"fmt"
"strings"
2022-06-27 17:09:06 +03:00
"github.com/usememos/memos/api"
"github.com/usememos/memos/common"
2022-02-03 10:32:03 +03:00
)
2022-05-19 13:32:04 +03:00
// resourceRaw is the store model for an Resource.
// Fields have exactly the same meanings as Resource.
type resourceRaw struct {
ID int
// Standard fields
CreatorID int
CreatedTs int64
UpdatedTs int64
// Domain specific fields
Filename string
Blob []byte
Type string
Size int64
}
func (raw *resourceRaw) toResource() *api.Resource {
return &api.Resource{
ID: raw.ID,
// Standard fields
CreatorID: raw.CreatorID,
CreatedTs: raw.CreatedTs,
UpdatedTs: raw.UpdatedTs,
// Domain specific fields
Filename: raw.Filename,
Blob: raw.Blob,
Type: raw.Type,
Size: raw.Size,
}
}
2022-05-16 02:37:23 +03:00
func (s *Store) CreateResource(create *api.ResourceCreate) (*api.Resource, error) {
2022-05-19 13:32:04 +03:00
resourceRaw, err := createResource(s.db, create)
2022-02-03 10:32:03 +03:00
if err != nil {
return nil, err
}
2022-05-19 13:32:04 +03:00
resource := resourceRaw.toResource()
2022-02-03 10:32:03 +03:00
return resource, nil
}
2022-05-16 02:37:23 +03:00
func (s *Store) FindResourceList(find *api.ResourceFind) ([]*api.Resource, error) {
2022-05-19 13:32:04 +03:00
resourceRawList, err := findResourceList(s.db, find)
2022-02-03 10:32:03 +03:00
if err != nil {
return nil, err
}
2022-05-19 13:32:04 +03:00
resourceList := []*api.Resource{}
for _, raw := range resourceRawList {
resourceList = append(resourceList, raw.toResource())
}
return resourceList, nil
}
2022-05-16 02:37:23 +03:00
func (s *Store) FindResource(find *api.ResourceFind) (*api.Resource, error) {
2022-02-04 13:54:24 +03:00
list, err := findResourceList(s.db, find)
if err != nil {
return nil, err
}
if len(list) == 0 {
return nil, &common.Error{Code: common.NotFound, Err: fmt.Errorf("not found")}
}
2022-05-19 13:32:04 +03:00
resource := list[0].toResource()
return resource, nil
2022-02-04 13:54:24 +03:00
}
2022-05-16 02:37:23 +03:00
func (s *Store) DeleteResource(delete *api.ResourceDelete) error {
2022-02-03 10:32:03 +03:00
err := deleteResource(s.db, delete)
if err != nil {
return err
}
return nil
}
2022-05-21 19:59:22 +03:00
func createResource(db *sql.DB, create *api.ResourceCreate) (*resourceRaw, error) {
row, err := db.Query(`
2022-02-03 10:32:03 +03:00
INSERT INTO resource (
filename,
blob,
type,
size,
creator_id
)
VALUES (?, ?, ?, ?, ?)
RETURNING id, filename, blob, type, size, created_ts, updated_ts
`,
create.Filename,
create.Blob,
create.Type,
create.Size,
2022-05-02 21:05:43 +03:00
create.CreatorID,
2022-02-03 10:32:03 +03:00
)
if err != nil {
return nil, FormatError(err)
}
defer row.Close()
2022-05-19 13:32:04 +03:00
row.Next()
var resourceRaw resourceRaw
2022-02-03 10:32:03 +03:00
if err := row.Scan(
2022-05-19 13:32:04 +03:00
&resourceRaw.ID,
&resourceRaw.Filename,
&resourceRaw.Blob,
&resourceRaw.Type,
&resourceRaw.Size,
&resourceRaw.CreatedTs,
&resourceRaw.UpdatedTs,
2022-02-03 10:32:03 +03:00
); err != nil {
return nil, FormatError(err)
}
2022-05-19 13:32:04 +03:00
return &resourceRaw, nil
2022-02-03 10:32:03 +03:00
}
2022-05-21 19:59:22 +03:00
func findResourceList(db *sql.DB, find *api.ResourceFind) ([]*resourceRaw, error) {
2022-02-03 10:32:03 +03:00
where, args := []string{"1 = 1"}, []interface{}{}
2022-02-06 06:26:53 +03:00
2022-05-02 21:05:43 +03:00
if v := find.ID; v != nil {
2022-02-03 10:32:03 +03:00
where, args = append(where, "id = ?"), append(args, *v)
}
2022-05-02 21:05:43 +03:00
if v := find.CreatorID; v != nil {
2022-02-03 10:32:03 +03:00
where, args = append(where, "creator_id = ?"), append(args, *v)
}
if v := find.Filename; v != nil {
where, args = append(where, "filename = ?"), append(args, *v)
}
2022-05-21 19:59:22 +03:00
rows, err := db.Query(`
2022-02-03 10:32:03 +03:00
SELECT
id,
filename,
blob,
type,
size,
created_ts,
updated_ts
FROM resource
2022-05-22 07:35:57 +03:00
WHERE `+strings.Join(where, " AND ")+`
ORDER BY created_ts DESC`,
2022-02-03 10:32:03 +03:00
args...,
)
if err != nil {
return nil, FormatError(err)
}
defer rows.Close()
2022-05-19 13:32:04 +03:00
resourceRawList := make([]*resourceRaw, 0)
for rows.Next() {
2022-05-19 13:32:04 +03:00
var resourceRaw resourceRaw
2022-02-03 10:32:03 +03:00
if err := rows.Scan(
2022-05-19 13:32:04 +03:00
&resourceRaw.ID,
&resourceRaw.Filename,
&resourceRaw.Blob,
&resourceRaw.Type,
&resourceRaw.Size,
&resourceRaw.CreatedTs,
&resourceRaw.UpdatedTs,
2022-02-03 10:32:03 +03:00
); err != nil {
return nil, FormatError(err)
}
2022-05-19 13:32:04 +03:00
resourceRawList = append(resourceRawList, &resourceRaw)
}
if err := rows.Err(); err != nil {
2022-02-03 10:32:03 +03:00
return nil, FormatError(err)
}
2022-05-19 13:32:04 +03:00
return resourceRawList, nil
}
2022-05-21 19:59:22 +03:00
func deleteResource(db *sql.DB, delete *api.ResourceDelete) error {
result, err := db.Exec(`DELETE FROM resource WHERE id = ?`, delete.ID)
2022-02-03 10:32:03 +03:00
if err != nil {
return FormatError(err)
}
rows, _ := result.RowsAffected()
if rows == 0 {
2022-05-02 21:05:43 +03:00
return &common.Error{Code: common.NotFound, Err: fmt.Errorf("resource ID not found: %d", delete.ID)}
2022-02-03 10:32:03 +03:00
}
2022-02-03 10:32:03 +03:00
return nil
}