TODO: fix expand all

goserver/main.go

@@ -2,9 +2,12 @@ package main
 
 import (
 	"database/sql"
+	"encoding/json"
 	"fmt"
 	"log"
+	"net/http"
 	"strconv"
+	"strings"
 	"time"
 
 	"github.com/gin-gonic/gin"
@@ -65,63 +68,144 @@ func init() {
 type ObjectTwistcasting struct {
 	Uid           int64
 	Platform      string
-	UserId        int64
+	UserId        string
 	UserName      string
 	LiveUrl       string
-	Tags          string
+	Tags          []string
 	Followers     int64
 	Views         int64
 	Gratuity      int64
 	LiveTitle     string
-	LiveStartTime time.Time
-	LiveEndTime   time.Time
-	UpdateTime    time.Time
+	LiveStartTime *time.Time
+	LiveEndTime   *time.Time
+	UpdateTime    *time.Time
 }
 
-func TwistcastingQuery(cxt *gin.Context) {
+type Result struct {
+	Code  int
+	Error string
+	Data  interface{}
+}
+
+func TwitcastingQuery(cxt *gin.Context) {
 	var err error
-	sstart := cxt.Query("start")
-	send := cxt.Query("end")
-	_, err = strconv.Atoi(sstart)
+	page, err := strconv.Atoi(cxt.Query("page"))
 	if err != nil {
 		cxt.Error(err)
 	}
-	_, err = strconv.Atoi(send)
+	psize, err := strconv.Atoi(cxt.Query("psize"))
 	if err != nil {
 		cxt.Error(err)
 	}
+	if psize > 100 {
+		cxt.Error(fmt.Errorf("page size <= 100"))
+	}
 
-	ssql := fmt.Sprintf(SqlQuery, sstart, send)
+	start := (page - 1) * 200
+	// end := start + 200
 
+	ssql := fmt.Sprintf(SqlQuery, strconv.Itoa(start), strconv.Itoa(psize))
 	rows, err := StoreStreamer.Query(ssql)
 	if err != nil {
-		log.Panic(err)
+		cxt.Error(err)
 	}
 	var ots []*ObjectTwistcasting
 	for rows.Next() {
 		ot := &ObjectTwistcasting{}
-		rows.Scan(
+		var lstm, letm, utm sql.NullTime
+		var tags string
+		err = rows.Scan(
 			&ot.Uid,
 			&ot.Platform,
 			&ot.UserId,
+			&ot.UserName,
 			&ot.LiveUrl,
-			&ot.Tags,
+			&tags,
 			&ot.Followers,
 			&ot.Views,
 			&ot.Gratuity,
 			&ot.LiveTitle,
-			&ot.LiveStartTime,
-			&ot.LiveEndTime,
-			&ot.UpdateTime,
+			&lstm,
+			&letm,
+			&utm,
 		)
+		if err != nil {
+			cxt.Error(err)
+			return
+		}
+		if !lstm.Valid {
+			ot.LiveStartTime = nil
+		} else {
+			ot.LiveStartTime = &lstm.Time
+		}
+		if !letm.Valid {
+			ot.LiveEndTime = nil
+		} else {
+			ot.LiveEndTime = &letm.Time
+		}
+		if !utm.Valid {
+			ot.UpdateTime = nil
+		} else {
+			ot.UpdateTime = &utm.Time
+		}
+
+		if err = json.Unmarshal([]byte(tags), &ot.Tags); err != nil {
+			log.Println(tags)
+		}
+
 		ots = append(ots, ot)
 	}
+
+	r := &Result{Code: 200}
+	r.Data = ots
 	log.Println(len(ots))
+	if retdata, err := json.Marshal(r); err != nil {
+		cxt.Error(err)
+	} else {
+		cxt.JSON(r.Code, string(retdata))
+	}
+}
+
+func Cors() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		method := c.Request.Method               //请求方法
+		origin := c.Request.Header.Get("Origin") //请求头部
+		var headerKeys []string                  // 声明请求头keys
+		for k, _ := range c.Request.Header {
+			headerKeys = append(headerKeys, k)
+		}
+		headerStr := strings.Join(headerKeys, ", ")
+		if headerStr != "" {
+			headerStr = fmt.Sprintf("access-control-allow-origin, access-control-allow-headers, %s", headerStr)
+		} else {
+			headerStr = "access-control-allow-origin, access-control-allow-headers"
+		}
+		if origin != "" {
+			c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
+			c.Header("Access-Control-Allow-Origin", "*")                                       // 这是允许访问所有域
+			c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE,UPDATE") //服务器支持的所有跨域请求的方法,为了避免浏览次请求的多次'预检'请求
+			//  header的类型
+			c.Header("Access-Control-Allow-Headers", "Authorization, Content-Length, X-CSRF-Token, Token,session,X_Requested_With,Accept, Origin, Host, Connection, Accept-Encoding, Accept-Language,DNT, X-CustomHeader, Keep-Alive, User-Agent, X-Requested-With, If-Modified-Since, Cache-Control, Content-Type, Pragma")
+			//              允许跨域设置                                                                                                      可以返回其他子段
+			c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers,Cache-Control,Content-Language,Content-Type,Expires,Last-Modified,Pragma,FooBar") // 跨域关键设置 让浏览器可以解析
+			c.Header("Access-Control-Max-Age", "172800")                                                                                                                                                           // 缓存请求信息 单位为秒
+			c.Header("Access-Control-Allow-Credentials", "false")                                                                                                                                                  //  跨域请求是否需要带cookie信息 默认设置为true
+			c.Set("content-type", "application/json")                                                                                                                                                              // 设置返回格式是json
+		}
+
+		//放行所有OPTIONS方法
+		if method == "OPTIONS" {
+			c.JSON(http.StatusOK, "Options Request!")
+		}
+		// 处理请求
+		c.Next() //  处理请求
+	}
 }
 
 func main() {
 	engine := gin.New() //r := gin.Default() //使用默认中间件
 	engine.Use(gin.Logger())
-	engine.GET("twistcasting/data", TwistcastingQuery)
+	engine.Use(Cors())
+	engine.GET("twitcasting/query", TwitcastingQuery)
 	engine.Run(":5500")
 }

goserver/main_test.go

@@ -0,0 +1,7 @@
+package main
+
+import "testing"
+
+func TestMain(t *testing.T) {
+	main()
+}