go get github.com/ClickHouse/clickhouse-go/v2

type Clickhouse struct {
	Host       string    // 服務(wù)端主機(jī)
	Port       int       // 端口
	DB         string    // 數(shù)據(jù)庫(kù)
	User       string    // 用戶名
	Password   string    // 密碼
	Connection *sql.DB   // 建立連接后存放連接
	Rows       *sql.Rows // 運(yùn)行sql后的結(jié)果存放
}

var (
	host  = flag.String("host", "localhost", "clickhouse host")
	port  = flag.Int("port", 8123, "clickhouse port")
	user  = flag.String("user", "default", "clichouse user")
	pass  = flag.String("password", "", "clickhouse password")
	db    = flag.String("db", "default", "clickhouse database")
	query = flag.String("query", "show tables", "query you will run")
	mode  = flag.String("mode", "driver", "driver or dsn")
)

// 
func (c *Clickhouse) Conn() {
	c.Connection = clickhouse.OpenDB(&clickhouse.Options{
		Addr: []string{fmt.Sprintf("%s:%d", c.Host, c.Port)},
		Auth: clickhouse.Auth{
			Database: c.DB,
			Username: c.User,
			Password: c.Password,
		},
		Settings: clickhouse.Settings{
			"max_execution_time": 60,
		},
		DialTimeout: 5 * time.Second,
		Compression: &clickhouse.Compression{
			Method: clickhouse.CompressionBrotli,
			Level:  5,
		},
		// 必須添加協(xié)議方式
		Protocol: clickhouse.HTTP,
	})

}

func (c *Clickhouse) ConnDsn() {
	conn, err := sql.Open("clickhouse", fmt.Sprintf("http://%s:%d/%s?username=%s&password=%s", c.Host, c.Port, c.DB, c.User, c.Password))
	if err != nil {
		log.Printf("Connect to the server failed, %s.\n", err.Error())
		return
	}
	c.Connection = conn
}

func (c *Clickhouse) Close() {
	c.Connection.Close()
	c.Rows.Close()
}

func (c *Clickhouse) Select(query string) {
	rows, err := c.Connection.Query(query)
	if err != nil {
		log.Printf("Query select failed, %s.\n", err.Error())
		return
	}
	c.Rows = rows
}

func (c *Clickhouse) Show() {
	cols, err := c.Rows.Columns()
	if err != nil {
		log.Printf("Failed to get table columns, %s.\n", err.Error())
		return
	}
	// 一行數(shù)據(jù)，使用any是為了避開(kāi)數(shù)據(jù)類(lèi)型的問(wèn)題
	var rows = make([]any, len(cols))
	// 存實(shí)際的值，是byte數(shù)組，長(zhǎng)度以列的數(shù)量為準(zhǔn)
	var values = make([][]byte, len(cols))
	for i := 0; i < len(cols); i++ {
		rows[i] = &values[i]
	}
	// 打印表頭
	fmt.Println(strings.Join(cols, ","))
	for c.Rows.Next() {
		if err = c.Rows.Scan(rows...); err != nil {
			fmt.Println(err)
			return
		}
		var vString []string
		for _, v := range values {
			vString = append(vString, string(v))
		}
		// 逐行打印出來(lái)
		fmt.Println(strings.Join(vString, ","))
	}
}

go run main.go -host hostname -password paswword -query "select * from clusters" -db system -mode dsn