115 lines
2.8 KiB
Go
115 lines
2.8 KiB
Go
//go:build plus
|
|
|
|
package db
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/systemconfigs"
|
|
"github.com/iwind/TeaGo/actions"
|
|
"strings"
|
|
)
|
|
|
|
const clickhouseConfigCode = "clickhouseConfig"
|
|
|
|
type ClickHouseAction struct {
|
|
actionutils.ParentAction
|
|
}
|
|
|
|
func (this *ClickHouseAction) Init() {
|
|
this.Nav("db", "db", "clickhouse")
|
|
}
|
|
|
|
func (this *ClickHouseAction) RunGet(params struct{}) {
|
|
this.Data["mainTab"] = "clickhouse"
|
|
resp, err := this.RPC().SysSettingRPC().ReadSysSetting(this.AdminContext(), &pb.ReadSysSettingRequest{Code: clickhouseConfigCode})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
cfg := &systemconfigs.ClickHouseSetting{Port: 8123, Database: "default", Scheme: "http"}
|
|
if len(resp.ValueJSON) > 0 {
|
|
_ = json.Unmarshal(resp.ValueJSON, cfg)
|
|
}
|
|
if cfg.Port <= 0 {
|
|
cfg.Port = 8123
|
|
}
|
|
if cfg.Database == "" {
|
|
cfg.Database = "default"
|
|
}
|
|
if strings.TrimSpace(cfg.Scheme) == "" {
|
|
cfg.Scheme = "http"
|
|
}
|
|
this.Data["config"] = map[string]interface{}{
|
|
"host": cfg.Host,
|
|
"port": cfg.Port,
|
|
"user": cfg.User,
|
|
"password": cfg.Password,
|
|
"database": cfg.Database,
|
|
"scheme": cfg.Scheme,
|
|
"tlsSkipVerify": cfg.TLSSkipVerify,
|
|
"tlsServerName": cfg.TLSServerName,
|
|
}
|
|
this.Show()
|
|
}
|
|
|
|
func (this *ClickHouseAction) RunPost(params struct {
|
|
Host string
|
|
Port int
|
|
User string
|
|
Password string
|
|
Database string
|
|
Scheme string
|
|
TLSSkipVerify bool
|
|
TLSServerName string
|
|
|
|
Must *actions.Must
|
|
}) {
|
|
defer this.CreateLogInfo(codes.DBNode_LogUpdateDBNode, 0)
|
|
if params.Port <= 0 {
|
|
params.Port = 8123
|
|
}
|
|
if params.Database == "" {
|
|
params.Database = "default"
|
|
}
|
|
if params.Scheme != "https" {
|
|
params.Scheme = "http"
|
|
}
|
|
password := params.Password
|
|
if password == "" {
|
|
resp, _ := this.RPC().SysSettingRPC().ReadSysSetting(this.AdminContext(), &pb.ReadSysSettingRequest{Code: clickhouseConfigCode})
|
|
if len(resp.ValueJSON) > 0 {
|
|
var old systemconfigs.ClickHouseSetting
|
|
if json.Unmarshal(resp.ValueJSON, &old) == nil {
|
|
password = old.Password
|
|
}
|
|
}
|
|
}
|
|
cfg := &systemconfigs.ClickHouseSetting{
|
|
Host: params.Host,
|
|
Port: params.Port,
|
|
User: params.User,
|
|
Password: password,
|
|
Database: params.Database,
|
|
Scheme: params.Scheme,
|
|
TLSSkipVerify: params.TLSSkipVerify,
|
|
TLSServerName: strings.TrimSpace(params.TLSServerName),
|
|
}
|
|
valueJSON, err := json.Marshal(cfg)
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
_, err = this.RPC().SysSettingRPC().UpdateSysSetting(this.AdminContext(), &pb.UpdateSysSettingRequest{
|
|
Code: clickhouseConfigCode,
|
|
ValueJSON: valueJSON,
|
|
})
|
|
if err != nil {
|
|
this.ErrorPage(err)
|
|
return
|
|
}
|
|
this.Success()
|
|
}
|