Commit 466eb331 authored by szh's avatar szh

add stat_service

parent 9c84ca8d
package api
import (
"net/http"
"github.com/astaxie/beego/validation"
"github.com/gin-gonic/gin"
"chain33-pai/pkg/app"
"chain33-pai/pkg/e"
"chain33-pai/pkg/util"
"chain33-pai/service/auth_service"
)
type auth struct {
Username string `valid:"Required; MaxSize(50)"`
Password string `valid:"Required; MaxSize(50)"`
}
// @Summary Get Auth
// @Produce json
// @Param username query string true "userName"
// @Param password query string true "password"
// @Success 200 {object} app.Response
// @Failure 500 {object} app.Response
// @Router /auth [get]
func GetAuth(c *gin.Context) {
appG := app.Gin{C: c}
valid := validation.Validation{}
username := c.Query("username")
password := c.Query("password")
a := auth{Username: username, Password: password}
ok, _ := valid.Valid(&a)
if !ok {
app.MarkErrors(valid.Errors)
appG.Response(http.StatusBadRequest, e.INVALID_PARAMS, nil)
return
}
authService := auth_service.Auth{Username: username, Password: password}
isExist, err := authService.Check()
if err != nil {
appG.Response(http.StatusInternalServerError, e.ERROR_AUTH_CHECK_TOKEN_FAIL, nil)
return
}
if !isExist {
appG.Response(http.StatusUnauthorized, e.ERROR_AUTH, nil)
return
}
token, err := util.GenerateToken(username, password)
if err != nil {
appG.Response(http.StatusInternalServerError, e.ERROR_AUTH_TOKEN, nil)
return
}
appG.Response(http.StatusOK, e.SUCCESS, map[string]string{
"token": token,
})
}
package stat_service
import (
"chain33-pai/models"
)
type RaspMinerStat struct {
Addr string `json:"addr"`
MinedAmount int64 `json:"mined_amount"`
Height int64 `json:"height"`
Time int64 `json:"time"`
}
func (s *RaspMinerStat) GetMinerStat(addr string) (*models.RaspMinerStat,error) {
return models.GetAddr(addr)
}
func (s *RaspMinerStat) SetMineTime(addr string,start ,end int64) error {
stat,err := models.GetAddr(addr)
if err != nil {
return err
}
m := map[string]interface{}{"miner_start_time":start,"miner_end_time":end}
if stat.MinerStartTime <= 0 {
m["miner_start_time"] = start
m["miner_end_time"] = 0
}
if stat.MinerStartTime > 0 && stat.MinerEndTime <= 0 {
m["miner_start_time"] = stat.MinerStartTime
m["miner_end_time"] = end
}
return models.EditAddr(addr,m)
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment