d1http/query.go at master · pubflow/d1http · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package d1http

import (
	"context"
	"net/http"
	"strings"
)

func (c *Client) Raw(ctx context.Context, sql string, params ...any) ([]RawResult, error) {
	return c.RawIn(ctx, "", sql, params...)
}

func (c *Client) RawIn(ctx context.Context, databaseID, sql string, params ...any) ([]RawResult, error) {
	id, err := c.databaseID(databaseID)
	if err != nil {
		return nil, err
	}
	var env envelope[[]RawResult]
	err = c.do(ctx, http.MethodPost, apiPath("accounts", c.cfg.AccountID, "d1/database", id, "raw"), nil, rawRequest{SQL: sql, Params: params}, &env)
	return env.Result, err
}

func (c *Client) Query(ctx context.Context, sql string, params ...any) ([]QueryResult, error) {
	return c.QueryIn(ctx, "", sql, params...)
}

func (c *Client) QueryIn(ctx context.Context, databaseID, sql string, params ...any) ([]QueryResult, error) {
	id, err := c.databaseID(databaseID)
	if err != nil {
		return nil, err
	}
	var env envelope[[]QueryResult]
	err = c.do(ctx, http.MethodPost, apiPath("accounts", c.cfg.AccountID, "d1/database", id, "query"), nil, rawRequest{SQL: sql, Params: params}, &env)
	return env.Result, err
}

func (c *Client) Exec(ctx context.Context, sql string, params ...any) (*RawResult, error) {
	results, err := c.Raw(ctx, sql, params...)
	if err != nil {
		return nil, err
	}
	if len(results) == 0 {
		return nil, nil
	}
	return &results[len(results)-1], nil
}

func (c *Client) BatchRaw(ctx context.Context, statements []Statement) ([]RawResult, error) {
	return c.BatchRawIn(ctx, "", statements)
}

func (c *Client) BatchRawIn(ctx context.Context, databaseID string, statements []Statement) ([]RawResult, error) {
	sql, params := joinStatements(statements)
	return c.RawIn(ctx, databaseID, sql, params...)
}

func (c *Client) BatchQuery(ctx context.Context, statements []Statement) ([]QueryResult, error) {
	return c.BatchQueryIn(ctx, "", statements)
}

func (c *Client) BatchQueryIn(ctx context.Context, databaseID string, statements []Statement) ([]QueryResult, error) {
	sql, params := joinStatements(statements)
	return c.QueryIn(ctx, databaseID, sql, params...)
}

func joinStatements(statements []Statement) (string, []any) {
	parts := make([]string, 0, len(statements))
	params := []any{}
	for _, stmt := range statements {
		s := strings.TrimSpace(stmt.SQL)
		if s == "" {
			continue
		}
		parts = append(parts, strings.TrimRight(s, ";"))
		params = append(params, stmt.Params...)
	}
	return strings.Join(parts, "; ") + ";", params
}