github.com/tickoalcantara12/micro/v3@v3.0.0-20221007104245-9d75b9bcbab9/service/debug/log/kubernetes/stream.go (about)

     1  // Licensed under the Apache License, Version 2.0 (the "License");
     2  // you may not use this file except in compliance with the License.
     3  // You may obtain a copy of the License at
     4  //
     5  //     https://www.apache.org/licenses/LICENSE-2.0
     6  //
     7  // Unless required by applicable law or agreed to in writing, software
     8  // distributed under the License is distributed on an "AS IS" BASIS,
     9  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    10  // See the License for the specific language governing permissions and
    11  // limitations under the License.
    12  //
    13  // Original source: github.com/micro/go-micro/v3/debug/log/kubernetes/stream.go
    14  
    15  package kubernetes
    16  
    17  import (
    18  	"encoding/json"
    19  	"fmt"
    20  	"os"
    21  	"sync"
    22  
    23  	"github.com/tickoalcantara12/micro/v3/service/debug/log"
    24  )
    25  
    26  func write(l log.Record) error {
    27  	m, err := json.Marshal(l)
    28  	if err == nil {
    29  		_, err := fmt.Fprintf(os.Stderr, "%s", m)
    30  		return err
    31  	}
    32  	return err
    33  }
    34  
    35  type kubeStream struct {
    36  	// the k8s log stream
    37  	stream chan log.Record
    38  	sync.Mutex
    39  	// the stop chan
    40  	stop chan bool
    41  }
    42  
    43  func (k *kubeStream) Chan() <-chan log.Record {
    44  	return k.stream
    45  }
    46  
    47  func (k *kubeStream) Stop() error {
    48  	k.Lock()
    49  	defer k.Unlock()
    50  	select {
    51  	case <-k.stop:
    52  		return nil
    53  	default:
    54  		close(k.stop)
    55  		close(k.stream)
    56  	}
    57  	return nil
    58  }