实现
为了实现在飞牛上敲写代码，我将docker容器加上了iframe加载，然后变成了用systemctl来挂载code-server服务，最后想用飞牛统一网关，但是code-server默认运行在/根目录，而飞牛的统一网关要加/app/xxx的前缀，解决办法就是加中间层，最开始用的是node.js写了个中间层，结果是websocket连接不上，后面用go来手写很意外的成功了，然后后想手写优化，优化了一下就出来之前的问题，如下图所示：
后面，反复修改终于确认了问题根源，之前的nodejs版本就是这个问题解决不了，在代理给 Unix Socket 时，请求头里的 Host 可能变成了 unix 或者与 Origin 的协议不匹配。code-server 看到 Origin 和 Host 不一致，触发了安全防御机制，直接掐断了连接（导致 1006 错误）。而手动拼接字符串粗糙代码，手动复制了 Sec-WebSocket-* 和 Cookie，完全漏掉了浏览器的 Origin 请求头，code-server 发现没有 Origin 头，以为是某种非浏览器的本地客户端发起的请求，直接跳过了安全校验，所以放行了就连接成功了，所以修复起来就非常简单：要么让 Host 和 Origin 完美匹配，要么直接把 Origin 头删掉来绕过校验，这下终于code-server用上了统一网关了，fpk下载地址：https://github.com/ctllo-bit/coder
go中间层
这个golang转发层是最核心的，最初以为只要改写路径就可以了，结果是websocket连不上，才知道websocket也要转发，转发了还是1006 的错误，才又知道有个CORS的跨域与主机头安全校验的问题，最终都解决了，就写出了下面的http/websocket统一代理层，代码如下所示：
package main
import (
"context"
"flag"
"log"
"net"
"net/http"
"net/http/httputil"
"os"
"os/signal"
"strings"
"syscall"
"time"
)
var (
codeServerSocket = flag.String("socket", "/var/apps/coder/var/code-server.sock", "upstream code-server unix socket")
proxySocket = flag.String("proxy-socket", "/var/apps/coder/target/coder-proxy.sock", "this proxy's own unix socket")
prefix = flag.String("prefix", "/app/coder", "URL prefix to strip before forwarding")
)
func main() {
flag.Parse()
// 不再使用 httputil.NewSingleHostReverseProxy，直接初始化 ReverseProxy 结构体
backend := &httputil.ReverseProxy{
// ====== 使用 Rewrite 替代 Director ======
Rewrite: func(pr *httputil.ProxyRequest) {
pr.SetXForwarded() // 自动处理 X-Forwarded-* 头
// 处理路径前缀剥离
path := pr.In.URL.Path
if strings.HasPrefix(path, *prefix) {
path = strings.TrimPrefix(path, *prefix)
if !strings.HasPrefix(path, "/") {
path = "/" + path
}
}
// 设置目标 Scheme 和 Host
pr.Out.URL.Scheme = "http"
pr.Out.URL.Host = "unix"
pr.Out.URL.Path = path
// 将外部真实的 Host 传递给后端
pr.Out.Host = pr.In.Host
// 删除 Origin 头，绕过 code-server 严格的同源检测
pr.Out.Header.Del("Origin")
},
Transport: &http.Transport{
DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
var d net.Dialer
return d.DialContext(ctx, "unix", *codeServerSocket)
},
MaxIdleConns: 100,
IdleConnTimeout: 90 * time.Second,
},
ModifyResponse: func(r *http.Response) error {
if loc := r.Header.Get("Location"); strings.HasPrefix(loc, "/") && !strings.HasPrefix(loc, *prefix) {
r.Header.Set("Location", *prefix+loc)
}
return nil
},
ErrorHandler: func(w http.ResponseWriter, r *http.Request, err error) {
log.Printf("proxy error: %s %s -> %v", r.Method, r.URL.Path, err)
if strings.Contains(err.Error(), "connection refused") || strings.Contains(err.Error(), "no such file") {
http.Error(w, "code-server is not running or the socket does not exist", http.StatusServiceUnavailable)
} else {
http.Error(w, "Bad Gateway", http.StatusBadGateway)
}
},
}
if err := os.RemoveAll(*proxySocket); err != nil {
log.Fatalf("failed to remove old proxy socket: %v", err)
}
listener, err := net.Listen("unix", *proxySocket)
if err != nil {
log.Fatalf("failed to listen on proxy socket %s: %v", *proxySocket, err)
}
if err := os.Chmod(*proxySocket, 0666); err != nil {
log.Printf("warning: failed to set socket permissions: %v", err)
}
// 优雅停机逻辑
server := &http.Server{
Handler: backend, // 直接将 backend 作为 Server 的 Handler
}
go func() {
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
<-sigCh
log.Println("shutting down proxy...")
// 给活动连接 5 秒钟时间完成关闭
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
if err := server.Shutdown(ctx); err != nil {
log.Printf("proxy shutdown error: %v", err)
}
}()
log.Printf("coder proxy listening on %s", *proxySocket)
log.Printf("upstream: unix://%s prefix=%s", *codeServerSocket, *prefix)
// 使用 server.Serve 替代 http.Serve
if err := server.Serve(listener); err != nil && err != http.ErrServerClosed {
log.Fatalf("server error: %v", err)
}
os.RemoveAll(*proxySocket)
}
go程序编译：
GOOS=linux GOARCH=amd64 go build -o coder-proxy-linux-amd64 main.go
GOOS=linux GOARCH=arm64 go build -o coder-proxy-linux-arm64 main.go
