cli can pass -db /path/to/sql.db
This commit is contained in:
@@ -2,12 +2,33 @@ package cmd
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"os"
|
||||
"show-rss/src/cleanup"
|
||||
"show-rss/src/db"
|
||||
)
|
||||
|
||||
func Config(ctx context.Context) (context.Context, func(), error) {
|
||||
ctx, err := db.Inject(ctx, "/tmp/f.db")
|
||||
type Flags struct {
|
||||
DB string
|
||||
}
|
||||
|
||||
func NewFlags(args []string) (Flags, error) {
|
||||
var result Flags
|
||||
|
||||
fs := flag.NewFlagSet(os.Args[0], flag.ContinueOnError)
|
||||
fs.StringVar(&result.DB, "db", "/tmp/f.db", "path to sqlite.db")
|
||||
err := fs.Parse(args)
|
||||
|
||||
return result, err
|
||||
}
|
||||
|
||||
func Config(ctx context.Context, args []string) (context.Context, func(), error) {
|
||||
flags, err := NewFlags(args)
|
||||
if err != nil {
|
||||
return ctx, nil, err
|
||||
}
|
||||
|
||||
ctx, err = db.Inject(ctx, flags.DB)
|
||||
if err != nil {
|
||||
return ctx, nil, err
|
||||
}
|
||||
|
||||
@@ -10,11 +10,11 @@ import (
|
||||
"time"
|
||||
)
|
||||
|
||||
func Main(ctx context.Context) error {
|
||||
func Main(ctx context.Context, args []string) error {
|
||||
ctx, can := context.WithCancel(ctx)
|
||||
defer can()
|
||||
|
||||
ctx, can, err := Config(ctx)
|
||||
ctx, can, err := Config(ctx, args)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to inject: %w", err)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user