//curl -i -X GET localhost:9090/sshRules/9999
func (this *SshRulesController) GetByUid() {
uid := this.Ctx.Input.Params[":uid"]
if uid == "" {
this.Ctx.Output.SetStatus(403)
this.Ctx.Output.Body([]byte(`{"result":1,"error":"param error"}`))
this.StopRun()
}
dbconn, err := models.InitDbConn(0)
if err != nil {
logs.Error("init db conn error:", err, 0)
this.Ctx.Output.SetStatus(500)
this.Ctx.Output.Body([]byte(`{"result":1,"error":"init db conn error"}`))
this.StopRun()
}
defer dbconn.Close()
sshRulesM := new(models.SshRuleManage)
rulelist, err := sshRulesM.QueryByUid(dbconn, uid, 0)
if err != nil {
logs.Error("ssh rule query by uid error:", err, 0)
this.Ctx.Output.SetStatus(500)
this.Ctx.Output.Body([]byte(`{"result":1,"error":"` + err.Error() + `"}`))
this.StopRun()
}
this.Data["json"] = rulelist
this.ServeJson()
}
//curl -i -X GET localhost:9090/sshRules/9999
func (this *SshRulesController) DeleteByUid() {
uid := this.Ctx.Input.Params[":uid"]
if uid == "" {
this.Ctx.Output.SetStatus(403)
this.Ctx.Output.Body([]byte(`{"result":1,"error":"param error"}`))
this.StopRun()
}
logid, _ := this.GetInt("logid")
if logid == 0 {
this.Ctx.Output.SetStatus(403)
this.Ctx.Output.Body([]byte(`{"result":1,"error":"logid param error"}`))
this.StopRun()
}
logs.Normal("delete param:", "uid:", uid, "logid:", logid)
dbconn, err := models.InitDbConn(logid)
if err != nil {
logs.Error("init db conn err:", err, "logid:", logid)
this.Ctx.Output.SetStatus(500)
this.Ctx.Output.Body([]byte(`{"result":1,"error":"init db conn error"}`))
this.StopRun()
}
defer dbconn.Close()
dbconn.Exec("START TRANSACTION")
logs.Normal("start transaction", "logid:", logid)
sshRulesM := new(models.SshRuleManage)
delrulelist, _ := sshRulesM.QueryByUid(dbconn, uid, logid)
err = sshRulesM.DeleteByUid(dbconn, uid, logid)
if err != nil {
dbconn.Exec("ROLLBACK")
logs.Normal("ROLLBACK", "logid:", logid)
logs.Error("delete by uid err:", err, "logid:", logid)
this.Ctx.Output.SetStatus(500)
this.Ctx.Output.Body([]byte(`{"result":1,"error":"` + err.Error() + `"}`))
this.StopRun()
}
err = models.DeleteContainerUserFromProxy(delrulelist, logid)
if err != nil {
logs.Error("DeleteContainerUserFromProxy error:", err, "logid:", logid)
dbconn.Exec("ROLLBACK")
logs.Normal("ROLLBACK", logid)
}
dbconn.Exec("COMMIT")
logs.Normal("COMMIT", "logid:", logid)
logs.Normal("delete OK!", "logid:", logid)
this.Ctx.Output.Body([]byte(`{"result":0}`))
this.StopRun()
}
func (this *SshKeysController) reloadSshPublicKeys(dbconn *sql.DB, uid string, logid int64) error {
sshKeysM := new(models.SshKeysManage)
pubSshKeyMap, errall := sshKeysM.GetAll(dbconn, uid, logid)
logs.Normal("reloadSshPublicKeys get sshkey map:", pubSshKeyMap, "logid:", logid)
if errall != nil {
logs.Error("sshKeysM.GetAll error!", errall, " uid:", uid, "logid:", logid)
return errall
}
//check ssh rules,if the rules is null,then only change ssh keys;
// if the rules is not null,then change the proxy server
sshRuleM := new(models.SshRuleManage)
rulelist, err := sshRuleM.QueryByUid(dbconn, uid, logid)
if err != nil {
logs.Error("sshRuleM.QueryByUid error!", err, " uid:", uid, "logid:", logid)
return err
}
logs.Normal("sshRuleM.QueryByUid get rule list:", rulelist, "logid:", logid)
if len(rulelist) > 0 {
err = models.UpdateRule(rulelist, pubSshKeyMap, logid)
}
return err
}