85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
// +build ignore
|
|
|
|
// This example illustrates a streaming Gemini server.
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"sync"
|
|
"time"
|
|
|
|
"git.sr.ht/~adnano/go-gemini"
|
|
"git.sr.ht/~adnano/go-gemini/certificate"
|
|
)
|
|
|
|
func main() {
|
|
certificates := &certificate.Store{}
|
|
certificates.Register("localhost")
|
|
if err := certificates.Load("/var/lib/gemini/certs"); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
mux := &gemini.ServeMux{}
|
|
mux.HandleFunc("/", stream)
|
|
|
|
server := &gemini.Server{
|
|
Handler: mux,
|
|
ReadTimeout: 30 * time.Second,
|
|
WriteTimeout: 1 * time.Minute,
|
|
GetCertificate: certificates.GetCertificate,
|
|
}
|
|
|
|
var shutdownOnce sync.Once
|
|
var wg sync.WaitGroup
|
|
wg.Add(1)
|
|
defer wg.Wait()
|
|
mux.HandleFunc("/shutdown", func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
|
|
fmt.Fprintln(w, "Shutting down...")
|
|
w.Flush()
|
|
go shutdownOnce.Do(func() {
|
|
server.Shutdown(context.Background())
|
|
wg.Done()
|
|
})
|
|
})
|
|
|
|
if err := server.ListenAndServe(context.Background()); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
// stream writes an infinite stream to w.
|
|
func stream(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
|
|
ch := make(chan string)
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
|
|
go func(ctx context.Context) {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
default:
|
|
ch <- fmt.Sprint(time.Now().UTC())
|
|
}
|
|
time.Sleep(time.Second)
|
|
}
|
|
// Close channel when finished.
|
|
// In this example this will never be reached.
|
|
close(ch)
|
|
}(ctx)
|
|
|
|
for {
|
|
s, ok := <-ch
|
|
if !ok {
|
|
break
|
|
}
|
|
fmt.Fprintln(w, s)
|
|
if err := w.Flush(); err != nil {
|
|
cancel()
|
|
return
|
|
}
|
|
}
|
|
}
|