Как «отключить список каталогов» и «пользовательскую страницу 404» обрабатывать одновременно - PullRequest
Новая
       7

Как «отключить список каталогов» и «пользовательскую страницу 404» обрабатывать одновременно

0 голосов
/ 08 февраля 2019

Я использую gorilla mux для маршрутизации на моем http-сервере 

https://github.com/gorilla/mux

Это мой код для отключения списка каталогов 

type justFilesFilesystem struct {
  fs               http.FileSystem
  // readDirBatchSize - configuration parameter for Readdir func
  readDirBatchSize int
}

func (fs justFilesFilesystem) Open(name string) (http.File, error) {
  f, err := fs.fs.Open(name)
  if err != nil {
    return nil, err
  }
  return neuteredStatFile{File: f, readDirBatchSize: fs.readDirBatchSize}, nil
}

type neuteredStatFile struct {
  http.File
  readDirBatchSize int
}

func (e neuteredStatFile) Stat() (os.FileInfo, error) {
  s, err := e.File.Stat()
  if err != nil {
    return nil, err
  }
  if s.IsDir() {
  LOOP:
    for {
      fl, err := e.File.Readdir(e.readDirBatchSize)
      switch err {
      case io.EOF:
        break LOOP
      case nil:
        for _, f := range fl {
          if f.Name() == "" {
            return s, err
          }
        }
      default:
        return nil, err
      }
    }
    return nil, os.ErrNotExist
  }
  return s, err
}

, и это моя основная функция 

  mux := mux.NewRouter()
  // This line why not work?
  mux.NotFoundHandler = http.HandlerFunc(NotFound)

  mux.HandleFunc("/index",HandleIndex)

  fs := justFilesFilesystem{fs: http.Dir("assets"), readDirBatchSize: 0}
  staticFileHandler := http.StripPrefix("", http.FileServer(fs))
  mux.PathPrefix("").Handler(staticFileHandler).Methods("GET")

и, наконец, мой 404 дескриптор func 

func NotFound(w http.ResponseWriter, r *http.Request) {

  fmt.Fprint(w, "404")
}

Вопрос в том, как обработать пользовательскую ошибку 404 страницы, если файл не найден, а также отключить листинг каталога в то же время.

...