package main

import (
	"context"

	"github.com/dolphindb/api-go/api"
)

func main() {
	host := "192.168.0.153:8848"
	// init client
	db, err := api.NewDolphinDBClient(context.TODO(), host, nil)
	if err != nil {
		// Handle exception
		panic(err)
	}

	// connect to server
	err = db.Connect()
	if err != nil {
		// Handle exception
		panic(err)
	}

	// init login request
	loginReq := &api.LoginRequest{
		UserID:   "admin",
		Password: "123456",
	}

	// login dolphindb
	err = db.Login(loginReq)
	if err != nil {
		// Handle exception
		panic(err)
	}

	// init create database request
	dbReq := &api.DatabaseRequest{
		Directory:       "dfs://account1",
		PartitionType:   "VALUE",
		PartitionScheme: "1..1000", // 以project_id进行分区
		DBHandle:        "example",
	}

	// create database
	dt, err := db.Database(dbReq)
	if err != nil {
		// Handle exception
		panic(err)
	}

	// init create partitioned table request
	createReq := &api.CreatePartitionedTableRequest{
		SrcTable:             "sourceTable",
		PartitionedTableName: "tableName",
		PartitionColumns:     []string{"project_id"},
	}

	// create partitioned table with database handler
	_, err = dt.CreatePartitionedTable(createReq)
	if err != nil {
		// Handle exception
		panic(err)
	}
}