管理端全部功能跑通

EdgeAPI/internal/clickhouse/client.go

@@ -1,6 +1,7 @@
 package clickhouse
 
 import (
+	"bytes"
 	"context"
 	"crypto/tls"
 	"encoding/json"
@@ -12,13 +13,11 @@ import (
 	"time"
 )
 
-// Client 通过 HTTP 接口执行只读查询（SELECT），返回 JSONEachRow 解析为 map 或结构体
 type Client struct {
 	cfg     *Config
 	httpCli *http.Client
 }
 
-// NewClient 使用共享配置创建客户端
 func NewClient() *Client {
 	cfg := SharedConfig()
 	transport := &http.Transport{}
@@ -28,6 +27,7 @@ func NewClient() *Client {
 			ServerName:         cfg.TLSServerName,
 		}
 	}
+
 	return &Client{
 		cfg: cfg,
 		httpCli: &http.Client{
@@ -37,21 +37,20 @@ func NewClient() *Client {
 	}
 }
 
-// IsConfigured 是否已配置
 func (c *Client) IsConfigured() bool {
 	return c.cfg != nil && c.cfg.IsConfigured()
 }
 
-// Query 执行 SELECT，将每行 JSON 解析到 dest 切片；dest 元素类型需为 *struct 或 map
 func (c *Client) Query(ctx context.Context, query string, dest interface{}) error {
 	if !c.IsConfigured() {
 		return fmt.Errorf("clickhouse: not configured")
 	}
-	// 强制 JSONEachRow 便于解析
+
 	q := strings.TrimSpace(query)
 	if !strings.HasSuffix(strings.ToUpper(q), "FORMAT JSONEACHROW") {
 		query = q + " FORMAT JSONEachRow"
 	}
+
 	u := c.buildURL(query)
 	req, err := http.NewRequestWithContext(ctx, http.MethodGet, u, nil)
 	if err != nil {
@@ -60,28 +59,32 @@ func (c *Client) Query(ctx context.Context, query string, dest interface{}) erro
 	if c.cfg.User != "" || c.cfg.Password != "" {
 		req.SetBasicAuth(c.cfg.User, c.cfg.Password)
 	}
+
 	resp, err := c.httpCli.Do(req)
 	if err != nil {
 		return err
 	}
 	defer resp.Body.Close()
+
 	if resp.StatusCode != http.StatusOK {
 		body, _ := io.ReadAll(resp.Body)
 		return fmt.Errorf("clickhouse HTTP %d: %s", resp.StatusCode, string(body))
 	}
+
 	dec := json.NewDecoder(resp.Body)
 	return decodeRows(dec, dest)
 }
 
-// QueryRow 执行仅返回一行的查询，将结果解析到 dest（*struct 或 *map）
 func (c *Client) QueryRow(ctx context.Context, query string, dest interface{}) error {
 	if !c.IsConfigured() {
 		return fmt.Errorf("clickhouse: not configured")
 	}
+
 	q := strings.TrimSpace(query)
 	if !strings.HasSuffix(strings.ToUpper(q), "FORMAT JSONEACHROW") {
 		query = q + " FORMAT JSONEachRow"
 	}
+
 	u := c.buildURL(query)
 	req, err := http.NewRequestWithContext(ctx, http.MethodGet, u, nil)
 	if err != nil {
@@ -90,32 +93,109 @@ func (c *Client) QueryRow(ctx context.Context, query string, dest interface{}) e
 	if c.cfg.User != "" || c.cfg.Password != "" {
 		req.SetBasicAuth(c.cfg.User, c.cfg.Password)
 	}
+
 	resp, err := c.httpCli.Do(req)
 	if err != nil {
 		return err
 	}
 	defer resp.Body.Close()
+
 	if resp.StatusCode != http.StatusOK {
 		body, _ := io.ReadAll(resp.Body)
 		return fmt.Errorf("clickhouse HTTP %d: %s", resp.StatusCode, string(body))
 	}
+
 	dec := json.NewDecoder(resp.Body)
 	return decodeOneRow(dec, dest)
 }
 
+func (c *Client) Execute(ctx context.Context, query string) error {
+	if !c.IsConfigured() {
+		return fmt.Errorf("clickhouse: not configured")
+	}
+
+	u := c.buildURL(strings.TrimSpace(query))
+	req, err := http.NewRequestWithContext(ctx, http.MethodPost, u, nil)
+	if err != nil {
+		return err
+	}
+	if c.cfg.User != "" || c.cfg.Password != "" {
+		req.SetBasicAuth(c.cfg.User, c.cfg.Password)
+	}
+
+	resp, err := c.httpCli.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		body, _ := io.ReadAll(resp.Body)
+		return fmt.Errorf("clickhouse HTTP %d: %s", resp.StatusCode, string(body))
+	}
+	return nil
+}
+
+func (c *Client) InsertJSONEachRow(ctx context.Context, insertSQL string, rows []map[string]interface{}) error {
+	if len(rows) == 0 {
+		return nil
+	}
+	if !c.IsConfigured() {
+		return fmt.Errorf("clickhouse: not configured")
+	}
+
+	query := strings.TrimSpace(insertSQL)
+	if !strings.HasSuffix(strings.ToUpper(query), "FORMAT JSONEACHROW") {
+		query += " FORMAT JSONEachRow"
+	}
+
+	var payload bytes.Buffer
+	for _, row := range rows {
+		if row == nil {
+			continue
+		}
+		data, err := json.Marshal(row)
+		if err != nil {
+			return err
+		}
+		payload.Write(data)
+		payload.WriteByte('\n')
+	}
+
+	u := c.buildURL(query)
+	req, err := http.NewRequestWithContext(ctx, http.MethodPost, u, &payload)
+	if err != nil {
+		return err
+	}
+	req.Header.Set("Content-Type", "application/json")
+	if c.cfg.User != "" || c.cfg.Password != "" {
+		req.SetBasicAuth(c.cfg.User, c.cfg.Password)
+	}
+
+	resp, err := c.httpCli.Do(req)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		body, _ := io.ReadAll(resp.Body)
+		return fmt.Errorf("clickhouse HTTP %d: %s", resp.StatusCode, string(body))
+	}
+
+	return nil
+}
+
 func (c *Client) buildURL(query string) string {
 	scheme := "http"
 	if c.cfg != nil && strings.EqualFold(c.cfg.Scheme, "https") {
 		scheme = "https"
 	}
-	rawURL := fmt.Sprintf("%s://%s:%d/?query=%s&database=%s",
+	return fmt.Sprintf("%s://%s:%d/?query=%s&database=%s",
 		scheme, c.cfg.Host, c.cfg.Port, url.QueryEscape(query), url.QueryEscape(c.cfg.Database))
-	return rawURL
 }
 
-// decodeRows 将 JSONEachRow 流解析到 slice；元素类型须为 *struct 或 *[]map[string]interface{}
 func decodeRows(dec *json.Decoder, dest interface{}) error {
-	// dest 应为 *[]*SomeStruct 或 *[]map[string]interface{}
 	switch d := dest.(type) {
 	case *[]map[string]interface{}:
 		*d = (*d)[:0]
@@ -130,7 +210,7 @@ func decodeRows(dec *json.Decoder, dest interface{}) error {
 			*d = append(*d, row)
 		}
 	default:
-		return fmt.Errorf("clickhouse: unsupported dest type for Query (use *[]map[string]interface{} or implement decoder)")
+		return fmt.Errorf("clickhouse: unsupported dest type for Query (use *[]map[string]interface{})")
 	}
 }