This commit is contained in:
Ian Gulliver
2023-05-15 21:13:00 -07:00
parent d35f011d4f
commit 44aae391a8
2 changed files with 316 additions and 0 deletions

272
cover.html Normal file
View File

@@ -0,0 +1,272 @@
<!DOCTYPE html>
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
<title>store: Go Coverage Report</title>
<style>
body {
background: black;
color: rgb(80, 80, 80);
}
body, pre, #legend span {
font-family: Menlo, monospace;
font-weight: bold;
}
#topbar {
background: black;
position: fixed;
top: 0; left: 0; right: 0;
height: 42px;
border-bottom: 1px solid rgb(80, 80, 80);
}
#content {
margin-top: 50px;
}
#nav, #legend {
float: left;
margin-left: 10px;
}
#legend {
margin-top: 12px;
}
#nav {
margin-top: 10px;
}
#legend span {
margin: 0 5px;
}
.cov0 { color: rgb(192, 0, 0) }
.cov1 { color: rgb(128, 128, 128) }
.cov2 { color: rgb(116, 140, 131) }
.cov3 { color: rgb(104, 152, 134) }
.cov4 { color: rgb(92, 164, 137) }
.cov5 { color: rgb(80, 176, 140) }
.cov6 { color: rgb(68, 188, 143) }
.cov7 { color: rgb(56, 200, 146) }
.cov8 { color: rgb(44, 212, 149) }
.cov9 { color: rgb(32, 224, 152) }
.cov10 { color: rgb(20, 236, 155) }
</style>
</head>
<body>
<div id="topbar">
<div id="nav">
<select id="files">
<option value="file0">github.com/gopatchy/store/store.go (80.9%)</option>
</select>
</div>
<div id="legend">
<span>not tracked</span>
<span class="cov0">no coverage</span>
<span class="cov1">low coverage</span>
<span class="cov2">*</span>
<span class="cov3">*</span>
<span class="cov4">*</span>
<span class="cov5">*</span>
<span class="cov6">*</span>
<span class="cov7">*</span>
<span class="cov8">*</span>
<span class="cov9">*</span>
<span class="cov10">high coverage</span>
</div>
</div>
<div id="content">
<pre class="file" id="file0" style="display: none">package store
import (
"context"
"database/sql"
"encoding/json"
"fmt"
"github.com/gopatchy/metadata"
// Register sqlite3 db handler.
_ "github.com/mattn/go-sqlite3"
)
type Store struct {
db *sql.DB
}
func NewStore(conn string) (*Store, error) <span class="cov6" title="3">{
db, err := sql.Open("sqlite3", conn)
if err != nil </span><span class="cov0" title="0">{
return nil, err
}</span>
// TODO: Keep a set of prepared statements with PrepareContext()
// TODO: Consider tuning per https://phiresky.github.io/blog/2020/sqlite-performance-tuning/
<span class="cov6" title="3">return &amp;Store{
db: db,
}, nil</span>
}
func (s *Store) Close() <span class="cov6" title="3">{
s.db.Close()
}</span>
func (s *Store) Write(ctx context.Context, t string, obj any) error <span class="cov9" title="6">{
id := metadata.GetMetadata(obj).ID
js, err := json.Marshal(obj)
if err != nil </span><span class="cov0" title="0">{
return err
}</span>
<span class="cov9" title="6">err = s.exec(ctx, "INSERT INTO `%s` (id, obj) VALUES (?,?) ON CONFLICT(id) DO UPDATE SET obj=?;", t, id, js, js)
if err != nil </span><span class="cov0" title="0">{
return err
}</span>
<span class="cov9" title="6">return nil</span>
}
func (s *Store) Delete(ctx context.Context, t, id string) error <span class="cov1" title="1">{
err := s.exec(ctx, "DELETE FROM `%s` WHERE id=?", t, id)
if err != nil </span><span class="cov0" title="0">{
return err
}</span>
<span class="cov1" title="1">return nil</span>
}
func (s *Store) Read(ctx context.Context, t, id string, factory func() any) (any, error) <span class="cov7" title="4">{
rows, err := s.query(ctx, "SELECT obj FROM `%s` WHERE id=?;", t, id)
if err != nil </span><span class="cov0" title="0">{
return nil, err
}</span>
<span class="cov7" title="4">defer rows.Close()
if !rows.Next() </span><span class="cov1" title="1">{
return nil, nil
}</span>
<span class="cov6" title="3">var js []byte
err = rows.Scan(&amp;js)
if err != nil </span><span class="cov0" title="0">{
return nil, err
}</span>
<span class="cov6" title="3">obj := factory()
err = json.Unmarshal(js, obj)
if err != nil </span><span class="cov0" title="0">{
return nil, err
}</span>
<span class="cov6" title="3">return obj, nil</span>
}
func (s *Store) List(ctx context.Context, t string, factory func() any) ([]any, error) <span class="cov4" title="2">{
rows, err := s.query(ctx, "SELECT obj FROM `%s`;", t)
if err != nil </span><span class="cov0" title="0">{
return nil, err
}</span>
<span class="cov4" title="2">defer rows.Close()
ret := []any{}
for rows.Next() </span><span class="cov4" title="2">{
var js []byte
err = rows.Scan(&amp;js)
if err != nil </span><span class="cov0" title="0">{
return nil, err
}</span>
<span class="cov4" title="2">obj := factory()
err = json.Unmarshal(js, obj)
if err != nil </span><span class="cov0" title="0">{
return nil, err
}</span>
<span class="cov4" title="2">ret = append(ret, obj)</span>
}
<span class="cov4" title="2">return ret, nil</span>
}
func (s *Store) exec(ctx context.Context, query, t string, args ...any) error <span class="cov10" title="7">{
query = fmt.Sprintf(query, t)
_, err := s.db.ExecContext(ctx, query, args...)
if err == nil </span><span class="cov8" title="5">{
return nil
}</span>
<span class="cov4" title="2">_, err = s.db.ExecContext(ctx, s.tableSQL(t))
if err != nil </span><span class="cov0" title="0">{
return err
}</span>
<span class="cov4" title="2">_, err = s.db.ExecContext(ctx, query, args...)
if err != nil </span><span class="cov0" title="0">{
return err
}</span>
<span class="cov4" title="2">return nil</span>
}
func (s *Store) query(ctx context.Context, query, t string, args ...any) (*sql.Rows, error) <span class="cov9" title="6">{
query = fmt.Sprintf(query, t)
rows, err := s.db.QueryContext(ctx, query, args...)
if err == nil </span><span class="cov8" title="5">{
return rows, nil
}</span>
<span class="cov1" title="1">_, err = s.db.ExecContext(ctx, s.tableSQL(t))
if err != nil </span><span class="cov0" title="0">{
return nil, err
}</span>
<span class="cov1" title="1">return s.db.QueryContext(ctx, query, args...)</span>
}
func (s *Store) tableSQL(t string) string <span class="cov6" title="3">{
return fmt.Sprintf("CREATE TABLE IF NOT EXISTS `%s` (id TEXT NOT NULL PRIMARY KEY, obj TEXT NOT NULL);", t)
}</span>
</pre>
</div>
</body>
<script>
(function() {
var files = document.getElementById('files');
var visible;
files.addEventListener('change', onChange, false);
function select(part) {
if (visible)
visible.style.display = 'none';
visible = document.getElementById(part);
if (!visible)
return;
files.value = part;
visible.style.display = 'block';
location.hash = part;
}
function onChange() {
select(files.value);
window.scrollTo(0, 0);
}
if (location.hash != "") {
select(location.hash.substr(1));
}
if (!visible) {
select("file0");
}
})();
</script>
</html>