143 lines
3.4 KiB
Go
143 lines
3.4 KiB
Go
package server
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"git.sr.ht/~akilan1999/p2p-rendering-computation/config"
|
|
"git.sr.ht/~akilan1999/p2p-rendering-computation/p2p"
|
|
"git.sr.ht/~akilan1999/p2p-rendering-computation/server/docker"
|
|
"github.com/gin-gonic/gin"
|
|
"io/ioutil"
|
|
"net/http"
|
|
)
|
|
|
|
func Server() error{
|
|
r := gin.Default()
|
|
|
|
// Gets default information of the server
|
|
r.GET("/server_info", func(c *gin.Context) {
|
|
c.JSON(http.StatusOK, ServerInfo())
|
|
})
|
|
|
|
// Speed test with 50 mbps
|
|
r.GET("/50", func(c *gin.Context){
|
|
// Get Path from config
|
|
config, err := config.ConfigInit()
|
|
if err != nil {
|
|
c.String(http.StatusOK, fmt.Sprint(err))
|
|
}
|
|
c.File(config.SpeedTestFile)
|
|
})
|
|
|
|
// Route build to do a speed test
|
|
r.GET("/upload", func(c *gin.Context) {
|
|
file, _ := c.FormFile("file")
|
|
|
|
// Upload the file to specific dst.
|
|
// c.SaveUploadedFile(file, dst)
|
|
|
|
c.String(http.StatusOK, fmt.Sprintf("'%s' uploaded!", file.Filename))
|
|
})
|
|
|
|
//Gets Ip Table from server node
|
|
r.POST("/IpTable", func(c *gin.Context) {
|
|
// Getting IPV4 address of client
|
|
var ClientHost p2p.IpAddress
|
|
ClientHost.Ipv4 = c.ClientIP()
|
|
// Variable to store IP table information
|
|
var IPTable p2p.IpAddresses
|
|
//Add Client IP address to IPTable struct
|
|
IPTable.IpAddress = append(IPTable.IpAddress, ClientHost)
|
|
|
|
// Receive file from POST request
|
|
body, err := c.FormFile("json")
|
|
if err != nil {
|
|
c.String(http.StatusOK, fmt.Sprint(err))
|
|
}
|
|
|
|
// Open file
|
|
open, err := body.Open()
|
|
if err != nil {
|
|
c.String(http.StatusOK, fmt.Sprint(err))
|
|
}
|
|
|
|
// Open received file
|
|
file, err := ioutil.ReadAll(open)
|
|
if err != nil {
|
|
c.String(http.StatusOK, fmt.Sprint(err))
|
|
}
|
|
|
|
json.Unmarshal(file,&IPTable)
|
|
|
|
// Runs speed test to return only servers in the IP table pingable
|
|
err = IPTable.SpeedTestUpdatedIPTable()
|
|
if err != nil {
|
|
c.String(http.StatusOK, fmt.Sprint(err))
|
|
}
|
|
|
|
// Reads IP addresses from ip table
|
|
IpAddresses,err := p2p.ReadIpTable()
|
|
if err != nil {
|
|
c.String(http.StatusOK, fmt.Sprint(err))
|
|
}
|
|
|
|
c.JSON(http.StatusOK, IpAddresses)
|
|
})
|
|
|
|
// Starts docker container in server
|
|
r.GET("/startcontainer", func(c *gin.Context) {
|
|
// Get Number of ports to open and whether to use GPU or not
|
|
Ports := c.DefaultQuery("ports","0")
|
|
GPU := c.DefaultQuery("GPU","false")
|
|
var PortsInt int
|
|
|
|
// Convert Get Request value to int
|
|
fmt.Sscanf(Ports, "%d", &PortsInt)
|
|
|
|
// Creates container and returns back result to
|
|
// access container
|
|
resp, err := docker.BuildRunContainer(PortsInt,GPU)
|
|
|
|
if err != nil {
|
|
c.String(http.StatusInternalServerError, fmt.Sprintf("error: %s", err))
|
|
}
|
|
|
|
c.JSON(http.StatusOK, resp)
|
|
})
|
|
|
|
//Remove container
|
|
r.GET("/RemoveContainer", func(c *gin.Context) {
|
|
ID := c.DefaultQuery("id","0")
|
|
if err := docker.StopAndRemoveContainer(ID); err != nil {
|
|
c.String(http.StatusInternalServerError, fmt.Sprintf("error: %s", err))
|
|
}
|
|
c.String(http.StatusOK, "success")
|
|
})
|
|
|
|
// Future feature
|
|
/*r.GET("/create_vm/:virtualization", func(c *gin.Context) {
|
|
virtualization := c.Param("virtualization")
|
|
// Runs based on Preallocated VM size
|
|
if virtualization == "docker" {
|
|
sshinfo,err := docker.RunVM()
|
|
if err != nil {
|
|
c.String(http.StatusInternalServerError, fmt.Sprintf("error: %s", err))
|
|
}
|
|
if sshinfo != nil {
|
|
c.JSON(http.StatusOK, sshinfo)
|
|
}
|
|
|
|
} else {
|
|
c.String(200,"virtualization tool not selected")
|
|
}
|
|
})*/
|
|
|
|
// Port running on
|
|
err := r.Run(":8088")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|