125 lines
2.3 KiB
Go
125 lines
2.3 KiB
Go
// Copyright 2019 Drone.IO Inc. All rights reserved.
|
|
// Use of this source code is governed by the Drone Non-Commercial License
|
|
// that can be found in the LICENSE file.
|
|
|
|
package logs
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"io/ioutil"
|
|
|
|
"github.com/drone/drone/store/shared/db"
|
|
"github.com/drone/drone/core"
|
|
)
|
|
|
|
// New returns a new LogStore.
|
|
func New(db *db.DB) core.LogStore {
|
|
return &logStore{db}
|
|
}
|
|
|
|
type logStore struct {
|
|
db *db.DB
|
|
}
|
|
|
|
func (s *logStore) Find(ctx context.Context, step int64) (io.ReadCloser, error) {
|
|
out := &logs{ID: step}
|
|
err := s.db.View(func(queryer db.Queryer, binder db.Binder) error {
|
|
query, args, err := binder.BindNamed(queryKey, out)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
row := queryer.QueryRow(query, args...)
|
|
return scanRow(row, out)
|
|
})
|
|
return ioutil.NopCloser(
|
|
bytes.NewBuffer(out.Data),
|
|
), err
|
|
}
|
|
|
|
func (s *logStore) Create(ctx context.Context, step int64, r io.Reader) error {
|
|
data, err := ioutil.ReadAll(r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.db.Lock(func(execer db.Execer, binder db.Binder) error {
|
|
params := &logs{
|
|
ID: step,
|
|
Data: data,
|
|
}
|
|
stmt, args, err := binder.BindNamed(stmtInsert, params)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = execer.Exec(stmt, args...)
|
|
return err
|
|
})
|
|
}
|
|
|
|
func (s *logStore) Update(ctx context.Context, step int64, r io.Reader) error {
|
|
data, err := ioutil.ReadAll(r)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.db.Lock(func(execer db.Execer, binder db.Binder) error {
|
|
params := &logs{
|
|
ID: step,
|
|
Data: data,
|
|
}
|
|
stmt, args, err := binder.BindNamed(stmtUpdate, params)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = execer.Exec(stmt, args...)
|
|
return err
|
|
})
|
|
}
|
|
|
|
func (s *logStore) Delete(ctx context.Context, step int64) error {
|
|
return s.db.Lock(func(execer db.Execer, binder db.Binder) error {
|
|
params := &logs{
|
|
ID: step,
|
|
}
|
|
stmt, args, err := binder.BindNamed(stmtDelete, params)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = execer.Exec(stmt, args...)
|
|
return err
|
|
})
|
|
}
|
|
|
|
type logs struct {
|
|
ID int64 `db:"log_id"`
|
|
Data []byte `db:"log_data"`
|
|
}
|
|
|
|
const queryKey = `
|
|
SELECT
|
|
log_id
|
|
,log_data
|
|
FROM logs
|
|
WHERE log_id = :log_id
|
|
`
|
|
|
|
const stmtInsert = `
|
|
INSERT INTO logs (
|
|
log_id
|
|
,log_data
|
|
) VALUES (
|
|
:log_id
|
|
,:log_data
|
|
)
|
|
`
|
|
|
|
const stmtUpdate = `
|
|
UPDATE logs
|
|
SET log_data = :log_data
|
|
WHERE log_id = :log_id
|
|
`
|
|
|
|
const stmtDelete = `
|
|
DELETE FROM logs
|
|
WHERE log_id = :log_id
|
|
`
|