parent
cfb633b3b5
commit
3088c9d76c
@ -0,0 +1,86 @@ |
||||
package post |
||||
|
||||
import ( |
||||
"bytes" |
||||
"database/sql" |
||||
"errors" |
||||
"fmt" |
||||
"io" |
||||
) |
||||
|
||||
var ( |
||||
// ErrAssetNotFound is used to indicate an Asset could not be found in the
|
||||
// AssetStore.
|
||||
ErrAssetNotFound = errors.New("asset not found") |
||||
) |
||||
|
||||
// AssetStore implements the storage and retrieval of binary assets, which are
|
||||
// intended to be used by posts (e.g. images).
|
||||
type AssetStore interface { |
||||
|
||||
// Set sets the id to the contents of the given io.Reader.
|
||||
Set(id string, from io.Reader) error |
||||
|
||||
// Get writes the id's body to the given io.Writer, or returns
|
||||
// ErrAssetNotFound.
|
||||
Get(id string, into io.Writer) error |
||||
|
||||
// Delete's the body stored for the id, if any.
|
||||
Delete(id string) error |
||||
} |
||||
|
||||
type assetStore struct { |
||||
db *sql.DB |
||||
} |
||||
|
||||
// NewAssetStore initializes a new AssetStore using an existing SQLDB.
|
||||
func NewAssetStore(db *SQLDB) AssetStore { |
||||
return &assetStore{ |
||||
db: db.db, |
||||
} |
||||
} |
||||
|
||||
func (s *assetStore) Set(id string, from io.Reader) error { |
||||
|
||||
body, err := io.ReadAll(from) |
||||
if err != nil { |
||||
return fmt.Errorf("reading body fully into memory: %w", err) |
||||
} |
||||
|
||||
_, err = s.db.Exec( |
||||
`INSERT INTO assets (id, body) |
||||
VALUES (?, ?) |
||||
ON CONFLICT (id) DO UPDATE SET body=excluded.body`, |
||||
id, body, |
||||
) |
||||
|
||||
if err != nil { |
||||
return fmt.Errorf("inserting into assets: %w", err) |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
func (s *assetStore) Get(id string, into io.Writer) error { |
||||
|
||||
var body []byte |
||||
|
||||
err := s.db.QueryRow(`SELECT body FROM assets WHERE id = ?`, id).Scan(&body) |
||||
|
||||
if errors.Is(err, sql.ErrNoRows) { |
||||
return ErrAssetNotFound |
||||
} else if err != nil { |
||||
return fmt.Errorf("selecting from assets: %w", err) |
||||
} |
||||
|
||||
if _, err := io.Copy(into, bytes.NewReader(body)); err != nil { |
||||
return fmt.Errorf("writing body to io.Writer: %w", err) |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
func (s *assetStore) Delete(id string) error { |
||||
_, err := s.db.Exec(`DELETE FROM assets WHERE id = ?`, id) |
||||
return err |
||||
} |
@ -0,0 +1,67 @@ |
||||
package post |
||||
|
||||
import ( |
||||
"bytes" |
||||
"io" |
||||
"testing" |
||||
|
||||
"github.com/stretchr/testify/assert" |
||||
) |
||||
|
||||
type assetTestHarness struct { |
||||
store AssetStore |
||||
} |
||||
|
||||
func newAssetTestHarness(t *testing.T) assetTestHarness { |
||||
|
||||
db := NewInMemSQLDB() |
||||
t.Cleanup(func() { db.Close() }) |
||||
|
||||
store := NewAssetStore(db) |
||||
|
||||
return assetTestHarness{ |
||||
store: store, |
||||
} |
||||
} |
||||
|
||||
func (h *assetTestHarness) assertGet(t *testing.T, exp, id string) { |
||||
t.Helper() |
||||
buf := new(bytes.Buffer) |
||||
err := h.store.Get(id, buf) |
||||
assert.NoError(t, err) |
||||
assert.Equal(t, exp, buf.String()) |
||||
} |
||||
|
||||
func (h *assetTestHarness) assertNotFound(t *testing.T, id string) { |
||||
t.Helper() |
||||
err := h.store.Get(id, io.Discard) |
||||
assert.ErrorIs(t, ErrAssetNotFound, err) |
||||
} |
||||
|
||||
func TestAssetStore(t *testing.T) { |
||||
|
||||
h := newAssetTestHarness(t) |
||||
|
||||
h.assertNotFound(t, "foo") |
||||
h.assertNotFound(t, "bar") |
||||
|
||||
err := h.store.Set("foo", bytes.NewBufferString("FOO")) |
||||
assert.NoError(t, err) |
||||
|
||||
h.assertGet(t, "FOO", "foo") |
||||
h.assertNotFound(t, "bar") |
||||
|
||||
err = h.store.Set("foo", bytes.NewBufferString("FOOFOO")) |
||||
assert.NoError(t, err) |
||||
|
||||
h.assertGet(t, "FOOFOO", "foo") |
||||
h.assertNotFound(t, "bar") |
||||
|
||||
assert.NoError(t, h.store.Delete("foo")) |
||||
h.assertNotFound(t, "foo") |
||||
h.assertNotFound(t, "bar") |
||||
|
||||
assert.NoError(t, h.store.Delete("bar")) |
||||
h.assertNotFound(t, "foo") |
||||
h.assertNotFound(t, "bar") |
||||
} |
Loading…
Reference in new issue