openlan/pkg/api/server.go

package api

import (
	"github.com/gorilla/mux"
	"net/http"
)

type Server struct {
	Switcher Switcher
}

func (l Server) Router(router *mux.Router) {
	router.HandleFunc("/api/server", l.List).Methods("GET")
	router.HandleFunc("/api/server/{id}", l.List).Methods("GET")
}

func (l Server) List(w http.ResponseWriter, r *http.Request) {
	server := l.Switcher.Server()
	data := &struct {
		UpTime     int64            `json:"uptime"`
		Total      int              `json:"total"`
		Statistic  map[string]int64 `json:"statistic"`
		Connection []interface{}    `json:"connection"`
	}{
		UpTime:     l.Switcher.UpTime(),
		Statistic:  server.Statistics(),
		Connection: make([]interface{}, 0, 1024),
		Total:      server.TotalClient(),
	}
	for u := range server.ListClient() {
		if u == nil {
			break
		}
		data.Connection = append(data.Connection, &struct {
			UpTime     int64            `json:"uptime"`
			LocalAddr  string           `json:"localAddr"`
			RemoteAddr string           `json:"remoteAddr"`
			Statistic  map[string]int64 `json:"statistic"`
		}{
			UpTime:     u.UpTime(),
			LocalAddr:  u.LocalAddr(),
			RemoteAddr: u.RemoteAddr(),
			Statistic:  u.Statistics(),
		})
	}
	ResponseJson(w, data)
}