//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 }