zicla
7 years ago
8 changed files with 435 additions and 69 deletions
@ -0,0 +1,129 @@ |
|||
package rest |
|||
|
|||
import ( |
|||
"net/http" |
|||
"strconv" |
|||
) |
|||
|
|||
type SecurityVisitController struct { |
|||
BaseController |
|||
securityVisitDao *SecurityVisitDao |
|||
securityVisitService *SecurityVisitService |
|||
} |
|||
|
|||
//初始化方法
|
|||
func (this *SecurityVisitController) Init(context *Context) { |
|||
this.BaseController.Init(context) |
|||
|
|||
//手动装填本实例的Bean. 这里必须要用中间变量方可。
|
|||
b := context.GetBean(this.securityVisitDao) |
|||
if b, ok := b.(*SecurityVisitDao); ok { |
|||
this.securityVisitDao = b |
|||
} |
|||
|
|||
b = context.GetBean(this.securityVisitService) |
|||
if b, ok := b.(*SecurityVisitService); ok { |
|||
this.securityVisitService = b |
|||
} |
|||
|
|||
} |
|||
|
|||
//注册自己的路由。
|
|||
func (this *SecurityVisitController) RegisterRoutes() map[string]func(writer http.ResponseWriter, request *http.Request) { |
|||
|
|||
routeMap := make(map[string]func(writer http.ResponseWriter, request *http.Request)) |
|||
|
|||
//每个Controller需要主动注册自己的路由。
|
|||
routeMap["/api/security/visit/delete"] = this.Wrap(this.Delete, USER_ROLE_USER) |
|||
routeMap["/api/security/visit/detail"] = this.Wrap(this.Detail, USER_ROLE_USER) |
|||
routeMap["/api/security/visit/page"] = this.Wrap(this.Page, USER_ROLE_USER) |
|||
|
|||
return routeMap |
|||
} |
|||
|
|||
//查看详情。
|
|||
func (this *SecurityVisitController) Detail(writer http.ResponseWriter, request *http.Request) *WebResult { |
|||
|
|||
uuid := request.FormValue("uuid") |
|||
if uuid == "" { |
|||
return this.Error("图片缓存的uuid必填") |
|||
} |
|||
|
|||
securityVisit := this.securityVisitService.Detail(uuid) |
|||
|
|||
//验证当前之人是否有权限查看这么详细。
|
|||
user := this.checkUser(writer, request) |
|||
if user.Role != USER_ROLE_ADMINISTRATOR { |
|||
if securityVisit.UserUuid != user.Uuid { |
|||
panic("没有权限查看该图片缓存") |
|||
} |
|||
} |
|||
|
|||
return this.Success(securityVisit) |
|||
|
|||
} |
|||
|
|||
//按照分页的方式查询
|
|||
func (this *SecurityVisitController) Page(writer http.ResponseWriter, request *http.Request) *WebResult { |
|||
|
|||
//如果是根目录，那么就传入root.
|
|||
pageStr := request.FormValue("page") |
|||
pageSizeStr := request.FormValue("pageSize") |
|||
userUuid := request.FormValue("userUuid") |
|||
orderCreateTime := request.FormValue("orderCreateTime") |
|||
orderSize := request.FormValue("orderSize") |
|||
|
|||
user := this.checkUser(writer, request) |
|||
if user.Role != USER_ROLE_ADMINISTRATOR { |
|||
userUuid = user.Uuid |
|||
} |
|||
|
|||
var page int |
|||
if pageStr != "" { |
|||
page, _ = strconv.Atoi(pageStr) |
|||
} |
|||
|
|||
pageSize := 200 |
|||
if pageSizeStr != "" { |
|||
tmp, err := strconv.Atoi(pageSizeStr) |
|||
if err == nil { |
|||
pageSize = tmp |
|||
} |
|||
} |
|||
|
|||
sortArray := []OrderPair{ |
|||
{ |
|||
key: "create_time", |
|||
value: orderCreateTime, |
|||
}, |
|||
{ |
|||
key: "size", |
|||
value: orderSize, |
|||
}, |
|||
} |
|||
|
|||
pager := this.securityVisitDao.Page(page, pageSize, userUuid, sortArray) |
|||
|
|||
return this.Success(pager) |
|||
} |
|||
|
|||
//删除一条记录
|
|||
func (this *SecurityVisitController) Delete(writer http.ResponseWriter, request *http.Request) *WebResult { |
|||
|
|||
uuid := request.FormValue("uuid") |
|||
if uuid == "" { |
|||
return this.Error("图片缓存的uuid必填") |
|||
} |
|||
|
|||
securityVisit := this.securityVisitDao.FindByUuid(uuid) |
|||
|
|||
//判断图片缓存的所属人是否正确
|
|||
user := this.checkUser(writer, request) |
|||
if user.Role != USER_ROLE_ADMINISTRATOR && securityVisit.UserUuid != user.Uuid { |
|||
return this.Error(CODE_WRAPPER_UNAUTHORIZED) |
|||
} |
|||
|
|||
this.securityVisitDao.Delete(securityVisit) |
|||
|
|||
return this.Success("删除成功！") |
|||
} |
|||
@ -0,0 +1,90 @@ |
|||
package rest |
|||
|
|||
import ( |
|||
"github.com/jinzhu/gorm" |
|||
_ "github.com/jinzhu/gorm/dialects/mysql" |
|||
"github.com/nu7hatch/gouuid" |
|||
"time" |
|||
) |
|||
|
|||
type SecurityVisitDao struct { |
|||
BaseDao |
|||
} |
|||
|
|||
//按照Id查询文件
|
|||
func (this *SecurityVisitDao) FindByUuid(uuid string) *SecurityVisit { |
|||
|
|||
// Read
|
|||
var securityVisit SecurityVisit |
|||
db := this.context.DB.Where(&SecurityVisit{Base: Base{Uuid: uuid}}).First(&securityVisit) |
|||
if db.Error != nil { |
|||
return nil |
|||
} |
|||
return &securityVisit |
|||
} |
|||
|
|||
//按照Id查询文件
|
|||
func (this *SecurityVisitDao) CheckByUuid(uuid string) *SecurityVisit { |
|||
|
|||
// Read
|
|||
var securityVisit SecurityVisit |
|||
db := this.context.DB.Where(&SecurityVisit{Base: Base{Uuid: uuid}}).First(&securityVisit) |
|||
this.PanicError(db.Error) |
|||
|
|||
return &securityVisit |
|||
|
|||
} |
|||
|
|||
//按分页条件获取分页
|
|||
func (this *SecurityVisitDao) Page(page int, pageSize int, userUuid string, sortArray []OrderPair) *Pager { |
|||
|
|||
var wp = &WherePair{} |
|||
|
|||
if userUuid != "" { |
|||
wp = wp.And(&WherePair{Query: "user_uuid = ?", Args: []interface{}{userUuid}}) |
|||
} |
|||
|
|||
var conditionDB *gorm.DB |
|||
conditionDB = this.context.DB.Model(&SecurityVisit{}).Where(wp.Query, wp.Args...) |
|||
|
|||
count := 0 |
|||
db := conditionDB.Count(&count) |
|||
this.PanicError(db.Error) |
|||
|
|||
var securityVisits []*SecurityVisit |
|||
db = conditionDB.Order(this.GetSortString(sortArray)).Offset(page * pageSize).Limit(pageSize).Find(&securityVisits) |
|||
this.PanicError(db.Error) |
|||
pager := NewPager(page, pageSize, count, securityVisits) |
|||
|
|||
return pager |
|||
} |
|||
|
|||
//创建
|
|||
func (this *SecurityVisitDao) Create(securityVisit *SecurityVisit) *SecurityVisit { |
|||
|
|||
timeUUID, _ := uuid.NewV4() |
|||
securityVisit.Uuid = string(timeUUID.String()) |
|||
securityVisit.CreateTime = time.Now() |
|||
securityVisit.UpdateTime = time.Now() |
|||
db := this.context.DB.Create(securityVisit) |
|||
this.PanicError(db.Error) |
|||
|
|||
return securityVisit |
|||
} |
|||
|
|||
//修改一条记录
|
|||
func (this *SecurityVisitDao) Save(securityVisit *SecurityVisit) *SecurityVisit { |
|||
|
|||
securityVisit.UpdateTime = time.Now() |
|||
db := this.context.DB.Save(securityVisit) |
|||
this.PanicError(db.Error) |
|||
|
|||
return securityVisit |
|||
} |
|||
|
|||
//删除一条记录
|
|||
func (this *SecurityVisitDao) Delete(securityVisit *SecurityVisit) { |
|||
|
|||
db := this.context.DB.Delete(&securityVisit) |
|||
this.PanicError(db.Error) |
|||
} |
|||
@ -0,0 +1,21 @@ |
|||
package rest |
|||
|
|||
/** |
|||
* 系统的所有访问记录均记录在此 |
|||
*/ |
|||
type SecurityVisit struct { |
|||
Base |
|||
SessionId string `json:"sessionId"` |
|||
UserUuid string `json:"userUuid"` |
|||
Ip string `json:"ip"` |
|||
Host string `json:"host"` |
|||
Uri string `json:"uri"` |
|||
Params string `json:"params"` |
|||
Cost int64 `json:"cost"` |
|||
Success bool `json:"success"` |
|||
} |
|||
|
|||
// set File's table name to be `profiles`
|
|||
func (SecurityVisit) TableName() string { |
|||
return TABLE_PREFIX + "security_visit" |
|||
} |
|||
@ -0,0 +1,32 @@ |
|||
package rest |
|||
|
|||
//@Service
|
|||
type SecurityVisitService struct { |
|||
Bean |
|||
securityVisitDao *SecurityVisitDao |
|||
userDao *UserDao |
|||
} |
|||
|
|||
//初始化方法
|
|||
func (this *SecurityVisitService) Init(context *Context) { |
|||
|
|||
//手动装填本实例的Bean. 这里必须要用中间变量方可。
|
|||
b := context.GetBean(this.securityVisitDao) |
|||
if b, ok := b.(*SecurityVisitDao); ok { |
|||
this.securityVisitDao = b |
|||
} |
|||
|
|||
b = context.GetBean(this.userDao) |
|||
if b, ok := b.(*UserDao); ok { |
|||
this.userDao = b |
|||
} |
|||
|
|||
} |
|||
|
|||
//获取某个文件的详情，会把父级依次倒着装进去。如果中途出错，直接抛出异常。
|
|||
func (this *SecurityVisitService) Detail(uuid string) *SecurityVisit { |
|||
|
|||
securityVisit := this.securityVisitDao.CheckByUuid(uuid) |
|||
|
|||
return securityVisit |
|||
} |
|||
Loading…
Reference in new issue