// GetProxyByContainerID returns a proxy object by container ID
func GetProxyByContainerID(dbConn *sql.DB, containerID string) (types.Proxy, error) {
var rows *sql.Rows
proxy := types.Proxy{}
var err error
queryStr := fmt.Sprintf("select p.usename, p.passwd, p.projectid, p.id, p.containerid, c.name , p.port, p.host, p.databasename from proxy p, container c where p.containerid = c.id and c.id = %s", containerID)
logit.Info.Println("GetProxyByContainerID:" + queryStr)
rows, err = dbConn.Query(queryStr)
if err != nil {
return proxy, err
}
defer rows.Close()
for rows.Next() {
if err = rows.Scan(
&proxy.Usename,
&proxy.Passwd,
&proxy.ProjectID,
&proxy.ID, &proxy.ContainerID,
&proxy.ContainerName, &proxy.Port, &proxy.Host, &proxy.Database); err != nil {
return proxy, err
}
}
if err = rows.Err(); err != nil {
return proxy, err
}
var unencrypted string
unencrypted, err = sec.DecryptPassword(proxy.Passwd)
if err != nil {
return proxy, err
}
proxy.Passwd = unencrypted
//see if the container is up
var resp cpmcontainerapi.StatusResponse
resp, err = cpmcontainerapi.StatusClient(proxy.ContainerName)
proxy.ContainerStatus = resp.Status
if err != nil {
return proxy, err
}
//test the remote host connectivity
var hoststatus string
hoststatus, err = util.FastPing(proxy.Port, proxy.Host)
if err != nil {
return proxy, err
}
if hoststatus == "OFFLINE" {
proxy.Status = hoststatus
return proxy, err
}
//test the database port on the remote host
proxy.Status, err = GetDatabaseStatus(proxy, containerID)
if err != nil {
return proxy, err
}
return proxy, nil
}
// GetNode returns the container node definition
func GetNode(w rest.ResponseWriter, r *rest.Request) {
dbConn, err := util.GetConnection(CLUSTERADMIN_DB)
if err != nil {
logit.Error.Println(err.Error())
rest.Error(w, err.Error(), 400)
return
}
defer dbConn.Close()
err = secimpl.Authorize(dbConn, r.PathParam("Token"), "perm-read")
if err != nil {
logit.Error.Println(err.Error())
rest.Error(w, err.Error(), http.StatusUnauthorized)
return
}
ID := r.PathParam("ID")
if ID == "" {
logit.Error.Println("error node ID required")
rest.Error(w, "node ID required", http.StatusBadRequest)
return
}
node, err2 := admindb.GetContainer(dbConn, ID)
if node.ID == "" {
rest.NotFound(w, r)
return
}
if err2 != nil {
logit.Error.Println(err2.Error())
rest.Error(w, err2.Error(), http.StatusBadRequest)
return
}
var currentStatus = "UNKNOWN"
request := &swarmapi.DockerInspectRequest{}
var inspectInfo swarmapi.DockerInspectResponse
request.ContainerName = node.Name
inspectInfo, err = swarmapi.DockerInspect(request)
if err != nil {
logit.Error.Println(err.Error())
currentStatus = CONTAINER_NOT_FOUND
}
if currentStatus != "CONTAINER NOT FOUND" {
var pgport types.Setting
pgport, err = admindb.GetSetting(dbConn, "PG-PORT")
if err != nil {
logit.Error.Println(err.Error())
rest.Error(w, err.Error(), http.StatusBadRequest)
return
}
currentStatus, err = util.FastPing(pgport.Value, node.Name)
if err != nil {
logit.Error.Println(err.Error())
rest.Error(w, err.Error(), http.StatusBadRequest)
return
}
//logit.Info.Println("pinging db finished")
}
clusternode := new(types.ClusterNode)
clusternode.ID = node.ID
clusternode.ClusterID = node.ClusterID
clusternode.Name = node.Name
clusternode.Role = node.Role
clusternode.Image = node.Image
clusternode.CreateDate = node.CreateDate
clusternode.Status = currentStatus
clusternode.ProjectID = node.ProjectID
clusternode.ProjectName = node.ProjectName
clusternode.ClusterName = node.ClusterName
clusternode.ServerID = inspectInfo.ServerID
clusternode.IPAddress = inspectInfo.IPAddress
w.WriteJson(clusternode)
}
func configureCluster(profile string, dbConn *sql.DB, cluster types.Cluster, autocluster bool) error {
logit.Info.Println("configureCluster:GetCluster")
//get master node for this cluster
master, err := admindb.GetContainerMaster(dbConn, cluster.ID)
if err != nil {
logit.Error.Println(err.Error())
return err
}
var pgport types.Setting
pgport, err = admindb.GetSetting(dbConn, "PG-PORT")
if err != nil {
logit.Error.Println(err.Error())
return err
}
var sleepSetting types.Setting
sleepSetting, err = admindb.GetSetting(dbConn, "SLEEP-PROV")
if err != nil {
logit.Error.Println("configureCluster:" + err.Error())
return err
}
//logit.Info.Println("configureCluster:GetContainerMaster")
//configure master postgresql.conf file
var data string
info := new(template.PostgresqlParameters)
info.PG_PORT = pgport.Value
err = template.GetTuningParms(dbConn, profile, info)
if err != nil {
logit.Error.Println("configureCluster:" + err.Error())
return err
}
if cluster.ClusterType == "synchronous" {
info.CLUSTER_TYPE = "*"
data, err = template.Postgresql("master", info)
} else {
//TODO verify these next 2 lines look erroneous
info.CLUSTER_TYPE = "*"
data, err = template.Postgresql("master", info)
//info.CLUSTER_TYPE = ""
//data, err = template.Postgresql("master", info)
}
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:master postgresql.conf generated")
//write master postgresql.conf file remotely
_, err = cpmcontainerapi.RemoteWritefileClient("/pgdata/postgresql.conf", data, master.Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:master postgresql.conf copied to remote")
//get domain name
var domainname types.Setting
domainname, err = admindb.GetSetting(dbConn, "DOMAIN-NAME")
if err != nil {
logit.Error.Println(err.Error())
return err
}
//configure master pg_hba.conf file
rules := make([]template.Rule, 0)
data, err = template.Hba(dbConn, "master", master.Name, pgport.Value, cluster.ID, domainname.Value, rules)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:master pg_hba.conf generated")
//write master pg_hba.conf file remotely
_, err = cpmcontainerapi.RemoteWritefileClient("/pgdata/pg_hba.conf", data, master.Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:master pg_hba.conf copied remotely")
//restart postgres after the config file changes
var stopResp cpmcontainerapi.StopPGResponse
stopResp, err = cpmcontainerapi.StopPGClient(master.Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster: master stoppg output was" + stopResp.Output)
var startResp cpmcontainerapi.StartPGResponse
startResp, err = cpmcontainerapi.StartPGClient(master.Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:master startpg output was" + startResp.Output)
//sleep loop until the master's PG can respond
var sleepTime time.Duration
sleepTime, err = time.ParseDuration(sleepSetting.Value)
var found = false
var currentStatus string
var masterhost = master.Name
for i := 0; i < 20; i++ {
//currentStatus, err = GetPGStatus2(dbConn, master.Name, masterhost)
currentStatus, err = util.FastPing(pgport.Value, master.Name)
if currentStatus == "RUNNING" {
logit.Info.Println("master is running...continuing")
found = true
break
} else {
logit.Info.Println("sleeping waiting on master..")
time.Sleep(sleepTime)
}
}
if !found {
logit.Info.Println("configureCluster: timed out waiting on master pg to start")
return errors.New("timeout waiting for master pg to respond")
}
standbynodes, err2 := admindb.GetAllStandbyContainers(dbConn, cluster.ID)
if err2 != nil {
logit.Error.Println(err.Error())
return err
}
//configure all standby nodes
var stopPGResp cpmcontainerapi.StopPGResponse
i := 0
for i = range standbynodes {
if standbynodes[i].Role == STANDBY {
//stop standby
if !autocluster {
stopPGResp, err = cpmcontainerapi.StopPGClient(standbynodes[i].Name)
if err != nil {
logit.Error.Println(err.Error())
logit.Error.Println("configureCluster:stop output was" + stopPGResp.Output)
return err
}
//logit.Info.Println("configureCluster:stop output was" + stopPGResp.Output)
}
//
var username = "******"
var password = "******"
//create base backup from master
var backupresp cpmcontainerapi.BasebackupResponse
backupresp, err = cpmcontainerapi.BasebackupClient(masterhost+"."+domainname.Value, standbynodes[i].Name, username, password)
if err != nil {
logit.Error.Println(err.Error())
logit.Error.Println("configureCluster:basebackup output was" + backupresp.Output)
return err
}
//logit.Info.Println("configureCluster:basebackup output was" + backupresp.Output)
data, err = template.Recovery(masterhost, pgport.Value, "postgres")
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:standby recovery.conf generated")
//write standby recovery.conf file remotely
_, err = cpmcontainerapi.RemoteWritefileClient("/pgdata/recovery.conf", data, standbynodes[i].Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:standby recovery.conf copied remotely")
info := new(template.PostgresqlParameters)
info.PG_PORT = pgport.Value
info.CLUSTER_TYPE = ""
err = template.GetTuningParms(dbConn, profile, info)
if err != nil {
logit.Error.Println(err.Error())
return err
}
data, err = template.Postgresql(STANDBY, info)
if err != nil {
logit.Error.Println(err.Error())
return err
}
//write standby postgresql.conf file remotely
_, err = cpmcontainerapi.RemoteWritefileClient("/pgdata/postgresql.conf", data, standbynodes[i].Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:standby postgresql.conf copied remotely")
//configure standby pg_hba.conf file
data, err = template.Hba(dbConn, STANDBY, standbynodes[i].Name, pgport.Value, cluster.ID, domainname.Value, rules)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:standby pg_hba.conf generated")
//write standby pg_hba.conf file remotely
_, err = cpmcontainerapi.RemoteWritefileClient("/pgdata/pg_hba.conf", data, standbynodes[i].Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:standby pg_hba.conf copied remotely")
//start standby
var stResp cpmcontainerapi.StartPGOnStandbyResponse
stResp, err = cpmcontainerapi.StartPGOnStandbyClient(standbynodes[i].Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:standby startpg output was" + stResp.Output)
}
i++
}
logit.Info.Println("configureCluster: sleeping 5 seconds before configuring pgpool...")
clustersleepTime, _ := time.ParseDuration("5s")
time.Sleep(clustersleepTime)
pgpoolNode, err4 := admindb.GetContainerPgpool(dbConn, cluster.ID)
//logit.Info.Println("configureCluster: lookup pgpool node")
if err4 != nil {
logit.Error.Println(err.Error())
return err
}
//logit.Info.Println("configureCluster:" + pgpoolNode.Name)
//configure the pgpool includes all standby nodes AND the master node
poolnames := make([]string, len(standbynodes)+1)
i = 0
for i = range standbynodes {
poolnames[i] = standbynodes[i].Name + "." + domainname.Value
i++
}
poolnames[i] = master.Name + "." + domainname.Value
//generate pgpool.conf HOST_LIST
data, err = template.Poolconf(poolnames)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:pgpool pgpool.conf generated")
//write pgpool.conf to remote pool node
_, err = cpmcontainerapi.RemoteWritefileClient(util.GetBase()+"/bin/"+"pgpool.conf", data, pgpoolNode.Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:pgpool pgpool.conf copied remotely")
//generate pool_passwd
data, err = template.Poolpasswd()
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:pgpool pool_passwd generated")
//write pgpool.conf to remote pool node
_, err = cpmcontainerapi.RemoteWritefileClient(util.GetBase()+"/bin/"+"pool_passwd", data, pgpoolNode.Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:pgpool pool_passwd copied remotely")
//g enerate pool_hba.conf
cars := make([]template.Rule, 0)
data, err = template.Poolhba(cars)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:pgpool pool_hba generated")
//write pgpool.conf to remote pool node
_, err = cpmcontainerapi.RemoteWritefileClient(util.GetBase()+"/bin/"+"pool_hba.conf", data, pgpoolNode.Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster:pgpool pool_hba copied remotely")
//start pgpool
var startPoolResp cpmcontainerapi.StartPgpoolResponse
startPoolResp, err = cpmcontainerapi.StartPgpoolClient(pgpoolNode.Name)
if err != nil {
logit.Error.Println(err.Error())
return err
}
logit.Info.Println("configureCluster: pgpool startpgpool output was" + startPoolResp.Output)
//finally, update the cluster to show that it is
//initialized!
cluster.Status = "initialized"
err = admindb.UpdateCluster(dbConn, cluster)
if err != nil {
logit.Error.Println(err.Error())
return err
}
return nil
}
func performConfigUpdate(dbConn *sql.DB, ContainerID string) error {
logit.Info.Println("performConfigUpdate....")
cars, err := GetAllContainerAccessRule(dbConn, ContainerID)
if err != nil {
logit.Error.Println(err.Error())
return err
}
container, err := admindb.GetContainer(dbConn, ContainerID)
if err != nil {
logit.Error.Println("GetNode: " + err.Error())
return err
}
var currentStatus string
//currentStatus, err = GetPGStatus2(dbConn, container.Name, container.Name)
var pgport types.Setting
pgport, err = admindb.GetSetting(dbConn, "PG-PORT")
if err != nil {
logit.Error.Println(err.Error())
return err
}
currentStatus, err = util.FastPing(pgport.Value, container.Name)
if err != nil {
logit.Error.Println("GetNode:" + err.Error())
return err
}
if currentStatus != "RUNNING" {
logit.Info.Println("performConfigUpdate....starting postgres")
if container.Role == "pgpool" {
var spgresp cpmcontainerapi.StartPgpoolResponse
spgresp, err = cpmcontainerapi.StartPgpoolClient(container.Name)
logit.Info.Println("AdminStartpg:" + spgresp.Output)
} else {
var srep cpmcontainerapi.StartPGResponse
srep, err = cpmcontainerapi.StartPGClient(container.Name)
logit.Info.Println("AdminStartpg:" + srep.Output)
}
if err != nil {
logit.Error.Println("AdminStartpg:" + err.Error())
return err
}
}
//make template changes here
logit.Info.Println("performConfigUpdate....making template changes")
templateChange(dbConn, container.Name, cars, container.Role)
//restart postgres
if container.Role == "pgpool" {
logit.Info.Println("performConfigUpdate....stopping pgpool")
var stoppoolResp cpmcontainerapi.StopPgpoolResponse
stoppoolResp, err = cpmcontainerapi.StopPgpoolClient(container.Name)
logit.Info.Println("AdminStoppg:" + stoppoolResp.Output)
} else {
logit.Info.Println("performConfigUpdate....stopping postgres")
var stoppgResp cpmcontainerapi.StopPGResponse
stoppgResp, err = cpmcontainerapi.StopPGClient(container.Name)
logit.Info.Println("AdminStoppg:" + stoppgResp.Output)
}
if err != nil {
logit.Error.Println("AdminStoppg:" + err.Error())
return err
}
if container.Role == "pgpool" {
logit.Info.Println("performConfigUpdate....starting pgpool")
var spgresp cpmcontainerapi.StartPgpoolResponse
spgresp, err = cpmcontainerapi.StartPgpoolClient(container.Name)
logit.Info.Println("AdminStartpg:" + spgresp.Output)
} else {
logit.Info.Println("performConfigUpdate....starting postgres")
var srep cpmcontainerapi.StartPGResponse
srep, err = cpmcontainerapi.StartPGClient(container.Name)
logit.Info.Println("AdminStartpg:" + srep.Output)
}
if err != nil {
logit.Error.Println("AdminStartpg:" + err.Error())
return err
}
return err
}