apps/kubevpn
1
0
Fork 0
mirror of https://github.com/kubenetworks/kubevpn.git synced 2025-10-30 18:26:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: set cleanup timeout to 5 seconds

Browse Source
This commit is contained in:
fengcaiwen
2023-04-09 17:07:54 +08:00
parent 2b97dd3038
commit b2a6596405
8 changed files with 62 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/webhook/dhcp.go
View File

@@ -1,6 +1,7 @@
package webhook
import (
"context"
"fmt"
"net"
"net/http"
@@ -21,11 +22,12 @@ type dhcpServer struct {
func (d *dhcpServer) rentIP(w http.ResponseWriter, r *http.Request) {
podName := r.Header.Get(config.HeaderPodName)
namespace := r.Header.Get(config.HeaderPodNamespace)
ctx := context.Background()
log.Infof("handling rent ip request, pod name: %s, ns: %s", podName, namespace)
cmi := d.clientset.CoreV1().ConfigMaps(namespace)
dhcp := handler.NewDHCPManager(cmi, namespace)
v4, v6, err := dhcp.RentIPRandom()
v4, v6, err := dhcp.RentIPRandom(ctx)
if err != nil {
log.Error(err)
w.WriteHeader(http.StatusBadRequest)
@@ -56,7 +58,7 @@ func (d *dhcpServer) releaseIP(w http.ResponseWriter, r *http.Request) {
log.Infof("handling release ip request, pod name: %s, ns: %s", podName, namespace)
cmi := d.clientset.CoreV1().ConfigMaps(namespace)
dhcp := handler.NewDHCPManager(cmi, namespace)
if err := dhcp.ReleaseIP(ips...); err != nil {
if err := dhcp.ReleaseIP(context.Background(), ips...); err != nil {
log.Error(err)
w.WriteHeader(http.StatusBadRequest)
return